Snowflake自定义函数ADDWORKDAYS报错[603]：字段传参失败但常量正常

问题根源 #

你的自定义UDF传入常量日期时正常，但批量处理表字段时触发内部错误，核心原因如下：

• ROW_NUMBER() OVER(ORDER BY 1)排序逻辑不稳定：GENERATOR生成的行无固定顺序，ORDER BY 1无法保证RN的连续性和正确性，批量执行时会导致执行引擎上下文混乱。
• 嵌套子查询隔离性缺失：原函数的嵌套结构未正确隔离每个输入的START_DATE，批量处理多行数据时，子查询结果会出现交叉干扰，触发Snowflake内部执行错误。
• 额外缺陷：GENERATOR固定生成5行，若目标工作日超出5天范围（比如连续周末加节假日），会返回空值，不符合需求。

修正后的函数代码 #

改用递归CTE生成目标工作日，既保证逻辑正确性，又能处理任意天数的需求，同时避免内部执行错误：

CREATE OR REPLACE FUNCTION xxx.ADDWORKDAYS(START_DATE DATE, DAYS NUMBER(5,0))
RETURNS DATE
LANGUAGE SQL
AS
$$
WITH RECURSIVE workdays AS (
    -- 初始行：从起始日期的下一天开始检查
    SELECT DATEADD(DAY, 1, START_DATE) AS dt, 0 AS cnt
    UNION ALL
    -- 递归生成后续日期，直到找到足够的工作日
    SELECT DATEADD(DAY, 1, w.dt), 
           CASE WHEN DAYNAME(w.dt) NOT IN ('Sat','Sun') AND NOT EXISTS(SELECT 1 FROM xxx.HOLIDAYS h WHERE h.FULL_DATE = w.dt) THEN w.cnt + 1 ELSE w.cnt END
    FROM workdays w
    WHERE w.cnt < DAYS
)
-- 取第N个工作日
SELECT dt
FROM workdays
WHERE cnt = DAYS
$$;

修正逻辑说明 #

1. 递归CTE结构：从起始日期的下一天开始，逐天检查是否为工作日（非周末+非节假日）。
2. 计数逻辑：每找到一个工作日，计数器cnt加1，直到计数器等于输入的DAYS值。
3. 稳定性保证：递归逻辑逐天递进，避免了原函数中窗口函数排序不稳定的问题，同时每个输入的START_DATE都会独立生成递归序列，不会出现批量处理时的交叉干扰。

验证测试 #

执行原查询语句即可得到预期结果：

SELECT
    D.FULL_DATE
  , xxx.ADDWORKDAYS(D.FULL_DATE, 1) AS WORKDAYS_1    
FROM xxx.DATE_TABLE D

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