DBT中使用宏定义变量后复用出现未定义错误的咨询

问题现象 #

通过宏number_of_rows_to_process为变量number_of_rows赋值后，单独输出该变量正常；但用它计算steps变量时，触发number_of_rows未定义错误。改用固定值赋值则无此问题。

问题根源 #

排查发现宏内部也定义了同名的number_of_rows变量，且宏的执行逻辑存在缺陷：要么未正确返回查询到的行数，要么局部变量作用域冲突导致外部变量解析异常。

解决方案 #

1. 修正宏的返回逻辑

确保宏正确返回查询结果，而非仅定义内部变量。示例宏写法：

{% macro number_of_rows_to_process(model_ref) %}
    {% set count_query %}
        SELECT COUNT(*) AS row_count FROM {{ model_ref }}
    {% endset %}
    {% set query_results = run_query(count_query) %}
    {% if execute %}
        {{ query_results.columns[0].values()[0] }}
    {% else %}
        0
    {% endif %}
{% endmacro %}

核心：通过run_query执行计数查询后，直接返回结果值，避免内部变量干扰外部作用域。

2. 避免变量名冲突

将宏内部的查询列名或变量名改为与外部不同的名称（如示例中的row_count），彻底规避作用域冲突风险。

3. 强制类型转换

宏返回的结果可能是字符串类型，赋值时强制转为整数，确保后续计算无类型解析问题：

{% set number_of_rows = number_of_rows_to_process(ref("some_model")) | int %}

验证代码 #

修改后完整运行代码：

{% set step_size = 6000 %}
{% set number_of_rows = number_of_rows_to_process(ref("some_model")) | int %}
{{ number_of_rows }}
{% set steps = ((number_of_rows + (step_size - 1)) / step_size)|round(0, 'ceil')|int %}
{{ steps }}

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