Dump query queue info if assessment flag turned on (#87)

google · Aug 30, 2022 · 0ccb489 · 0ccb489
1 parent fdd3e46
commit 0ccb489
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 0 deletions.
diff --git a/...e/edwmigration/dumper/application/dumper/connector/redshift/RedshiftRawLogsConnector.java b/...e/edwmigration/dumper/application/dumper/connector/redshift/RedshiftRawLogsConnector.java
@@ -118,6 +118,15 @@ public void addTasksTo(List<? super Task<?>> out, ConnectorArguments arguments)
             makeTasks(arguments, intervals,
                 RedshiftRawLogsDumpFormat.QueryMetricsHistory.ZIP_ENTRY_PREFIX,
                 queryMetricsTemplateQuery, "starttime", parallelTask);
+
+            String queryQueueInfoTemplateQuery
+                = "SELECT database, query, xid, userid, queue_start_time, "
+                + " exec_start_time, service_class, slots, queue_elapsed, exec_elapsed, "
+                + " wlm_total_elapsed, commit_queue_elapsed, commit_exec_time "
+                + "FROM SVL_QUERY_QUEUE_INFO WHERE ##";
+            makeTasks(arguments, intervals,
+                RedshiftRawLogsDumpFormat.QueryQueueInfo.ZIP_ENTRY_PREFIX,
+                queryQueueInfoTemplateQuery, "queue_start_time", parallelTask);
         }
     }
 

diff --git a/.../java/com/google/edwmigration/dumper/plugin/lib/dumper/spi/RedshiftRawLogsDumpFormat.java b/.../java/com/google/edwmigration/dumper/plugin/lib/dumper/spi/RedshiftRawLogsDumpFormat.java
@@ -70,4 +70,14 @@ public static boolean isZipEntryName(@Nonnull String name) {
             return name.startsWith(ZIP_ENTRY_PREFIX) && name.endsWith(ZIP_ENTRY_SUFFIX);
         }
     }
+
+    interface QueryQueueInfo {
+        String ZIP_ENTRY_PREFIX = "query_queue_info_";
+
+        enum Header {
+            database, query, xid, userid, queue_start_time, exec_start_time,
+            service_class, slots, queue_elapsed, exec_elapsed, wlm_total_elapsed,
+            commit_queue_elapsed, commit_exec_time;
+        }
+    }
 }