如何用Jinja优化DBT中的多Case When语句？

针对你要优化的多Case When场景，用Jinja的字典映射+循环是最高效的方式，既减少重复代码，后期维护也方便。直接上解决方案：

第一步：定义映射关系 #

在你的DBT模型文件开头，用Jinja定义一个字典，把what_id的前缀和对应的目标字段一一对应起来。你可以根据实际需求随时添加或修改键值对：

{% set what_id_to_field = {
    '006': 'opportunity_id',
    '001': 'account_id',
    '003': 'contact_id',
    '00Q': 'lead_id'  # 示例：可自行添加其他what_id前缀对应的字段
} %}

第二步：用循环生成Case When逻辑 #

在SQL查询中，通过Jinja的for循环遍历上面的字典，自动生成所有Case When分支，代替原来手写的重复代码：

SELECT
    activity_id,
    what_id,
    {% for prefix, field in what_id_to_field.items() %}
        CASE WHEN what_id LIKE '{{ prefix }}%' THEN {{ field }} END AS {{ field }}_from_what_id{% if not loop.last %},{% endif %}
    {% endfor %}
FROM {{ ref('dim_activities') }}

代码细节说明 #

• {% for prefix, field in what_id_to_field.items() %}：遍历字典的每个键值对，prefix对应what_id的前缀（比如'006'），field对应要返回的目标字段（比如'oppotunity_id'）
• CASE WHEN what_id LIKE '{{ prefix }}%' THEN {{ field }} END：自动生成单个Case When分支，匹配以指定前缀开头的what_id，返回对应字段值
• AS {{ field }}_from_what_id：给生成的字段起别名，避免和原表字段重名
• {% if not loop.last %},{% endif %}：自动处理字段间的逗号，防止最后一个字段后多逗号导致SQL语法错误

对比原手写写法 #

原来你可能需要写一堆重复的Case When：

SELECT
    activity_id,
    what_id,
    CASE WHEN what_id LIKE '006%' THEN opportunity_id END AS opportunity_id_from_what_id,
    CASE WHEN what_id LIKE '001%' THEN account_id END AS account_id_from_what_id,
    CASE WHEN what_id LIKE '003%' THEN contact_id END AS contact_id_from_what_id
FROM dim_activities

用Jinja优化后，你只需要维护what_id_to_field这个字典，新增或修改规则时不用动SQL主体，大幅降低维护成本。

内容的提问来源于stack exchange，提问作者Lorena Gonzales Saar