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

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

2 个答案: