Update seed executions with compile completed at/query started at

swdotcom · Aug 16, 2022 · a8ebe2d · a8ebe2d
1 parent 85e6868
commit a8ebe2d
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 3 deletions.
diff --git a/macros/create_seed_executions_table_if_not_exists.sql b/macros/create_seed_executions_table_if_not_exists.sql
@@ -40,12 +40,14 @@
     create table {{database_name}}.{{schema_name}}.{{table_name}} (
         command_invocation_id STRING,
         node_id STRING,
-        run_started_at TIMESTAMP_TZ,
+        run_started_at TIMESTAMP_NTZ,
         was_full_refresh BOOLEAN,
         thread_id STRING,
         status STRING,
-        compile_started_at TIMESTAMP_TZ,
-        query_completed_at TIMESTAMP_TZ,
+        compile_started_at TIMESTAMP_NTZ,
+        compile_completed_at TIMESTAMP_NTZ,
+        query_started_at TIMESTAMP_NTZ,
+        query_completed_at TIMESTAMP_NTZ,
         total_node_runtime INTEGER,
         rows_affected INTEGER,
         materialization STRING,
@@ -64,6 +66,8 @@
         thread_id STRING,
         status STRING,
         compile_started_at TIMESTAMP,
+        compile_completed_at TIMESTAMP,
+        query_started_at TIMESTAMP,
         query_completed_at TIMESTAMP,
         total_node_runtime INTEGER,
         rows_affected INTEGER,

diff --git a/macros/upload_seed_executions.sql b/macros/upload_seed_executions.sql
@@ -44,20 +44,26 @@
                     {% for stage in model.timing if stage.name == "compile" %}
                         {% if loop.length == 0 %}
                             null, {# compile_started_at #}
+                            null, {# compile_completed_at #}
                         {% else %}
                             '{{ stage.started_at }}', {# compile_started_at #}
+                            '{{ stage.completed_at }}', {# compile_completed_at #}
                         {% endif %}
                     {% endfor %}
 
                     {% for stage in model.timing if stage.name == "execute" %}
                         {% if loop.length == 0 %}
+                            null, {# query_started_at #}
                             null, {# query_completed_at #}
                         {% else %}
+                            '{{ stage.started_at }}', {# query_started_at #}
                             '{{ stage.completed_at }}', {# query_completed_at #}
                         {% endif %}
                     {% endfor %}
                 {% else %}
                     null, {# compile_started_at #}
+                    null, {# compile_completed_at #}
+                    null, {# query_started_at #}
                     null, {# query_completed_at #}
                 {% endif %}