dbt dbt_utils 包中的pivot简单说明

dbt-utils 中的pivot macro 实际上就是实现了我们平时会碰到的一些sql 行转列问题

dbt-utils pivot实现处理

行转列的方法很多有基于case 判断的，也有直接使用数据库提供的pivot 函数的，dbt 使用了case 模式

• 参考实现

{% macro pivot(column,

               values,

               alias=True,

               agg='sum',

               cmp='=',

               prefix='',

               suffix='',

               then_value=1,

               else_value=0,

               quote_identifiers=True,

               distinct=False) %}

    {{ return(adapter.dispatch('pivot', 'dbt_utils')(column, values, alias, agg, cmp, prefix, suffix, then_value, else_value, quote_identifiers, distinct)) }}

{% endmacro %}

 

{% macro default__pivot(column,

               values,

               alias=True,

               agg='sum',

               cmp='=',

               prefix='',

               suffix='',

               then_value=1,

               else_value=0,

               quote_identifiers=True,

               distinct=False) %}

 # 此处核心是进行循环的case 处理，当然还会结合实际上的聚合函数以及别名配置进行一些处理

  {% for value in values %}

    {{ agg }}(

      {% if distinct %} distinct {% endif %}

      case

      when {{ column }} {{ cmp }} '{{ dbt.escape_single_quotes(value) }}'

        then {{ then_value }}

      else {{ else_value }}

      end

    )

    {% if alias %}

      {% if quote_identifiers %}

            as {{ adapter.quote(prefix ~ value ~ suffix) }}

      {% else %}

        as {{ dbt_utils.slugify(prefix ~ value ~ suffix) }}

      {% endif %}

    {% endif %}

    {% if not loop.last %},{% endif %}

  {% endfor %}

{% endmacro %}

参考资料

https://github.com/dbt-labs/dbt-utils
https://docs.dremio.com/current/reference/sql/commands/SELECT-statements/#parameters
https://blogs.oracle.com/sql/post/how-to-convert-rows-to-columns-and-back-again-with-sql-aka-pivot-and-unpivot