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

Question

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

例子：

   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;

    $$;

我对此很陌生。 有人可以解释如何解决这个问题。

Answer 1

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

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

Answer 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
$$
;