Question

我想获取雪花列表中表的所有主键。我用过

SHOW PRIMARY KEYS IN TABLE t1

但输出是一组列：

"created_on"    "database_name" "schema_name"   "table_name"    "column_name"   "key_sequence"  "constraint_name"   "comment"

而且我只需要 column_name 列。我试着做：

select a.* from (SHOW PRIMARY KEYS IN TABLE t1) a

但我收到此错误：

SQL Error [1003] [42000]: SQL compilation error:
syntax error line 1 at position 17 unexpected 'SHOW'.

如果有人有想法，请帮忙。谢谢。

Answer 1

您可以使用 RESULT_SCAN 函数来处理 SHOW 命令的输出：

https://docs.snowflake.com/en/sql-reference/functions/result_scan.html

SHOW PRIMARY KEYS IN TABLE t1;

SELECT * FROM table(RESULT_SCAN( LAST_QUERY_ID() ));

另一个示例：

SHOW PRIMARY KEYS IN TABLE test1;
select last_query_id();

This is the result:
-------------------
019c7ab2-3200-9e9b-0000-b4550014773e

然后我将它与 RESULT_SCAN 一起使用：

SELECT * FROM table(RESULT_SCAN( '019c7ab2-3200-9e9b-0000-b4550014773e'));

请注意，last_query_id 将始终返回当前会话中的最后一个查询。它不会返回与其他会话一起执行的查询 ID。

Answer 2

不久前，我为一位需要以编程方式将表的 PK 列作为表返回的客户编写了 UDTF（用户定义表函数）。这可能与您要执行的操作一致，但如果您希望将返回作为列表而不是表格，则可能需要进行一些调整。

您可以在此处查看其工作原理示例https://snowflake.pavlik.us/index.php/2020/07/29/getting-snowflake-primary-key-columns-as-a-table/

/********************************************************************************************************
*                                                                                                       *
* User defined table function (UDTF) to get primary keys for a table.                                   *
*                                                                                                       *
* @param  {string}:  TABLE_DDL    The DDL for the table to get the PKs. Usually use get_ddl.            *
* @return {table}:                A table with the columns comprising the table's primary key           *
*                                                                                                       *
********************************************************************************************************/
create or replace function GET_PK_COLUMNS(TABLE_DDL string)
returns table (PK_COLUMN string)
language javascript
as
$$
{
    processRow: function get_params(row, rowWriter, context){
  
        var pkCols = getPKs(row.TABLE_DDL);
        for (i = 0; i < pkCols.length; i++) {
            rowWriter.writeRow({PK_COLUMN: pkCols[i]}); 
        }
          
        function getPKs(tableDDL) {
            var c;
            var keyword = "primary key";
            var ins = -1;
            var s = tableDDL.split("\n");
            for (var i = 0; i < s.length; i++) {  
                ins = s[i].indexOf(keyword);
                if (ins != -1) {
                    var colList = s[i].substring(ins + keyword.length);
                    colList = colList.replace("(", "");
                    colList = colList.replace(")", "");
                    var colArray = colList.split(",");
                    for (pkc = 0; c < colArray.length; pkc++) {
                        colArray[pkc] = colArray[pkc].trim();
                    }
                    return colArray;
                }
            }
            return [];  // No PK
        }
    }
}
$$;

如何获取雪花中的表主键列表？

2 个答案: