Databricks SQL中DECLARE变量的等效语法咨询及报错解决

Databricks SQL（基于Spark SQL）不支持SQL Server风格的DECLARE变量声明和SET赋值语法，这就是你遇到mismatched input 'DECLARE'错误的原因。下面是几种实用的替代方案：

• 用CTE定义静态变量
  把变量值塞进公共表表达式（CTE）里，后续查询直接引用，适合简单的单值变量场景。比如你原来的SQL Server代码如果是：

  DECLARE @TableName sysname = 'sales_data';
  DECLARE @StartDate varchar(10) = '2023-01-01';
  DECLARE @EndDate varchar(10) = '2023-12-31';
  
  SELECT * FROM @TableName WHERE sale_date BETWEEN @StartDate AND @EndDate;
  

  改成Databricks SQL的写法：

  WITH vars AS (
      SELECT 
          'sales_data' AS table_name,
          '2023-01-01' AS start_date,
          '2023-12-31' AS end_date
  )
  SELECT s.* 
  FROM vars v
  JOIN sales_data s ON s.sale_date BETWEEN v.start_date AND v.end_date;
  

  注意：如果要动态引用表名，这种方式行不通，得用下面的动态SQL方案。

• 动态SQL处理动态对象
  如果需要用动态表名、列名这类场景，用EXECUTE IMMEDIATE拼接字符串就行，搭配Databricks SQL的会话变量：

  -- 设置会话变量
  SET table_name = 'sales_data';
  SET start_date = '2023-01-01';
  SET end_date = '2023-12-31';
  
  -- 执行动态SQL
  EXECUTE IMMEDIATE CONCAT(
      'SELECT * FROM ', table_name, 
      ' WHERE sale_date BETWEEN ''', start_date, ''' AND ''', end_date, ''''
  );
  

  这里的SET是Databricks的会话变量设置方式，变量在当前会话内有效，直接写变量名就能引用。

• Notebook里用Widgets做交互式变量
  如果是在Databricks Notebook的%sql单元格里跑查询，还可以用Widgets做可交互的变量，调整值很方便：

  -- 创建交互式控件
  CREATE WIDGET TEXT table_name DEFAULT 'sales_data';
  CREATE WIDGET TEXT start_date DEFAULT '2023-01-01';
  CREATE WIDGET TEXT end_date DEFAULT '2023-12-31';
  
  -- 使用变量查询
  SELECT * FROM ${table_name} WHERE sale_date BETWEEN ${start_date} AND ${end_date};
  

  运行后Notebook顶部会出现输入框，直接修改就能刷新查询结果。

• 临时视图存复杂变量集合
  如果变量值是通过查询得到的，或者有一堆变量要复用，可以建临时视图来存储：

  CREATE OR REPLACE TEMP VIEW vars AS
  SELECT 
      (SELECT MAX(sale_date) FROM sales_data) AS max_date,
      (SELECT MIN(sale_date) FROM sales_data) AS min_date;
  
  SELECT * FROM sales_data WHERE sale_date BETWEEN (SELECT min_date FROM vars) AND (SELECT max_date FROM vars);
  

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