add query type column

wangbo · Feb 21, 2024 · f13fdc9 · f13fdc9
1 parent 89ce566
commit f13fdc9
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 0 deletions.
diff --git a/fe/fe-core/src/main/java/org/apache/doris/qe/QeProcessorImpl.java b/fe/fe-core/src/main/java/org/apache/doris/qe/QeProcessorImpl.java
@@ -307,6 +307,10 @@ public String getSql() {
         public long getStartExecTime() {
             return startExecTime;
         }
+
+        public QueryType getQueryType() {
+            return queryType;
+        }
     }
 
     private class WriteProfileTask implements Runnable {

diff --git a/...e-core/src/main/java/org/apache/doris/tablefunction/ActiveQueriesTableValuedFunction.java b/...e-core/src/main/java/org/apache/doris/tablefunction/ActiveQueriesTableValuedFunction.java
@@ -50,6 +50,7 @@ public class ActiveQueriesTableValuedFunction extends MetadataTableValuedFunctio
             new Column("ShuffleSendRows", PrimitiveType.BIGINT),
             new Column("Database", ScalarType.createStringType()),
             new Column("FrontendInstance", ScalarType.createStringType()),
+            new Column("QueryType", ScalarType.createStringType()),
             new Column("Sql", ScalarType.createStringType()));
 
     private static final ImmutableMap<String, Integer> COLUMN_TO_INDEX;

diff --git a/fe/fe-core/src/main/java/org/apache/doris/tablefunction/MetadataGenerator.java b/fe/fe-core/src/main/java/org/apache/doris/tablefunction/MetadataGenerator.java
@@ -463,6 +463,7 @@ private static TRow makeQueryStatisticsTRow(SimpleDateFormat sdf, String queryId
             trow.addToColumnValue(new TCell().setStringVal(""));
         }
         trow.addToColumnValue(new TCell().setStringVal(selfNode));
+        trow.addToColumnValue(new TCell().setStringVal(queryInfo.getQueryType()));
         trow.addToColumnValue(new TCell().setStringVal(queryInfo.getSql()));
 
         return trow;