如何在Athena中实现仅当列存在时执行对应查询的兼容逻辑？

你原来的写法失败是因为Athena基于Presto，SQL会在解析阶段就校验所有引用的列是否存在——不管CASE分支的逻辑会不会被执行。只要查询里写了new_column，解析器就会直接检查表结构，发现列不存在就报错，根本不会走到执行CASE逻辑的阶段。

核心思路是绕开解析阶段的列校验：把整行数据转成JSON格式，通过JSON提取的方式获取新列的值（列不存在时返回NULL），再结合列存在性判断来执行对应逻辑。这样解析阶段只会校验CAST(tb AS JSON)这个合法操作，不会检查new_column是否存在。

完整SQL示例：

WITH has_column AS (
    -- 检查新列是否存在，建议加上table_schema避免跨schema干扰
    SELECT COUNT(*) > 0 AS has_new_col
    FROM information_schema.columns
    WHERE table_name = 'my_table'
      AND column_name = 'new_column'
      AND table_schema = '你的schema名称' -- 替换成实际的schema，比如'default'
),
table_data AS (
    SELECT 
        id,
        current_column,
        -- 将整行转成JSON，提取new_column的值，不存在则返回NULL
        JSON_EXTRACT_SCALAR(CAST(tb AS JSON), '$.new_column') AS new_column
    FROM "my_table" tb
)
SELECT 
    COUNT(td.id) AS s,
    check_col.has_new_col
FROM table_data td
CROSS JOIN has_column check_col
WHERE 
    CASE
        WHEN check_col.has_new_col THEN td.new_column = 'value'
        ELSE td.current_column = 'value'
    END

• CAST(tb AS JSON)会把表的每一行转换成JSON对象，列名对应JSON的key，适合绝大多数Athena支持的表格式（Parquet、ORC、CSV等）。
• 如果新列是数值/日期等非字符串类型，可以用JSON_EXTRACT配合类型转换，比如TRY_CAST(JSON_EXTRACT(CAST(tb AS JSON), '$.new_column') AS INT)。
• 必须指定table_schema，否则如果多个schema下有同名表，会导致列存在性判断出错。

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