Merge pull request #148 from dbt-labs/feat/147/add-dispatching

Add dispatch to macros
dbt-labs · Dec 7, 2023 · ee4899e · ee4899e
2 parents e8b5a02 + 15c4704
commit ee4899e
Show file tree

Hide file tree

Showing 5 changed files with 26 additions and 4 deletions.
diff --git a/macros/create_base_models.sql b/macros/create_base_models.sql
@@ -1,4 +1,8 @@
 {% macro create_base_models(source_name, tables) %}
+    {{ return(adapter.dispatch('create_base_models', 'codegen')(source_name, tables)) }}
+{% endmacro %}
+
+{% macro default__create_base_models(source_name, tables) %}
 
 {% set source_name = ""~ source_name ~"" %}
 

diff --git a/macros/generate_base_model.sql b/macros/generate_base_model.sql
@@ -1,4 +1,8 @@
 {% macro generate_base_model(source_name, table_name, leading_commas=False, case_sensitive_cols=False, materialized=None) %}
+  {{ return(adapter.dispatch('generate_base_model', 'codegen')(source_name, table_name, leading_commas, case_sensitive_cols, materialized)) }}
+{% endmacro %}
+
+{% macro default__generate_base_model(source_name, table_name, leading_commas, case_sensitive_cols, materialized) %}
 
 {%- set source_relation = source(source_name, table_name) -%}
 

diff --git a/macros/generate_model_import_ctes.sql b/macros/generate_model_import_ctes.sql
@@ -1,4 +1,8 @@
-{% macro generate_model_import_ctes(model_name, leading_commas = false) %}
+{% macro generate_model_import_ctes(model_name, leading_commas = False) %}
+    {{ return(adapter.dispatch('generate_model_import_ctes', 'codegen')(model_name, leading_commas)) }}
+{% endmacro %}
+
+{% macro default__generate_model_import_ctes(model_name, leading_commas) %}
 
     {%- if execute -%}
     {%- set nodes = graph.nodes.values() -%}

diff --git a/macros/generate_model_yaml.sql b/macros/generate_model_yaml.sql
@@ -1,4 +1,8 @@
 {% macro generate_column_yaml(column, model_yaml, column_desc_dict, include_data_types, parent_column_name="") %}
+  {{ return(adapter.dispatch('generate_column_yaml', 'codegen')(column, model_yaml, column_desc_dict, include_data_types, parent_column_name)) }}
+{% endmacro %}
+
+{% macro default__generate_column_yaml(column, model_yaml, column_desc_dict, include_data_types, parent_column_name) %}
     {% if parent_column_name %}
         {% set column_name = parent_column_name ~ "." ~ column.name %}
     {% else %}
@@ -20,7 +24,11 @@
     {% do return(model_yaml) %}
 {% endmacro %}
 
-{% macro generate_model_yaml(model_names=[], upstream_descriptions=False, include_data_types=True) %}
+{% macro generate_model_yaml(model_names=[], upstream_descriptions=False, include_data_types=True) -%}
+  {{ return(adapter.dispatch('generate_model_yaml', 'codegen')(model_names, upstream_descriptions, include_data_types)) }}
+{%- endmacro %}
+
+{% macro default__generate_model_yaml(model_names, upstream_descriptions, include_data_types) %}
 
     {% set model_yaml=[] %}
 

diff --git a/macros/generate_source.sql b/macros/generate_source.sql
@@ -13,9 +13,11 @@
 
 {% endmacro %}
 
-
----
 {% macro generate_source(schema_name, database_name=target.database, generate_columns=False, include_descriptions=False, include_data_types=True, table_pattern='%', exclude='', name=schema_name, table_names=None, include_database=False, include_schema=False) %}
+    {{ return(adapter.dispatch('generate_source', 'codegen')(schema_name, database_name, generate_columns, include_descriptions, include_data_types, table_pattern, exclude, name, table_names, include_database, include_schema)) }}
+{% endmacro %}
+
+{% macro default__generate_source(schema_name, database_name, generate_columns, include_descriptions, include_data_types, table_pattern, exclude, name, table_names, include_database, include_schema) %}
 
 {% set sources_yaml=[] %}
 {% do sources_yaml.append('version: 2') %}