Azure Databricks SQL 中能否使用For、Loop、While等循环迭代结构？

首先明确结论：Azure Databricks环境下可以实现循环迭代类的逻辑开发，但不同运行场景的支持方式有差异：

• 纯标准Spark SQL语法（无过程式扩展）本身没有内置FOR、WHILE这类过程式循环关键字，如果你是在Databricks SQL仓库的纯SQL查询场景下运行标准SQL，不能直接写原生的循环语句。
• 如果你使用Databricks Notebook的SQL单元格，或者开启了Databricks SQL的脚本功能，就可以通过过程式扩展语法实现迭代：
  • 原生支持WHILE循环构造，示例代码如下：

DECLARE counter INT = 1;
WHILE counter <= 5 DO
  -- 此处编写每次迭代需要执行的SQL逻辑
  INSERT INTO access_log (record_date) VALUES (DATE_ADD(CURRENT_DATE(), counter));
  SET counter = counter + 1;
END WHILE;

• 目前还没有原生的FOR循环语法，要实现固定次数的迭代逻辑，有两种常见实现方案：
  1. 用上述WHILE循环手动维护计数器模拟FOR逻辑
  2. 用纯SQL的sequence函数结合EXPLODE实现，不需要用到过程式语法，示例如下：

-- 模拟遍历1到10的FOR循环，处理每个序号对应的业务逻辑
SELECT 
  seq_id,
  -- 此处编写对应序号的计算逻辑
  CONCAT('order_', seq_id) as order_no
FROM TABLE(EXPLODE(SEQUENCE(1, 10))) AS t(seq_id);

• 要是你的迭代逻辑复杂度很高，更建议直接在Notebook中切换到Python/Scala单元格，用编程语言的循环语法批量提交SQL任务，灵活度会高很多。

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