From bdd7b68ebdb3cec6fb4a1d21e5b786d17af40688 Mon Sep 17 00:00:00 2001
From: eric-laurello <102970824+eric-laurello@users.noreply.github.com>
Date: Fri, 8 Sep 2023 08:59:36 -0400
Subject: [PATCH] add select tags (#135)

---
 dbt_project.yml             | 13 ++++++++++++-
 macros/custom_query_tag.sql | 11 -----------
 packages.yml                |  4 +++-
 3 files changed, 15 insertions(+), 13 deletions(-)
 delete mode 100644 macros/custom_query_tag.sql

diff --git a/dbt_project.yml b/dbt_project.yml
index cfd17fa..630f9bd 100644
--- a/dbt_project.yml
+++ b/dbt_project.yml
@@ -56,4 +56,15 @@ on-run-start:
   - '{{sp_create_bulk_get_pool_balances()}}'
 
 on-run-end:
-  - '{{ apply_meta_as_tags(results) }}'
\ No newline at end of file
+  - '{{ apply_meta_as_tags(results) }}'
+
+dispatch:
+  - macro_namespace: dbt
+    search_order:
+      - osmosis-models
+      - dbt_snowflake_query_tags
+      - dbt
+
+query-comment:
+  comment: '{{ dbt_snowflake_query_tags.get_query_comment(node) }}'
+  append: true # Snowflake removes prefixed comments.
\ No newline at end of file
diff --git a/macros/custom_query_tag.sql b/macros/custom_query_tag.sql
deleted file mode 100644
index 809e1bb..0000000
--- a/macros/custom_query_tag.sql
+++ /dev/null
@@ -1,11 +0,0 @@
-{% 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
diff --git a/packages.yml b/packages.yml
index 62c8c0b..4ad5d9d 100644
--- a/packages.yml
+++ b/packages.yml
@@ -6,4 +6,6 @@ packages:
   - package: dbt-labs/dbt_external_tables
     version: 0.8.0
   - git: https://github.com/FlipsideCrypto/fsc-utils.git
-    revision: "v1.3.0"
\ No newline at end of file
+    revision: "v1.3.0"
+  - package: get-select/dbt_snowflake_query_tags
+    version: [">=2.0.0", "<3.0.0"]
\ No newline at end of file