From 8012421e142da8da53f325aa2ff1d45bb8dd50cb Mon Sep 17 00:00:00 2001
From: eric-laurello <102970824+eric-laurello@users.noreply.github.com>
Date: Tue, 13 Dec 2022 21:55:32 -0500
Subject: [PATCH] custom query tags (#28)

---
 macros/custom_query_tag.sql | 11 +++++++++++
 1 file changed, 11 insertions(+)
 create mode 100644 macros/custom_query_tag.sql

diff --git a/macros/custom_query_tag.sql b/macros/custom_query_tag.sql
new file mode 100644
index 0000000..809e1bb
--- /dev/null
+++ b/macros/custom_query_tag.sql
@@ -0,0 +1,11 @@
+{% macro set_query_tag() -%}
+ {% set new_json = {"repo":project_name, "object":this.table, "profile":target.profile_name, "env":target.name, "existing_tag":get_current_query_tag()  } %}
+{% set new_query_tag = tojson(new_json) | as_text %}
+  {% if new_query_tag %}
+    {% set original_query_tag = get_current_query_tag() %}
+    {{ log("Setting query_tag to '" ~ new_query_tag ~ "'. Will reset to '" ~ original_query_tag ~ "' after materialization.") }}
+    {% do run_query("alter session set query_tag = '{}'".format(new_query_tag)) %}
+    {{ return(original_query_tag)}}
+  {% endif %}
+  {{ return(none)}}
+{% endmacro %}
\ No newline at end of file