如何将表的列名存储在雪花的数组中

Posted 2023-03-23

【中文标题】如何将表的列名存储在雪花的数组中

【英文标题】：How to store the column names of a table in an array in Snowflake

【发布时间】：2020-09-10 00:38:46

【问题描述】：

我对 Snowflake 很陌生。我只需要检索表的列名，然后在存储过程中将其作为数组返回。

例子：

   Column_A    Column_B    Column_C    Column_D
      1           2           2           2

应该返回 [Column_A, Column_B, Column_C, Column_D]

这是我的尝试：

create or replace procedure sample()
    return RETURNTYPE?
    language javascript
    AS
    $$
      var cmd = 'SELECT COLUMN_NAME' FROM table1;
      var statement = snowflake.createStatement(sqlTest: cmd);
      var res = statement.execute();
      var arr = [];
      //HOW CAN I PUSH COLUMN NAMES INTO THIS ARRAY
      
      return ARRAY;

    $$;

我是新手。谁能解释一下如何解决这个问题。

【参考方案1】：

我会简单地利用information_schema 和array_agg() 函数来完成这项工作，而不是使用存储过程。像这样的东西在没有所有 javascript 作为 Snowflake 中的本机函数的情况下也能工作：

SELECT array_agg(column_name)
FROM information_schema.columns
where table_name = 'TABLE_NAME';

【参考方案2】：

查看 Mike Walton 的不使用存储过程的解决方案。将此答案留给任何寻求存储过程答案的人。

---

一种解决方案是在存储过程中运行SELECT *，然后语句将具有列的名称：

CREATE OR REPLACE PROCEDURE get_columns(TABLE_NAME VARCHAR)
RETURNS ARRAY
LANGUAGE JAVASCRIPT
AS
$$
var stmt = snowflake.createStatement(
    sqlText: "SELECT * FROM " + TABLE_NAME + " LIMIT 1;",    
);
stmt.execute();

var cols=[];
for (i = 1; i <= stmt.getColumnCount(); i++) 
  cols.push(stmt.getColumnName(i));

return cols
$$
;

【讨论】：

我猜@R0bert 想要一个存储过程