[Enhancement] add warehouse parameter to audit log table

Signed-off-by: Rohit Satardekar <[email protected]>

Author: rohitrs1983

README.md

@@ -51,7 +51,8 @@ CREATE TABLE starrocks_audit_db__.starrocks_audit_tbl__ (
   `planMemCosts` DOUBLE COMMENT "查询规划阶段内存占用（字节）",
   `pendingTimeMs` BIGINT COMMENT "查询在队列中等待的时间（毫秒）",
   `candidateMVs` varchar(65533) NULL COMMENT "候选MV列表",
-  `hitMvs` varchar(65533) NULL COMMENT "命中MV列表"
+  `hitMvs` varchar(65533) NULL COMMENT "命中MV列表",
+  `warehouse` VARCHAR(128) NULL COMMENT "仓库名称" 
 ) ENGINE = OLAP
 DUPLICATE KEY (`queryId`, `timestamp`, `queryType`)
 COMMENT "审计日志表"

lib/starrocks-fe.jar

@@ -196,7 +196,8 @@ private void assembleAudit(AuditEvent event) {
                 "\"planMemCosts\":" + event.planMemCosts + "," +
                 "\"pendingTimeMs\":" + event.pendingTimeMs + "," +
                 "\"candidateMVs\":\"" + candidateMvsVal + "\"," +
-                "\"hitMvs\":\"" + hitMVsVal + "\"}";
+                "\"hitMvs\":\"" + hitMVsVal + "\"," +
+                "\"warehouse\":\"" + event.warehouse + "\"}";
         if (auditBuffer.length() > 0) {
             auditBuffer.append(",");
         }

src/main/java/com/starrocks/plugin/audit/AuditLoaderPlugin.java

@@ -81,7 +81,7 @@ private HttpURLConnection getConnection(String urlStr, String label) throws IOEx
 
         conn.addRequestProperty("label", label);
         conn.addRequestProperty("max_filter_ratio", "1.0");
-        conn.addRequestProperty("columns", "queryId,timestamp,queryType,clientIp,user,authorizedUser,resourceGroup,catalog,db,state,errorCode,queryTime,scanBytes,scanRows,returnRows,cpuCostNs,memCostBytes,stmtId,isQuery,feIp,stmt,digest,planCpuCosts,planMemCosts,pendingTimeMs,candidateMVs,hitMvs");
+        conn.addRequestProperty("columns", "queryId,timestamp,queryType,clientIp,user,authorizedUser,resourceGroup,catalog,db,state,errorCode,queryTime,scanBytes,scanRows,returnRows,cpuCostNs,memCostBytes,stmtId,isQuery,feIp,stmt,digest,planCpuCosts,planMemCosts,pendingTimeMs,candidateMVs,hitMvs,warehouse");
         if(!StringUtils.isBlank(this.streamLoadFilter)) {
             conn.addRequestProperty("where", streamLoadFilter);
         }
@@ -104,7 +104,7 @@ private String toCurl(HttpURLConnection conn) {
             sb.append("-H \"").append("where\":").append(streamLoadFilter).append(" \\\n  ");
         }
         sb.append("-H \"").append("columns\":").append("\"queryId, timestamp, queryType, clientIp, user, authorizedUser, resourceGroup, catalog, db, state, errorCode," +
-                "queryTime, scanBytes, scanRows, returnRows, cpuCostNs, memCostBytes, stmtId, isQuery, feIp, stmt, digest, planCpuCosts, planMemCosts, pendingTimeMs, candidateMVs, hitMvs\" \\\n  ");
+                "queryTime, scanBytes, scanRows, returnRows, cpuCostNs, memCostBytes, stmtId, isQuery, feIp, stmt, digest, planCpuCosts, planMemCosts, pendingTimeMs, candidateMVs, hitMvs, warehouse\" \\\n  ");
         sb.append("\"").append(conn.getURL()).append("\"");
         return sb.toString();
     }