fix ai rules mysql

actiontech · Jan 24, 2025 · 95dd379 · 95dd379
1 parent 7e11e74
commit 95dd379
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/sqle/driver/mysql/mysql.go b/sqle/driver/mysql/mysql.go
@@ -377,7 +377,10 @@ func (i *MysqlDriverImpl) audit(ctx context.Context, sql string) (*driverV2.Audi
 
 		handler, ok := rulepkg.RuleHandlerMap[rule.Name]
 		if !ok || handler.Func == nil {
-			continue
+			handler, ok = rulepkg.AIRuleHandlerMap[rule.Name]
+			if !ok || handler.Func == nil {
+				continue
+			}
 		}
 		if i.IsOfflineAudit() && !handler.IsAllowOfflineRule(nodes[0]) {
 			continue