如何编写通用SQL查询：满足条件返回多列，否则仅返回id列？

原查询的核心问题有两个：

• 无论country是否为'Germany'，都会固定返回两列，非德国数据会产生空列
• 若数据库不存在population列，查询会直接在解析阶段报错

要同时满足「动态返回列数」和「跨库兼容」的需求，只能通过动态SQL实现——因为静态SQL的结果集列数是固定的，且解析阶段会强制校验所有引用的列是否存在。

以下是针对主流数据库的实现方案：

MySQL/MariaDB 实现 #

-- 检查当前表是否存在population列
SET @has_population = EXISTS(
    SELECT 1 
    FROM INFORMATION_SCHEMA.COLUMNS 
    WHERE TABLE_SCHEMA = DATABASE()
      AND TABLE_NAME = 'your_table'
      AND COLUMN_NAME = 'population'
);

-- 动态生成并执行查询
PREPARE sql_stmt FROM 
    CASE WHEN @has_population THEN
        -- 有population列时，分开查询德国和非德国数据
        'SELECT id, population FROM your_table WHERE country = ''Germany''; SELECT id FROM your_table WHERE country != ''Germany'''
    ELSE
        -- 无population列时，统一返回id
        'SELECT id FROM your_table'
    END;

EXECUTE sql_stmt;
DEALLOCATE PREPARE sql_stmt;

PostgreSQL 实现 #

DO $$
DECLARE
    has_population BOOLEAN;
BEGIN
    -- 检查列是否存在
    SELECT EXISTS(
        SELECT 1 
        FROM information_schema.columns 
        WHERE table_schema = current_schema()
          AND table_name = 'your_table'
          AND column_name = 'population'
    ) INTO has_population;

    -- 执行对应查询逻辑
    IF has_population THEN
        EXECUTE 'SELECT id, population FROM your_table WHERE country = ''Germany''';
        EXECUTE 'SELECT id FROM your_table WHERE country != ''Germany''';
    ELSE
        EXECUTE 'SELECT id FROM your_table';
    END IF;
END $$;

关键说明 #

• 动态SQL会先校验目标表是否存在population列，再决定执行对应的查询逻辑
• 对于存在population列的数据库，会分别返回德国数据（含id和population）和非德国数据（仅id）；对于无该列的数据库，直接返回所有id
• 这种方式规避了静态SQL的列存在性校验问题，同时满足了无空列的要求

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