[Improvement][Audit] add sql type for query audit (apache#37790)

Co-authored-by: garenshi <[email protected]>

fe/fe-core/src/main/java/org/apache/doris/analysis/AlterCatalogCommentStmt.java

@@ -46,4 +46,9 @@ public void analyze(Analyzer analyzer) throws UserException {
     public String toSql() {
         return "ALTER CATALOG " + catalogName + " MODIFY COMMENT " + comment;
     }
+
+    @Override
+    public StmtType stmtType() {
+        return StmtType.ALTER;
+    }
 }

fe/fe-core/src/main/java/org/apache/doris/analysis/AlterCatalogNameStmt.java

@@ -56,4 +56,9 @@ public void analyze(Analyzer analyzer) throws UserException {
     public String toSql() {
         return "ALTER CATALOG " + catalogName + " RENAME " + newCatalogName;
     }
+
+    @Override
+    public StmtType stmtType() {
+        return StmtType.ALTER;
+    }
 }

fe/fe-core/src/main/java/org/apache/doris/analysis/AlterCatalogPropertyStmt.java

@@ -54,4 +54,9 @@ public String toSql() {
     public boolean needAuditEncryption() {
         return true;
     }
+
+    @Override
+    public StmtType stmtType() {
+        return StmtType.ALTER;
+    }
 }

fe/fe-core/src/main/java/org/apache/doris/analysis/AlterColocateGroupStmt.java

@@ -79,4 +79,9 @@ public String toSql() {
         sb.append("PROPERTIES(").append(new PrintableMap<>(properties, " = ", true, false)).append(")");
         return sb.toString();
     }
+
+    @Override
+    public StmtType stmtType() {
+        return StmtType.ALTER;
+    }
 }

fe/fe-core/src/main/java/org/apache/doris/analysis/AlterColumnStatsStmt.java

@@ -219,4 +219,9 @@ public String toSql() {
     public String getValue(StatsType statsType) {
         return statsTypeToValue.get(statsType);
     }
+
+    @Override
+    public StmtType stmtType() {
+        return StmtType.ALTER;
+    }
 }

fe/fe-core/src/main/java/org/apache/doris/analysis/AlterDatabasePropertyStmt.java

@@ -76,4 +76,9 @@ public String toSql() {
         return "ALTER DATABASE " + dbName + " SET PROPERTIES ("
                 + new PrintableMap<String, String>(properties, "=", true, false, ",") + ")";
     }
+
+    @Override
+    public StmtType stmtType() {
+        return StmtType.ALTER;
+    }
 }

fe/fe-core/src/main/java/org/apache/doris/analysis/AlterDatabaseQuotaStmt.java

@@ -94,4 +94,9 @@ public String toSql() {
                 + quotaType.name()
                 + " QUOTA " + quotaValue;
     }
+
+    @Override
+    public StmtType stmtType() {
+        return StmtType.ALTER;
+    }
 }

fe/fe-core/src/main/java/org/apache/doris/analysis/AlterDatabaseRename.java

@@ -73,4 +73,9 @@ public String toSql() {
         return "ALTER DATABASE " + dbName + " RENAME " + newDbName;
     }
 
+    @Override
+    public StmtType stmtType() {
+        return StmtType.ALTER;
+    }
+
 }

fe/fe-core/src/main/java/org/apache/doris/analysis/AlterJobStatusStmt.java

@@ -74,4 +74,9 @@ public void analyze(Analyzer analyzer) throws UserException {
             throw new AnalysisException("Can't alter inner job status");
         }
     }
+
+    @Override
+    public StmtType stmtType() {
+        return StmtType.ALTER;
+    }
 }

fe/fe-core/src/main/java/org/apache/doris/analysis/AlterPolicyStmt.java

@@ -96,4 +96,9 @@ public String toSql() {
         return sb.toString();
     }
 
+    @Override
+    public StmtType stmtType() {
+        return StmtType.ALTER;
+    }
+
 }

fe/fe-core/src/main/java/org/apache/doris/analysis/AlterRepositoryStmt.java

@@ -83,4 +83,9 @@ public String toSql() {
         sb.append("PROPERTIES(").append(new PrintableMap<>(properties, " = ", true, false)).append(")");
         return sb.toString();
     }
+
+    @Override
+    public StmtType stmtType() {
+        return StmtType.ALTER;
+    }
 }

fe/fe-core/src/main/java/org/apache/doris/analysis/AlterResourceStmt.java

@@ -87,4 +87,9 @@ public String toSql() {
     public boolean needAuditEncryption() {
         return true;
     }
+
+    @Override
+    public StmtType stmtType() {
+        return StmtType.ALTER;
+    }
 }

fe/fe-core/src/main/java/org/apache/doris/analysis/AlterRoleStmt.java

@@ -62,4 +62,9 @@ public String toSql() {
         sb.append(" COMMENT \"").append(comment).append("\"");
         return sb.toString();
     }
+
+    @Override
+    public StmtType stmtType() {
+        return StmtType.ALTER;
+    }
 }

fe/fe-core/src/main/java/org/apache/doris/analysis/AlterRoutineLoadStmt.java

@@ -264,4 +264,10 @@ private void checkDataSourceProperties() throws UserException {
         dataSourceProperties.setTimezone(job.getTimezone());
         dataSourceProperties.analyze();
     }
+
+    @Override
+    public StmtType stmtType() {
+        return StmtType.ALTER;
+    }
+
 }

fe/fe-core/src/main/java/org/apache/doris/analysis/AlterSqlBlockRuleStmt.java

@@ -138,4 +138,9 @@ public String toSql() {
                 .append(")");
         return sb.toString();
     }
+
+    @Override
+    public StmtType stmtType() {
+        return StmtType.ALTER;
+    }
 }

fe/fe-core/src/main/java/org/apache/doris/analysis/AlterSystemStmt.java

@@ -69,4 +69,9 @@ public String toSql() {
     public String toString() {
         return toSql();
     }
+
+    @Override
+    public StmtType stmtType() {
+        return StmtType.ALTER;
+    }
 }

fe/fe-core/src/main/java/org/apache/doris/analysis/AlterTableStmt.java

@@ -210,4 +210,9 @@ public String toSql() {
     public String toString() {
         return toSql();
     }
+
+    @Override
+    public StmtType stmtType() {
+        return StmtType.ALTER;
+    }
 }

fe/fe-core/src/main/java/org/apache/doris/analysis/AlterUserStmt.java

@@ -159,4 +159,9 @@ public String toSql() {
 
         return sb.toString();
     }
+
+    @Override
+    public StmtType stmtType() {
+        return StmtType.ALTER;
+    }
 }

fe/fe-core/src/main/java/org/apache/doris/analysis/AlterViewStmt.java

@@ -111,4 +111,9 @@ public String toSql() {
     public String toString() {
         return toSql();
     }
+
+    @Override
+    public StmtType stmtType() {
+        return StmtType.ALTER;
+    }
 }

fe/fe-core/src/main/java/org/apache/doris/analysis/AlterWorkloadGroupStmt.java

@@ -66,4 +66,9 @@ public String toSql() {
         sb.append("PROPERTIES(").append(new PrintableMap<>(properties, " = ", true, false)).append(")");
         return sb.toString();
     }
+
+    @Override
+    public StmtType stmtType() {
+        return StmtType.ALTER;
+    }
 }

fe/fe-core/src/main/java/org/apache/doris/analysis/AnalyzeStmt.java

@@ -114,4 +114,9 @@ public boolean forceFull() {
     public boolean usingSqlForExternalTable() {
         return analyzeProperties.usingSqlForExternalTable();
     }
+
+    @Override
+    public StmtType stmtType() {
+        return StmtType.ANALYZE;
+    }
 }

fe/fe-core/src/main/java/org/apache/doris/analysis/BackupStmt.java

@@ -127,4 +127,10 @@ public String toSql() {
         sb.append("\n)");
         return sb.toString();
     }
+
+    @Override
+    public StmtType stmtType() {
+        return StmtType.BACKUP;
+    }
+
 }

fe/fe-core/src/main/java/org/apache/doris/analysis/CancelAlterSystemStmt.java

@@ -90,4 +90,9 @@ public String toSql() {
 
         return sb.toString();
     }
+
+    @Override
+    public StmtType stmtType() {
+        return StmtType.CANCEL;
+    }
 }

fe/fe-core/src/main/java/org/apache/doris/analysis/CancelAlterTableStmt.java

@@ -104,4 +104,9 @@ public String toString() {
         return toSql();
     }
 
+    @Override
+    public StmtType stmtType() {
+        return StmtType.CANCEL;
+    }
+
 }

fe/fe-core/src/main/java/org/apache/doris/analysis/CancelBackupStmt.java

@@ -82,4 +82,9 @@ public String toSql() {
     public String toString() {
         return toSql();
     }
+
+    @Override
+    public StmtType stmtType() {
+        return StmtType.CANCEL;
+    }
 }

fe/fe-core/src/main/java/org/apache/doris/analysis/CancelExportStmt.java

@@ -169,4 +169,9 @@ public String toString() {
         return toSql();
     }
 
+    @Override
+    public StmtType stmtType() {
+        return StmtType.CANCEL;
+    }
+
 }

fe/fe-core/src/main/java/org/apache/doris/analysis/CancelJobTaskStmt.java

@@ -69,4 +69,9 @@ public void analyze(Analyzer analyzer) throws UserException {
         }
         this.taskId = ((IntLiteral) compoundPredicate.getChildren().get(1).getChild(1)).getLongValue();
     }
+
+    @Override
+    public StmtType stmtType() {
+        return StmtType.CANCEL;
+    }
 }

fe/fe-core/src/main/java/org/apache/doris/analysis/CancelLoadStmt.java

@@ -173,4 +173,9 @@ public String toString() {
         return toSql();
     }
 
+    @Override
+    public StmtType stmtType() {
+        return StmtType.CANCEL;
+    }
+
 }

fe/fe-core/src/main/java/org/apache/doris/analysis/CleanLabelStmt.java

@@ -68,4 +68,9 @@ public String toSql() {
     public RedirectStatus getRedirectStatus() {
         return RedirectStatus.FORWARD_WITH_SYNC;
     }
+
+    @Override
+    public StmtType stmtType() {
+        return StmtType.CLEAN;
+    }
 }

fe/fe-core/src/main/java/org/apache/doris/analysis/CleanProfileStmt.java

@@ -47,4 +47,9 @@ public String toSql() {
     public RedirectStatus getRedirectStatus() {
         return RedirectStatus.FORWARD_WITH_SYNC;
     }
+
+    @Override
+    public StmtType stmtType() {
+        return StmtType.CLEAN;
+    }
 }

fe/fe-core/src/main/java/org/apache/doris/analysis/CleanQueryStatsStmt.java

@@ -141,4 +141,9 @@ public String toSql() {
     public enum Scope {
         ALL, DB, TABLE
     }
+
+    @Override
+    public StmtType stmtType() {
+        return StmtType.CLEAN;
+    }
 }

fe/fe-core/src/main/java/org/apache/doris/analysis/CreateCatalogStmt.java

@@ -132,4 +132,9 @@ public String toSql() {
     public boolean needAuditEncryption() {
         return true;
     }
+
+    @Override
+    public StmtType stmtType() {
+        return StmtType.CREATE;
+    }
 }

fe/fe-core/src/main/java/org/apache/doris/analysis/CreateDataSyncJobStmt.java

@@ -127,4 +127,9 @@ public Map<String, String> getProperties() {
     public DataSyncJobType getDataSyncJobType() {
         return dataSyncJobType;
     }
+
+    @Override
+    public StmtType stmtType() {
+        return StmtType.CREATE;
+    }
 }

fe/fe-core/src/main/java/org/apache/doris/analysis/CreateDbStmt.java

@@ -93,4 +93,9 @@ public String toSql() {
         }
         return stringBuilder.toString();
     }
+
+    @Override
+    public StmtType stmtType() {
+        return StmtType.CREATE;
+    }
 }

fe/fe-core/src/main/java/org/apache/doris/analysis/CreateEncryptKeyStmt.java

@@ -91,4 +91,9 @@ public String toSql() {
                 .append(encryptKeyName.getKeyName()).append(" AS \"").append(keyString).append("\"");
         return stringBuilder.toString();
     }
+
+    @Override
+    public StmtType stmtType() {
+        return StmtType.CREATE;
+    }
 }

fe/fe-core/src/main/java/org/apache/doris/analysis/CreateFileStmt.java

@@ -162,4 +162,9 @@ public String toSql() {
     public RedirectStatus getRedirectStatus() {
         return RedirectStatus.FORWARD_WITH_SYNC;
     }
+
+    @Override
+    public StmtType stmtType() {
+        return StmtType.CREATE;
+    }
 }

fe/fe-core/src/main/java/org/apache/doris/analysis/CreateFunctionStmt.java

@@ -902,4 +902,9 @@ public String toSql() {
     public RedirectStatus getRedirectStatus() {
         return RedirectStatus.FORWARD_WITH_SYNC;
     }
+
+    @Override
+    public StmtType stmtType() {
+        return StmtType.CREATE;
+    }
 }

fe/fe-core/src/main/java/org/apache/doris/analysis/CreateMaterializedViewStmt.java

@@ -723,4 +723,9 @@ public static boolean isMVColumnNormal(String name) {
     public String toSql() {
         return null;
     }
+
+    @Override
+    public StmtType stmtType() {
+        return StmtType.CREATE;
+    }
 }

fe/fe-core/src/main/java/org/apache/doris/analysis/CreatePolicyStmt.java

@@ -153,4 +153,9 @@ public String toSql() {
         }
         return sb.toString();
     }
+
+    @Override
+    public StmtType stmtType() {
+        return StmtType.CREATE;
+    }
 }