Databricks SQL实现通用工作日累加函数报错求助

错误原因 #

你遇到的UNSUPPORTED_SUBQUERY_EXPRESSION_CATEGORY.CORRELATED_REFERENCE错误，本质是Databricks SQL的UDF执行上下文限制：不允许在CTE的SELECT列表、窗口函数等非WHERE/HAVING子句中直接引用外层函数的参数（比如你的dt_initial），这种跨层级的关联引用在UDF中是被禁止的。

修正方案：递归CTE实现 #

改用递归CTE逐天迭代，避免对外部参数的关联引用，同时逻辑更高效（无需生成冗余日期）：

CREATE OR REPLACE FUNCTION add_weekdays(dt_initial DATE, num_days INT)
RETURNS DATE
RETURN (
  WITH RECURSIVE business_days AS (
    -- 初始节点：传入初始日期和剩余需要累加的工作日数
    SELECT dt_initial AS current_date, num_days AS remaining_days
    UNION ALL
    -- 递归迭代：每天加1天，判断是否为工作日，更新剩余天数
    SELECT 
      DATE_ADD(current_date, 1) AS current_date,
      CASE 
        WHEN WEEKDAY(DATE_ADD(current_date, 1)) BETWEEN 0 AND 4 THEN remaining_days - 1 
        ELSE remaining_days 
      END AS remaining_days
    FROM business_days
    WHERE remaining_days > 0
  )
  -- 取剩余天数为0的最后一个日期（即目标工作日）
  SELECT current_date 
  FROM business_days 
  WHERE remaining_days = 0 
  ORDER BY current_date DESC 
  LIMIT 1
);

方案优势 #

1. 规避了关联引用限制：递归CTE内部的所有引用都是CTE自身的列，不再直接引用外层函数参数
2. 高效无冗余：无需生成num_days*2个冗余日期，仅迭代到目标工作日为止
3. 兼容边界场景：支持num_days=0（直接返回初始日期）、初始日期为周末等情况

测试示例 #

-- 周五加1个工作日，返回下周一
SELECT add_weekdays('2024-05-24', 1); -- 输出：2024-05-27

-- 周一加3个工作日，返回周四
SELECT add_weekdays('2024-05-27', 3); -- 输出：2024-05-30

内容的提问来源于stack exchange，提问作者Leandro Guimarães