在Presto/Trino中如何循环执行带不同参数的SQL并合并输出结果

该需求完全可以在Presto/Trino中实现，无需逐次运行SQL再手动UNION合并结果，通过参数定义+数组展开+关联原有查询逻辑的方式，单次查询即可得到所有日期维度的汇总结果。

实现思路 #

• 先将你需要传入的固定参数（周数）、待遍历的日期列表统一在参数CTE中定义
• 将日期数组展开为单行列，每一行对应你要迭代的一个日期
• 将你的原有链式CTE和日期行做交叉关联，原有逻辑中引用的date参数替换为展开后的日期字段，number_weeks直接引用参数CTE中的值即可
• 该方案性能优于逐次执行SQL再UNION，Trino会自动优化执行计划，避免重复扫描公共数据源

完整示例代码 #

-- 1. 定义公共参数
WITH params AS (
    SELECT 
        4 AS number_weeks, -- 周数可直接在此修改
        ARRAY[DATE '2024-01-01', DATE '2024-01-08', DATE '2024-01-15'] AS target_dates -- 待遍历的日期列表
),
-- 2. 展开日期数组为单行单日期
date_list AS (
    SELECT 
        d AS run_date,
        number_weeks
    FROM params, UNNEST(target_dates) AS t(d)
),
-- 3. 以下替换为你自己的链式CTE逻辑，所有原来用到date的地方替换为run_date
-- 示例第一个CTE
user_behavior AS (
    SELECT 
        dl.run_date,
        t.user_id,
        t.pay_amount
    FROM date_list dl
    LEFT JOIN your_source_table t
        ON t.event_date >= date_add('week', -dl.number_weeks, dl.run_date)
        AND t.event_date <= dl.run_date
),
-- 示例第二个CTE
date_agg AS (
    SELECT
        run_date,
        COUNT(DISTINCT user_id) AS uv,
        SUM(pay_amount) AS total_gmv
    FROM user_behavior
    GROUP BY run_date
)
-- 4. 最终输出自动是所有日期结果的合并汇总
SELECT * FROM date_agg;

扩展说明 #

如果不同的遍历日期需要对应不同的周数，可以直接将参数写在展开的VALUES里，不需要单独定义params：

WITH date_list AS (
    SELECT * FROM (
        VALUES
        (DATE '2024-01-01', 4),
        (DATE '2024-01-08', 2),
        (DATE '2024-01-15', 6)
    ) AS t(run_date, number_weeks)
)
-- 后续CTE逻辑和上面示例一致

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