Snowflake SQL存储过程调用触发Key Error问题求助

你遇到的Key Error大概率是规则对象的键名与目标表实际列名不匹配，或是存储过程中遍历规则对象的逻辑存在问题，结合你的ROOT_DEPTH表场景，按以下步骤排查修复：

• 核对规则对象键名与表列名
  确认传入的规则对象中，列名键值和表中实际列名完全一致：

  • 如果表列是用双引号定义的区分大小写名称，规则对象的键名必须严格匹配大小写；
  • 如果是无引号定义的列名，Snowflake会自动转为大写，规则对象的键名也需要对应大写（比如你的ROOT_DEPTH_CODE列，规则对象里不能写root_depth_code）。

• 检查存储过程的规则遍历逻辑
  若存储过程用FLATTEN解析规则对象，要确保获取键名的方式正确。比如用SELECT KEY FROM TABLE(FLATTEN(input => :rule_obj))时，返回的KEY值必须和列名完全一致，避免额外的字符串处理导致键名变形。

• 验证参数传递格式
  调用存储过程时，规则对象的JSON格式必须合法，键名不能有多余空格或特殊字符。正确的调用示例：

  CALL YOUR_PROCEDURE_NAME('ROOT_DEPTH', PARSE_JSON('{"ROOT_DEPTH_CODE": 1}'));
  

• 添加调试日志定位问题
  在存储过程中加入日志输出，查看实际处理的键名：

  FOR rec IN (SELECT KEY AS col_name, VALUE AS rule FROM TABLE(FLATTEN(input => :rule_obj))) DO
      CALL SYSTEM$LOG('当前处理列名: ' || rec.col_name);
      -- 后续验证逻辑
  END FOR;
  

  通过日志可以直接对比存储过程拿到的键名和表列名，快速定位不匹配的问题。

修复后的示例存储过程 #

CREATE OR REPLACE PROCEDURE VALIDATE_TABLE_COLUMNS(table_name VARCHAR, rule_obj VARIANT)
RETURNS VARCHAR
LANGUAGE SQL
AS
$$
DECLARE
    result VARCHAR DEFAULT '';
    total_rows NUMBER;
    null_count NUMBER;
    null_ratio NUMBER;
BEGIN
    -- 获取表总行数
    EXECUTE IMMEDIATE 'SELECT COUNT(*) FROM ' || table_name INTO total_rows;
    
    FOR rec IN (SELECT KEY AS col_name, VALUE AS rule FROM TABLE(FLATTEN(input => :rule_obj))) DO
        IF rec.rule = 1 THEN
            -- 统计空值数量
            EXECUTE IMMEDIATE 'SELECT COUNT(*) FROM ' || table_name || ' WHERE ' || rec.col_name || ' IS NULL' INTO null_count;
            null_ratio := (null_count / total_rows) * 100;
            
            IF null_ratio > 10 THEN
                result := result || '列' || rec.col_name || '空值占比' || ROUND(null_ratio, 2) || '%，超过10%阈值；';
            ELSE
                result := result || '列' || rec.col_name || '空值占比' || ROUND(null_ratio, 2) || '%，符合要求；';
            END IF;
        END IF;
    END FOR;
    
    RETURN result;
END;
$$;

-- 调用示例
CALL VALIDATE_TABLE_COLUMNS('ROOT_DEPTH', PARSE_JSON('{"ROOT_DEPTH_CODE": 1}'));

内容的提问来源于stack exchange，提问作者Pandit Abilash