Update trino__get_columns_in_relation to use information_schema.columns

starburstdata · Dec 18, 2024 · 56505ee · 56505ee
1 parent cb25ea8
commit 56505ee
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 2 deletions.
diff --git a/.changes/unreleased/Under the Hood-20241105-083613.yaml b/.changes/unreleased/Under the Hood-20241105-083613.yaml
@@ -0,0 +1,7 @@
+kind: Under the Hood
+body: Update trino__get_columns_in_relation to use information_schema.columns
+time: 2024-11-05T08:36:13.788945-05:00
+custom:
+  Author: posulliv
+  Issue: "443"
+  PR: "444"
diff --git a/dbt/include/trino/macros/adapters.sql b/dbt/include/trino/macros/adapters.sql
@@ -5,7 +5,12 @@
 
 {% macro trino__get_columns_in_relation(relation) -%}
   {%- set sql -%}
-    describe {{ relation }}
+    select column_name, data_type
+    from {{ relation.information_schema() }}.columns
+    where
+      table_catalog = '{{ relation.database | lower }}'
+      and table_schema = '{{ relation.schema | lower }}'
+      and table_name = '{{ relation.identifier  | lower}}'
   {%- endset -%}
   {%- set result = run_query(sql) -%}
 
@@ -20,7 +25,7 @@
 
   {% set columns = [] %}
   {% for row in result %}
-    {% do columns.append(api.Column.from_description(row['Column'].lower(), row['Type'])) %}
+    {% do columns.append(api.Column.from_description(row['column_name'].lower(), row['data_type'])) %}
   {% endfor %}
   {% do return(columns) %}
 {% endmacro %}