Merge pull request #801 from elementary-data/improve-temp-tables-macro

Improve temp tables macro

Author: GuyEshdat

macros/utils/table_operations/create_temp_table.sql

@@ -1,14 +1,36 @@
 {% macro create_temp_table(database_name, schema_name, table_name, sql_query) %}
+    {% do return(adapter.dispatch('create_temp_table','elementary')(database_name, schema_name, table_name, sql_query)) %}
+{%- endmacro %}
+
+{% macro default__create_temp_table(database_name, schema_name, table_name, sql_query) %}
     {% set temp_table_exists, temp_table_relation = dbt.get_or_create_relation(database=database_name,
                                                                                schema=schema_name,
                                                                                identifier=table_name,
                                                                                type='table') -%}
-    {% set temp_table_relation = elementary.edr_make_temp_relation(temp_table_relation) %}
+    {% set temp_table_relation = elementary.make_temp_table_relation(temp_table_relation) %}
     {% if temp_table_exists %}
         {% do adapter.drop_relation(temp_table_relation) %}
         {% do elementary.run_query(dbt.create_table_as(True, temp_table_relation, sql_query)) %}
     {% else %}
         {% do elementary.run_query(dbt.create_table_as(True, temp_table_relation, sql_query)) %}
     {% endif %}
+    {{ return(temp_table_relation) }}{% endmacro %}
+
+{% macro snowflake__create_temp_table(database_name, schema_name, table_name, sql_query) %}
+    {% set temp_table_exists, temp_table_relation = dbt.get_or_create_relation(database=database_name,
+                                                                               schema=schema_name,
+                                                                               identifier=table_name,
+                                                                               type='table') -%}
+    {% set temp_table_relation = elementary.make_temp_table_relation(temp_table_relation) %}
+    {% set create_query %}
+        create or replace temporary table {{ temp_table_relation }} 
+        as (
+            {{ sql_query }}
+        );
+        
+    {% endset %}
+
+    {% do elementary.run_query(create_query) %}
+    
     {{ return(temp_table_relation) }}
 {% endmacro %}