如何在Databricks SQL中编写for或while循环？

方案1：原生WHILE循环（适用于Databricks SQL 2022.35及以上版本） #

Databricks SQL高版本已经原生支持变量声明和WHILE流程控制，你提供的SQL Server示例可以直接改写为对应语法：

-- 声明计数器变量
DECLARE Counter INT DEFAULT 1;
WHILE (Counter <= 10)
DO
  -- 输出计数器值，字符串拼接用CONCAT避免类型错误
  PRINT CONCAT('The counter value is = ', Counter);
  -- 计数器自增
  SET Counter = Counter + 1;
END WHILE;

如果需要把循环生成的结果供给后续SQL使用，可以在循环中往临时表插入数据，后续SQL直接读临时表即可。

方案2：递归CTE实现循环（全版本兼容，纯SQL实现） #

如果你的Databricks版本较低不支持原生WHILE语法，可以用递归CTE实现同等逻辑，生成的结果可以直接作为数据集供后续SQL调用，无需依赖其他语言：

WITH RECURSIVE counter_seq(counter) AS (
  -- 初始值：对应你示例里的SET @Counter=1
  SELECT 1
  UNION ALL
  -- 递归逻辑：对应计数器自增，终止条件为<=10
  SELECT counter + 1 FROM counter_seq WHERE counter < 10
)
-- 输出结果，也可以用CREATE TEMP VIEW temp_counter AS 把结果存为临时视图
SELECT CONCAT('The counter value is = ', counter) AS counter_info FROM counter_seq;

注意事项 #

• 递归CTE默认最大递归深度为1000，如果你需要更大的循环范围，可以提前执行配置调整：SET spark.sql.recursiveCTE.maxDepth = 你需要的最大数值;
• 非必要场景优先使用集合操作代替循环逻辑，SQL的集合运算性能远高于逐行/逐次循环。

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