使用传递的变量插入表雪花

Posted 2023-03-29

【中文标题】使用传递的变量插入表雪花

【英文标题】：Inserting into a table snowflake with a passed variable

【发布时间】：2021-09-17 17:21:23

【问题描述】：

我有一个有 3 个属性的表

属性数据类型标志（0 或 1）

我的存储过程采用表单中的一串记录 “Att1 Datatype1 1，Att2 Datatype2 0，Att3 Datatype3 0...） 所以例如传递的字符串可以是 "DATABASE VARCHAR 1，SCHEMA VARCHAR 1，TIMESTAMP TIMESTAMP 0"

我的代码获取表的列名并将其存储在字符串中 我的代码采用传递的字符串并将它们放入由 , 分隔的数组元素中 但是，当我尝试插入我的表时，问题就出现了 我不断收到错误“无效标识符”

CREATE OR REPLACE PROCEDURE "ADMINDB"."TOOLKIT".ADD_ATTRIBUTES_SESSION_META ("P_ATTRIBUTE_DATATYPE_FLAG" VARCHAR(16777216))
RETURNS VARCHAR
LANGUAGE javascript
COMMENT='Add Attributes to the table SESSION_ATTRIBUTES_META'
EXECUTE AS CALLER
AS
$$
  var v_list = P_ATTRIBUTE_DATATYPE_FLAG;
  var arr_list = [];
  arr_list = P_ATTRIBUTE_DATATYPE_FLAG.split(',');
  var v_string;
  var arr_col_att = [];
  var v_sqlCode = `SELECT * FROM ` + "ADMINDB" + "." + "TOOLKIT" + "." + "SESSION_ATTRIBUTES_META";
  
    
  try
    var sqlStmt = snowflake.createStatement(sqlText :  v_sqlCode);
    var sqlRS = sqlStmt.execute();
  catch(err)
     errMessage =  "Failed: Code: " + err.code + "\n  State: " + err.state;
     errMessage += "\n  Message: " + err.message + v_sqlCode;
     errMessage += "\nStack Trace:\n" + err.stackTraceTxt + v_sqlCode;  
     throw 'Encountered error in executing v_sqlCode. \n' + errMessage;
   
   
  for (i = 1; i <= sqlStmt.getColumnCount(); i++) 
    arr_col_att.push(sqlStmt.getColumnName(i));


     arr_col_att[0] = arr_col_att[0].replace(/\s/g, ',');
    var v_col_att = arr_col_att.toString();
    v_string = arr_list[0].toString();
    v_string = v_string.replace(/\s/g, ',');
    
    v_sqlCode = `INSERT INTO ` + "ADMINDB" + "." + "TOOLKIT" + "." 
+ "SESSION_ATTRIBUTES_META" + `(` + v_col_att + `) VALUES( ` + v_string + `)`;
                
    try
      var sqlStmt = snowflake.createStatement(sqlText :  v_sqlCode);
      var sqlRS = sqlStmt.execute();
    catch(err)
      errMessage =  "Failed: Code: " + err.code + "\n  State: " + err.state;
      errMessage += "\n  Message: " + err.message + v_sqlCode;
      errMessage += "\nStack Trace:\n" + err.stackTraceTxt + v_sqlCode;  
      throw 'Encountered error in executing v_sqlCode. \n' + errMessage;
       
  return "SUCCESS!";
$$;

CALL "ADMINDB"."TOOLKIT".ADD_ATTRIBUTES_SESSION_META('DATABASE VARCHAR 1,SCHEMA VARCHAR 1,TIMESTAMP TIMESTAMP 0');

【问题讨论】：

检查 Snowflake Web UI 中的 Query History 选项卡，了解您的插入语句是如何从 SP 发送的。这将有助于找出问题所在。

在构造 SQL 语句时需要引号字符。您必须拆分 v_string 变量并在前两个元素周围添加单引号。

【参考方案1】：

修改第 33 行和第 36 行，修复问题：

CREATE OR REPLACE PROCEDURE "ADMINDB"."TOOLKIT".ADD_ATTRIBUTES_SESSION_META ("P_ATTRIBUTE_DATATYPE_FLAG" VARCHAR(16777216))
RETURNS VARCHAR
LANGUAGE JAVASCRIPT
COMMENT='Add Attributes to the table SESSION_ATTRIBUTES_META'
EXECUTE AS CALLER
AS
$$
  var v_list = P_ATTRIBUTE_DATATYPE_FLAG;
  var arr_list = [];
  arr_list = P_ATTRIBUTE_DATATYPE_FLAG.split(',');
  var v_string;
  var arr_col_att = [];
  var v_sqlCode = `SELECT * FROM ` + "ADMINDB" + "." + "TOOLKIT" + "." + "SESSION_ATTRIBUTES_META";
  
    
  try
    var sqlStmt = snowflake.createStatement(sqlText :  v_sqlCode);
    var sqlRS = sqlStmt.execute();
  catch(err)
     errMessage =  "Failed: Code: " + err.code + "\n  State: " + err.state;
     errMessage += "\n  Message: " + err.message + v_sqlCode;
     errMessage += "\nStack Trace:\n" + err.stackTraceTxt + v_sqlCode;  
     throw 'Encountered error in executing v_sqlCode. \n' + errMessage;
   
   
  for (i = 1; i <= sqlStmt.getColumnCount(); i++) 
    arr_col_att.push(sqlStmt.getColumnName(i));


     arr_col_att[0] = arr_col_att[0].replace(/\s/g, ',');
    var v_col_att = arr_col_att.toString();
    v_string = arr_list[0].toString();
    v_string = v_string.replace(/\s/g, "','" );
    
    v_sqlCode = `INSERT INTO ` + "ADMINDB" + "." + "TOOLKIT" + "." 
+ "SESSION_ATTRIBUTES_META" + `(` + v_col_att + `) VALUES( '` + v_string + `' )`;
                
    try
      var sqlStmt = snowflake.createStatement(sqlText :  v_sqlCode);
      var sqlRS = sqlStmt.execute();
    catch(err)
      errMessage =  "Failed: Code: " + err.code + "\n  State: " + err.state;
      errMessage += "\n  Message: " + err.message + v_sqlCode;
      errMessage += "\nStack Trace:\n" + err.stackTraceTxt + v_sqlCode;  
      throw 'Encountered error in executing v_sqlCode. \n' + errMessage;
       
  return "SUCCESS!";
$$;

CALL "ADMINDB"."TOOLKIT".ADD_ATTRIBUTES_SESSION_META('DATABASE VARCHAR 1,SCHEMA VARCHAR 1,TIMESTAMP TIMESTAMP 0');

据我所知，您目前还没有在参数中实现处理多条记录，但我认为您可以处理它。

【讨论】：

上帝保佑你，我一直在这个语法错误上很久了。我绝对可以做多条记录