如何在Data Factory匿名块脚本任务中设置变量并自定义返回字段

一、修改Snowflake匿名块输出的自定义键名 #

你当前用return row_count的方式，Snowflake会默认把结果键名设为anonymous block。要改成自定义名称（比如row_count），不需要用return，而是在匿名块末尾通过SELECT语句返回带指定别名的列：

DECLARE 
    row_count int := -1;
BEGIN
    SET row_count := (SELECT COUNT(*) FROM ourTable where date = '2024-11-12');
    -- 用SELECT返回带自定义别名的结果
    SELECT row_count AS row_count;
END;

修改后，脚本活动返回的结果集中，列名就是你定义的row_count，后续提取路径会更清晰。

二、更高效的数据提取方式 #

1. 优化脚本活动的表达式提取 #

如果继续使用脚本活动，修改后的结果可以用更简洁的表达式提取：

@int(activity('Check if DATE exists').output.resultSets[0].rows[0].row_count)

（注意去掉原表达式中键名的单引号，ADF中引用无特殊字符的键名不需要引号）

2. 使用Lookup活动替代脚本活动（更推荐） #

ADF的Lookup活动专门用于获取单行或少量查询结果，比脚本活动更适配这类场景：

• 配置Lookup活动，直接执行简化的查询语句（无需匿名块）：

  SELECT COUNT(*) AS row_count FROM ourTable where date = '2024-11-12'
  

• 提取结果时直接通过firstRow获取单行数据，无需遍历数组：

  @int(activity('Lookup Row Count').output.firstRow.row_count)
  

这种方式代码更简洁，也省去了遍历resultSets和rows数组的步骤，性能更优。

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