Snowflake存储过程中如何按查询结果条件执行Task？

第二种写法的核心错误 #

SQL存储过程的CASE表达式是仅用于返回值的逻辑判断结构，无法在其中嵌入EXECUTE TASK这类执行型语句（这类语句属于DDL/DML范畴，不属于返回值表达式），因此第二种写法从语法层面就不成立，会直接触发编译报错。

第一种写法的潜在问题及修复 #

第一种写法的逻辑框架是正确的，但可能因以下细节问题导致异常：

1. 转义字符错误：代码中的>是HTML转义后的大于号，在Snowflake中执行时需替换为原生的>，否则会触发语法错误。
2. ACCOUNT_USAGE视图延迟：Snowflake的ACCOUNT_USAGE类视图通常存在1-2小时的数据同步延迟，若查询的是当天刚生成的数据，可能还未同步到视图中，导致v_count始终为0，任务无法触发。可尝试改用INFORMATION_SCHEMA下的对应视图（若业务场景允许），或放宽日期条件进行测试。
3. 权限不足：
   • 调用存储过程的用户需拥有EXECUTE TASK权限（针对TASK_DEMO.TASK_1）。
   • 调用者需拥有SELECT权限访问SNOWFLAKE.ACCOUNT_USAGE.GRANTS_TO_ROLES视图。
4. 语法细节验证：SQL存储过程中SELECT ... INTO语句要求查询返回单行单列，你的COUNT(*)满足该条件，但后续修改查询逻辑时需注意保持此规则。

修正后的可运行代码 #

CREATE OR REPLACE PROCEDURE CHECK_AND_EXECUTE_TASK()
  RETURNS STRING
  LANGUAGE SQL
  EXECUTE AS CALLER
AS
$$
DECLARE
  v_count INT;
BEGIN
  -- 查询符合条件的记录数
  SELECT COUNT(*) INTO v_count
  FROM SNOWFLAKE.ACCOUNT_USAGE.GRANTS_TO_ROLES 
  WHERE 
      GRANTED_BY = 'AAD_PROVISIONER'
      AND DATE(CREATED_ON) = CURRENT_DATE;
 
  -- 根据计数判断是否执行任务
  IF v_count > 0 THEN
    EXECUTE TASK TASK_DEMO.TASK_1;
    RETURN 'Task executed successfully.';
  ELSE
    RETURN 'No results found. Task not executed.';
  END IF;
END;
$$;

额外验证步骤 #

1. 单独执行查询语句，确认是否能返回预期计数：

SELECT COUNT(*)
FROM SNOWFLAKE.ACCOUNT_USAGE.GRANTS_TO_ROLES 
WHERE 
    GRANTED_BY = 'AAD_PROVISIONER'
    AND DATE(CREATED_ON) = CURRENT_DATE;

2. 验证调用者权限：

-- 检查任务执行权限
SHOW GRANTS ON TASK TASK_DEMO.TASK_1;
-- 检查ACCOUNT_USAGE视图访问权限
SHOW GRANTS ON VIEW SNOWFLAKE.ACCOUNT_USAGE.GRANTS_TO_ROLES;

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