Instrument run_hooks

dbt-adapters/src/dbt/include/global_project/macros/materializations/hooks.sql

@@ -1,17 +1,25 @@
-{% macro run_hooks(hooks, inside_transaction=True) %}
-  {% for hook in hooks | selectattr('transaction', 'equalto', inside_transaction)  %}
-    {% if not inside_transaction and loop.first %}
-      {% call statement(auto_begin=inside_transaction) %}
-        commit;
-      {% endcall %}
-    {% endif %}
-    {% set rendered = render(hook.get('sql')) | trim %}
-    {% if (rendered | length) > 0 %}
-      {% call statement(auto_begin=inside_transaction) %}
-        {{ rendered }}
-      {% endcall %}
-    {% endif %}
-  {% endfor %}
+{% macro run_hooks(hooks, inside_transaction=True, span_name='run_hooks') %}
+  {% if hooks|length > 0 %}
+      {% set tracer = modules.opentelemetry.trace.get_tracer("dbt-runner") %}
+      {% set hook_span = tracer.start_span(span_name, context=modules.opentelemetry.context.get_current())%}
+      {% set ctx = modules.opentelemetry.trace.set_span_in_context(hook_span) %}
+      {% set token = modules.opentelemetry.context.attach(ctx) %}
+      {% for hook in hooks | selectattr('transaction', 'equalto', inside_transaction)  %}
+        {% if not inside_transaction and loop.first %}
+          {% call statement(auto_begin=inside_transaction) %}
+            commit;
+{% endcall %}
+        {% endif %}
+        {% set rendered = render(hook.get('sql')) | trim %}
+        {% if (rendered | length) > 0 %}
+          {% call statement(auto_begin=inside_transaction) %}
+            {{ rendered }}
+          {% endcall %}
+        {% endif %}
+      {% endfor %}
+      {% set _ = modules.opentelemetry.context.detach(token) %}
+      {% set _ = hook_span.end() %}
+  {% endif %}
 {% endmacro %}
 
 

dbt-snowflake/src/dbt/include/snowflake/macros/relations/view/create.sql

@@ -49,7 +49,7 @@ eventually be retired in favor of a standardized approach. Changed line:
       type='view') -%}
   {% set grant_config = config.get('grants') %}
 
-  {{ run_hooks(pre_hooks) }}
+  {{ run_hooks(pre_hooks, span_name='pre-hook') }}
 
   -- If there's a table with the same name and we weren't told to full refresh,
   -- that's an error. If we were told to full refresh, drop it. This behavior differs
@@ -66,7 +66,7 @@ eventually be retired in favor of a standardized approach. Changed line:
   {% set should_revoke = should_revoke(exists_as_view, full_refresh_mode=True) %}
   {% do apply_grants(target_relation, grant_config, should_revoke=should_revoke) %}
 
-  {{ run_hooks(post_hooks) }}
+  {{ run_hooks(post_hooks, span_name='post-hook') }}
 
   {{ return({'relations': [target_relation]}) }}