Redshift中基于table_1列是否存在创建temp_table的实现方法

在Redshift里要实现这种“列存在则直接选取，不存在则新增列并赋值”的临时表创建需求，确实没法用静态SQL直接搞定——毕竟Redshift不支持在CREATE TABLE语句里直接做条件分支逻辑。不过我们可以借助动态SQL结合系统表查询来解决这个问题，具体方案如下：

1. 先通过系统表判断目标列是否存在 #

Redshift提供了information_schema.columns系统表，它记录了所有表的列信息，我们可以用它来检查table_1里是否存在column_n：

SELECT EXISTS(
    SELECT 1 
    FROM information_schema.columns 
    WHERE table_name = 'table_1' 
      AND column_name = 'column_n'
      AND table_schema = 'public' -- 如果你的表不在默认public schema下，替换成实际schema名
);

2. 用PL/pgSQL编写动态SQL块实现需求 #

我们可以写一个PL/pgSQL代码块，先判断列的存在性，再动态拼接创建临时表的SQL语句，最后执行生成的SQL：

DO $$
DECLARE
    column_exists BOOLEAN;
    create_temp_sql TEXT;
BEGIN
    -- 检查column_n是否存在于table_1中
    SELECT EXISTS(
        SELECT 1 
        FROM information_schema.columns 
        WHERE table_name = 'table_1' 
          AND column_name = 'column_n'
          AND table_schema = 'public'
    ) INTO column_exists;

    -- 根据列的存在性拼接不同的SQL语句
    IF column_exists THEN
        create_temp_sql := 'CREATE TEMP TABLE temp_table AS 
                           SELECT column_name_1, column_name_2, column_name_3, column_n 
                           FROM table_1';
    ELSE
        create_temp_sql := 'CREATE TEMP TABLE temp_table AS 
                           SELECT column_name_1, column_name_2, column_name_3, ''val_n'' AS column_n 
                           FROM table_1';
    END IF;

    -- 执行动态生成的SQL
    EXECUTE create_temp_sql;
END $$;

关键细节说明 #

• 动态SQL拼接逻辑：如果column_n存在，就直接在SELECT里包含该列；不存在的话，就用常量'val_n'生成一个名为column_n的新列，保证临时表结构符合要求。
• EXECUTE语句：这是实现动态逻辑的核心，用来执行我们拼接好的SQL字符串。
• Schema适配：如果你的table_1不在默认的public schema下，一定要修改table_schema的值为实际的schema名称，否则会查不到列信息。

验证结果 #

执行完上面的代码后，你可以用以下语句验证临时表的结构和数据：

-- 查看临时表的列结构
SELECT column_name, data_type FROM information_schema.columns WHERE table_name = 'temp_table';

-- 查看临时表的样本数据
SELECT * FROM temp_table LIMIT 10;

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