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

Question

我对 Snowflake 很陌生。我只需要检索 table 的列名，然后 return 它作为存储过程中的数组。

示例：

   Column_A    Column_B    Column_C    Column_D
      1           2           2           2

应该return[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

检查 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
$$
;

Answer 2

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

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

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

How to store the column names of a table in an array in Snowflake

javascript

sql

stored-procedures

snowflake-cloud-data-platform