在Databricks SQL笔记本中设置宏以检查表名中的年份限定符

我来给你详细梳理下怎么实现这个需求，在Databricks SQL笔记本里有几种灵活的方案，完全能满足你通过开头设置的年份限定符来控制后续单元格执行的要求。

第一步：定义全局变量 #

首先在笔记本的第一个单元格里，我们可以用DECLARE命令定义全局的年份限定符，甚至可以把需要检查的文件名统一放在数组里，方便后续复用修改：

-- 第一个单元格：设置全局配置变量
DECLARE Year_qualifier STRING DEFAULT 'y2022';
-- 把需要处理的文件名放在数组里，后续批量操作更方便
DECLARE target_file_names ARRAY<STRING> DEFAULT ARRAY('file1', 'file2', 'file3');

这些变量的作用域是整个笔记本，后续所有单元格都能直接引用它们。

方案一：直接动态生成表名（最简单） #

如果你只是想让后续查询自动使用带指定年份限定符的表，不需要额外检查，直接用变量拼接表名就行。只要你修改第一个单元格的Year_qualifier，所有后续查询都会自动切换到对应的年份表：

-- 后续单元格：自动使用指定年份的表
SELECT * FROM file1_${Year_qualifier};

比如你把Year_qualifier改成y2023，这条查询就会自动变成SELECT * FROM file1_y2023，完全不用手动修改查询语句。

方案二：带表存在检查的条件执行（更稳妥） #

如果想避免因为目标表不存在而报错，可以先检查表是否存在，再决定是否执行查询。这里需要用EXECUTE IMMEDIATE来动态执行拼接好的SQL语句：

-- 后续单元格：检查表存在后再执行查询
DECLARE current_table STRING DEFAULT CONCAT('file1_', Year_qualifier);

IF EXISTS (
    SELECT 1 FROM information_schema.tables 
    WHERE table_name = current_table
    AND table_schema = current_database() -- 可选：指定当前数据库，避免跨库匹配
) THEN
    -- 表存在，执行查询
    EXECUTE IMMEDIATE CONCAT('SELECT * FROM ', current_table);
ELSE
    -- 表不存在，返回提示信息
    SELECT CONCAT('Skipping query: 目标表 ', current_table, ' 在当前数据库 ', current_database(), ' 中不存在') AS 状态提示;
END IF;

方案三：批量处理多个文件（高效复用） #

如果要处理多个文件名，可以循环遍历我们在第一个单元格定义的target_file_names数组，批量处理每个文件对应的年份表：

-- 后续单元格：批量处理所有目标文件
FOR file_name IN target_file_names DO
    DECLARE current_table STRING DEFAULT CONCAT(file_name, '_', Year_qualifier);
    IF EXISTS (
        SELECT 1 FROM information_schema.tables 
        WHERE table_name = current_table
        AND table_schema = current_database()
    ) THEN
        EXECUTE IMMEDIATE CONCAT('SELECT * FROM ', current_table);
    ELSE
        SELECT CONCAT('跳过文件 ', file_name, ' 的查询：表 ', current_table, ' 不存在') AS 状态提示;
    END IF;
END FOR;

这样只要你在第一个单元格的数组里添加或删除文件名，就能自动批量处理对应的表，非常高效。

一些注意事项 #

• 确保你有查询information_schema.tables的权限，这样才能正常检查表是否存在
• 如果你的表不在当前默认数据库里，记得把table_schema = current_database()改成对应的数据库名称，比如table_schema = 'your_target_db'
• EXECUTE IMMEDIATE在这里是安全的，因为变量都是你自己控制的，不用担心SQL注入风险

备注：内容来源于stack exchange，提问作者Dr.Data