Add some macros and use in one place

dbt-labs · Oct 18, 2024 · 2f499d5 · 2f499d5
1 parent 636e6b8
commit 2f499d5
Showing 1 changed file with 34 additions and 21 deletions.
diff --git a/dbt/include/global_project/macros/materializations/snapshots/helpers.sql b/dbt/include/global_project/macros/materializations/snapshots/helpers.sql
@@ -98,27 +98,10 @@
             source_data.*
 
         from insertions_source_data as source_data
-        left outer join snapshotted_data on
-            {% if strategy.unique_key | is_list %}
-                {% for key in strategy.unique_key %}
-                    snapshotted_data.dbt_unique_key_{{ loop.index }} = source_data.dbt_unique_key_{{ loop.index }}
-                    {%- if not loop.last %} and {%- endif %}
-                {% endfor %}
-                where snapshotted_data.dbt_unique_key_1 is null
-                or (
-                    snapshotted_data.dbt_unique_key_1 is not null
-                and (
-                    {{ strategy.row_changed }}
-                )
-            {% else %}
-                snapshotted_data.dbt_unique_key = source_data.dbt_unique_key
-                where snapshotted_data.dbt_unique_key is null
-                or (
-                    snapshotted_data.dbt_unique_key is not null
-                and (
-                    {{ strategy.row_changed }}
-                )
-            {% endif %}
+        left outer join snapshotted_data
+            on {{ unique_key_join_on(strategy.unique_key, "snapshotted_data", "source_data") }}
+            where {{ unique_key_is_null(strategy.unique_key, "snapshotted_data") }}
+            or ({{ unique_key_is_not_null(strategy.unique_key, "snapshotted_data") }} and {{ strategy.row_changed }})
 
         )
 
@@ -259,3 +242,33 @@
         {{ unique_key }} as dbt_unique_key
     {% endif %}
 {% endmacro %}
+
+
+{% macro unique_key_join_on(unique_key, identifier, from_identifier) %}
+    {% if strategy.unique_key | is_list %}
+        {% for key in strategy.unique_key %}
+            {{ identifier }}.dbt_unique_key_{{ loop.index }} = {{ from_identifier }}.dbt_unique_key_{{ loop.index }}
+            {%- if not loop.last %} and {%- endif %}
+        {% endfor %}
+    {% else %}
+        {{ identifier }}.dbt_unique_key = {{ from_identifier }}.dbt_unique_key
+    {% endif %}
+{% endmacro %}
+
+
+{% macro unique_key_is_null(unique_key, identifier) %}
+    {% if unique_key | is_list %}
+        {{ identifier }}.dbt_unique_key_1 is null
+    {% else %}
+        {{ identifer }}.dbt_unique_key is null
+    {% endif %}
+{% endmacro %}
+
+
+{% macro unique_key_is_not_null(unique_key, identifier) %}
+    {% if unique_key | is_list %}
+        {{ identifier }}.dbt_unique_key_1 is not null
+    {% else %}
+        {{ identifer }}.dbt_unique_key is not null
+    {% endif %}
+{% endmacro %}