Snowflake SQL 存储过程并使用动态 SQL 保存查询值

Question

我正在用 Snowflake 编写 SQL 存储过程（语言 SQL 不是 Javascript）。

我正在尝试将 SELECT 语句中的计数保存到一个变量中，table 语句的 select 名称需要来自一个变量。

这是我目前所拥有的，但这是失败的。我不知道将 USING 放在哪里，或者我是否可以这样做？我觉得我只是在语法上还不正确。

create or replace procedure myprocedure(DBNAME varchar(16777216))
  returns int
  language sql
  as
  $$
    DECLARE
        excludeCount int;
        fullyQualifiedProceduresTable  varchar(16777216);    
        
    BEGIN
        fullyQualifiedProceduresTable := CONCAT(DBNAME, '.INFORMATION_SCHEMA.PROCEDURES');
        excludeCount := (SELECT count(*) as count from TABLE (?) WHERE PROCEDURE_OWNER = '<ROLE NAME>') USING fullyQualifiedProceduresTable ;
        
        IF (excludeCount > 0) THEN
            RETURN 1;
        ELSE
            RETURN 0;
        END IF;
    END;
$$;

Answer 1

我不确定你为什么需要一个程序。你能不能不...

set table_name='abc';
set excludeCount=(select case when count(*)>0 then 1 else 0 end from identifier($table_name));

Answer 2

试试这个

call my_proc('bus_day');

create or replace procedure my_proc(table_name varchar)
returns table(a integer)
language sql
as
$$
  declare
    res RESULTSET;
    query varchar default 'SELECT count(*) FROM ' || :table_name ;
  begin
    res := (execute immediate :query);
    return table (res);
  end;
$$;

Answer 3

这就是我使用略有不同的语法来使用变量的工作。语法取自 here

create or replace procedure myprocedure(DBNAME varchar(16777216))
  returns int
  language sql
  as
  $$
    DECLARE
        excludeCount int;
        fullyQualifiedProceduresTable  varchar(16777216);    
        
    BEGIN
        SELECT CONCAT(:DBNAME, '.INFORMATION_SCHEMA.PROCEDURES') into :fullyQualifiedProceduresTable;
        SELECT count(*) into :excludeCount from IDENTIFIER(:fullyQualifiedProceduresTable) WHERE PROCEDURE_OWNER = '<role>';
        
        IF (excludeCount > 0) THEN
            RETURN 1;
        ELSE
            RETURN 0;
        END IF;
    END;
$$;

Snowflake SQL 存储过程并使用动态 SQL 保存查询值

Snowflake SQL stored procedure and save value from query with dynamic SQL

sql

scripting

snowflake-cloud-data-platform