🚀 Optimized alarm route logic

Author: Cairry

alert/consumer/consumer.go

@@ -104,12 +104,7 @@ func (ag *AlertGroups) getNoticeId(alert *models.AlertCurEvent, faultCenter mode
 		labels := alert.Labels
 
 		for _, route := range faultCenter.NoticeRoutes {
-			val, ok := labels[route.Key].(string)
-			if !ok {
-				continue
-			}
-
-			if regexp.MustCompile(route.Value).MatchString(val) {
+			if evalCondition(labels, route.NoticeLabels) {
 				return route.NoticeIds
 			}
 		}
@@ -383,3 +378,37 @@ func (c *Consume) StopAllConsumers() {
 
 	logc.Infof(c.ctx.Ctx, "所有故障中心消费者已停止")
 }
+
+func evalCondition(metrics map[string]interface{}, muteLabels []models.NoticeLabels) bool {
+	for _, muteLabel := range muteLabels {
+		value, exists := metrics[muteLabel.Key]
+		if !exists {
+			return false
+		}
+
+		val, ok := value.(string)
+		if !ok {
+			continue
+		}
+
+		var matched bool
+		switch muteLabel.Operator {
+		case "==", "=":
+			matched = (val == muteLabel.Value)
+		case "!=":
+			matched = (val != muteLabel.Value)
+		case "=~":
+			matched = regexp.MustCompile(muteLabel.Value).MatchString(val)
+		case "!~":
+			matched = !regexp.MustCompile(muteLabel.Value).MatchString(val)
+		default:
+			matched = false
+		}
+
+		if !matched {
+			return false
+		}
+	}
+
+	return true
+}

internal/models/fault_center.go

@@ -49,9 +49,14 @@ type UpgradeStrategy struct {
 }
 
 type NoticeRoute struct {
-	Key       string   `json:"key"`
-	Value     string   `json:"value"`
-	NoticeIds []string `json:"noticeIds" gorm:"column:noticeIds;serializer:json"`
+	NoticeLabels []NoticeLabels `json:"labels" gorm:"column:labels;serializer:json"`
+	NoticeIds    []string       `json:"noticeIds" gorm:"column:noticeIds;serializer:json"`
+}
+
+type NoticeLabels struct {
+	Key      string `json:"key"`
+	Value    string `json:"value"`
+	Operator string `json:"operator"`
 }
 
 func (u *UpgradeStrategy) GetEnabled() bool {