Update snapshot_merge.sql

dbt-labs · Aug 28, 2024 · 13e5655 · 13e5655
1 parent 5575fa8
commit 13e5655
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/dbt/include/postgres/macros/materializations/snapshot_merge.sql b/dbt/include/postgres/macros/materializations/snapshot_merge.sql
@@ -2,12 +2,15 @@
 {% macro postgres__snapshot_merge_sql(target, source, insert_cols) -%}
     {%- set insert_cols_csv = insert_cols | join(', ') -%}
 
+    {%- set dbt_scd_id = config.get("dbt_scd_id_column_name") or "dbt_scd_id" -%}
+    {%- set dbt_valid_to = config.get("dbt_valid_to_column_name") or "dbt_valid_to" -%}
+
     update {{ target }}
-    set dbt_valid_to = DBT_INTERNAL_SOURCE.dbt_valid_to
+    set {{ dbt_valid_to }} = DBT_INTERNAL_SOURCE.{{ dbt_valid_to }}
     from {{ source }} as DBT_INTERNAL_SOURCE
-    where DBT_INTERNAL_SOURCE.dbt_scd_id::text = {{ target }}.dbt_scd_id::text
+    where DBT_INTERNAL_SOURCE.{{ dbt_scd_id }}::text = {{ target }}.{{ dbt_scd_id }}::text
       and DBT_INTERNAL_SOURCE.dbt_change_type::text in ('update'::text, 'delete'::text)
-      and {{ target }}.dbt_valid_to is null;
+      and {{ target }}.{{ dbt_valid_to }} is null;
 
     insert into {{ target }} ({{ insert_cols_csv }})
     select {% for column in insert_cols -%}