Question

在Snowflake中有一个要求，我必须先生成一些SQL，然后执行它以创建新表。

我已经通过创建UDF（目前已进行硬编码）成功生成了create table语句

CREATE OR REPLACE FUNCTION COOL_CARGO.test()
  RETURNS STRING 
  AS 
  $$
  SELECT substr(regexp_replace(GET_DDL('TABLE', 'COOL_CARGO.DIM_BRANCH'),('DIM_BRANCH'),'COOL_CARGO.DIM_BRANCH_ERR'), 0, LENGTH(regexp_replace(GET_DDL('TABLE', 'COOL_CARGO.DIM_BRANCH'),('DIM_BRANCH'),'COOL_CARGO.DIM_BRANCH_ERR')) -2)||','||'  etl_err_date timestamp NOT NULL DEFAULT CURRENT_TIMESTAMP,
  etl_id_run int DEFAULT NULL,
  etl_err_noe int DEFAULT NULL,
  etl_err_desc varchar(512) DEFAULT NULL,
  etl_err_col varchar(256) DEFAULT NULL,
  etl_err_cod varchar(256) DEFAULT NULL'||');'
 $$
 ;

这将输出以下

create or replace TABLE COOL_CARGO.DIM_BRANCH_ERR (
    TK_BRANCH NUMBER(38,0),
    GB_BRANCH_CODE VARCHAR(256),
    GB_BRANCH_NAME VARCHAR(256),
    GB_BRANCH_CITY VARCHAR(256),
    GB_BRANCH_STATE VARCHAR(256),
    BG_BRANCH_HOME_PORT VARCHAR(256),
    BG_BRANCH_COUNTRY_CODE VARCHAR(256),
    BG_BRANCH_COUNTRY_NAME VARCHAR(256)
,  etl_err_date timestamp NOT NULL DEFAULT CURRENT_TIMESTAMP,
  etl_id_run int DEFAULT NULL,
  etl_err_noe int DEFAULT NULL,
  etl_err_desc varchar(512) DEFAULT NULL,
  etl_err_col varchar(256) DEFAULT NULL,
  etl_err_cod varchar(256) DEFAULT NULL);

我现在需要创建一个将执行此create table语句的UDF，但是由于它似乎仅返回诸如字符串之类的内容，因此我无法通过例如从另一个函数调用它来使其执行。

CREATE OR REPLACE FUNCTION COOL_CARGO.run_test()
     RETURNS string
     AS
     $$
     COOL_CARGO.test()
      $$
     ;

然后我尝试运行该函数以创建表

select COOL_CARGO.run_test();

我不知道我想做些什么，如果不可能的话我会很生气。

这可以在Snowflake DB中完成吗？

Answer 1

您可以通过在 2019 中推出的Snowflake的新Stored Procedures feature来实现这一目标。它允许您构建一个任意的SQL字符串并使用SQL或JavaScript执行执行。

您现有的方法可以轻松地应用于存储过程中（以下示例是说明性的，尚未经过专门测试）：

create or replace procedure test()
  returns null
  language javascript
  as
  $$
  var orig_ddl_qry = "SELECT GET_DDL('TABLE', 'COOL_CARGO.DIM_BRANCH')";
  var get_ddl_stmt = snowflake.createStatement(
         {
         sqlText: orig_ddl_qry
         }
      );
  var get_ddl_res = get_ddl_stmt.execute();
  get_ddl_res.next();
  var orig_ddl_str = res.getColumnValue(1);
  var replaced_ddl_open = orig_ddl_str.replace("DIM_BRANCH", "COOL_CARGO.DIM_BRANCH_ERR").slice(0, -1);
  var new_ddl = replaced_ddl_open + ", etl_err_date timestamp NOT NULL DEFAULT CURRENT_TIMESTAMP, etl_id_run int DEFAULT NULL, etl_err_noe int DEFAULT NULL, etl_err_desc varchar(512) DEFAULT NULL, etl_err_col varchar(256) DEFAULT NULL, etl_err_cod varchar(256) DEFAULT NULL);";
  var create_stmt = snowflake.createStatement(
         {
         sqlText: new_ddl
         }
      );
  var create_ddl_res = create_stmt.execute();
  $$
  ;

CALL test();

Snowflake DB UDF可以执行SQL命令吗？

1 个答案: