fkx

sqle/driver/mysql/rule/rule.go

@@ -280,7 +280,7 @@ func AddResult(result *driverV2.AuditResults, currentRule driverV2.Rule, ruleNam
 		return
 	}
 	level := currentRule.Level
-	message := RuleHandlerMap[ruleName].Message
+	message := GetAllRule()[ruleName].Message
 	result.Add(level, ruleName, plocale.Bundle.LocalizeAll(message), args...)
 }
 
@@ -312,6 +312,17 @@ var (
 	AIRuleHandlerMap = map[string]RuleHandler{}
 )
 
+func GetAllRule() map[string]RuleHandler {
+	allRuleHandlerMap := make(map[string]RuleHandler, len(AIRuleHandlerMap)+len(RuleHandlerMap))
+	for ruleName, aiHandler := range AIRuleHandlerMap {
+		RuleHandlerMap[ruleName] = aiHandler
+	}
+	for ruleName, ruleHandler := range RuleHandlerMap {
+		allRuleHandlerMap[ruleName] = ruleHandler
+	}
+	return allRuleHandlerMap
+}
+
 const DefaultSingleParamKeyName = "first_key" // For most of the rules, it is just has one param, this is first params.
 
 const (