在其他宏中声明其他宏中的var

Question

主要想法是具有恒定变量，然后有可能在代码或宏的其他部分中使用它作为DBT。

包含常数的宏的示例：

{% macro constant_vars() -%}
    
{%
set var_1 = {
 "0": ["0"],
 "1": ["1", "11", "111"]
}
%}

{%
set var_2 = {
 "2": ["2"],
 "3": ["3"]
}
%}
{%- endmacro -%}

使用以前的宏：

{% macro evaluate(
    column_to_check
) -%}

CASE
    {% for map_key in var_1 -%}
    WHEN ({{column_to_check}} IN UNNEST( {{ var_1[map_key] }})) THEN
        '{{ map_key }}'
    {% endfor -%}
    ELSE
        "-1"
END
{%- endmacro -%}

为dbt创建的sql句子中的常数：

SELECT 
[..]
  evaluate(column1)
[..]
FROM
 table

dbt编译的查询：

SELECT 
[..]
CASE
    WHEN (column1 IN UNNEST(["0"])) THEN
        '0'
    WHEN (column1 IN UNNEST(["1", "11", "111"])) THEN
        '1'
    
    ELSE
        "-1"
END
[..]
FROM
 table

是否有可能？存在另一种做到这一点的方式？

谢谢！

Answer 1

您使用{％设置my_var = ...％}在范围中声明的变量，这意味着在您设置的位置之外不会在任何情况下声明它们。

DBT还支持var宏，可以充当全局变量。

”您的dbt_project.yml这样的文件：

vars:
  var_1:
    0: ["0"]
    1: ["1", "11", "111"]

然后使用{{var（'var_1'）}}}访问值。

另一个选择是返回来自宏的变量。如果值不是静态/文字，则可以很好地工作：

-- in macros/get_var_1.sql
{% macro get_var_1() %}
{% set var_1 = {
    "0": ["0"],
    "1": ["1", "11", "111"]
} %}
{{ return(var_1) }}
{% endmacro %}

然后您可以调用宏并将其分配给模型中的变量：

-- in models/my_model.sql
{% set my_var = get_var_1() %}