WIP: 通知方式分组 支持将不同的报警｜监控｜计划任务的通知 发送到指定的通知分组

service/singleton/notification.go

@@ -13,46 +13,95 @@ import (
 const firstNotificationDelay = time.Minute * 15
 
 // 通知方式
-var notifications []model.Notification
-var notificationsLock sync.RWMutex
+var (
+	NotificationList    map[string]map[uint64]*model.Notification // [NotificationMethodTag][NotificationID] -> model.Notification
+	NotificationIDToTag map[uint64]string                         // [NotificationID] -> NotificationTag
+	notificationsLock   sync.RWMutex
+)
 
-// LoadNotifications 从 DB 加载通知方式到 singleton.notifications 变量
+// InitNotification 初始化 Tag <-> ID <-> Notification 的映射
+func InitNotification() {
+	NotificationList = make(map[string]map[uint64]*model.Notification)
+	NotificationIDToTag = make(map[uint64]string)
+}
+
+// LoadNotifications 从 DB 初始化通知方式相关参数
 func LoadNotifications() {
+	InitNotification()
 	notificationsLock.Lock()
+	defer notificationsLock.Unlock()
+
+	var notifications []model.Notification
 	if err := DB.Find(&notifications).Error; err != nil {
 		panic(err)
 	}
-	notificationsLock.Unlock()
+	for _, n := range notifications {
+		// 旧版本的Tag可能不存在 自动设置为默认值
+		if n.Tag == "" {
+			SetDefaultNotificationTagInDB(&n)
+		}
+		AddNotificationToList(&n)
+	}
 }
 
-func OnRefreshOrAddNotification(n model.Notification) {
+// SetDefaultNotificationTagInDB 设置默认通知方式的 Tag
+func SetDefaultNotificationTagInDB(n *model.Notification) {
+	n.Tag = "default"
+	if err := DB.Save(n).Error; err != nil {
+		log.Println("[ERROR]", err)
+	}
+}
+
+// OnRefreshOrAddNotification 刷新通知方式相关参数
+func OnRefreshOrAddNotification(n *model.Notification) {
 	notificationsLock.Lock()
 	defer notificationsLock.Unlock()
+
 	var isEdit bool
-	for i := 0; i < len(notifications); i++ {
-		if notifications[i].ID == n.ID {
-			notifications[i] = n
-			isEdit = true
-		}
+	if _, ok := NotificationList[n.Tag][n.ID]; ok {
+		isEdit = true
 	}
 	if !isEdit {
-		notifications = append(notifications, n)
+		AddNotificationToList(n)
+	} else {
+		UpdateNotificationInList(n)
 	}
 }
 
+// AddNotificationToList 添加通知方式到map中
+func AddNotificationToList(n *model.Notification) {
+	notificationsLock.Lock()
+	defer notificationsLock.Unlock()
+
+	// 当前 Tag 不存在，创建对应该 Tag 的 子 map 后再添加
+	if _, ok := NotificationList[n.Tag]; !ok {
+		NotificationList[n.Tag] = make(map[uint64]*model.Notification)
+	}
+	NotificationList[n.Tag][n.ID] = n
+	NotificationIDToTag[n.ID] = n.Tag
+
+}
+
+// UpdateNotificationInList 在 map 中更新通知方式
+func UpdateNotificationInList(n *model.Notification) {
+	notificationsLock.Lock()
+	defer notificationsLock.Unlock()
+
+	NotificationList[n.Tag][n.ID] = n
+}
+
+// OnDeleteNotification 在map中删除通知方式
 func OnDeleteNotification(id uint64) {
 	notificationsLock.Lock()
 	defer notificationsLock.Unlock()
-	for i := 0; i < len(notifications); i++ {
-		if notifications[i].ID == id {
-			notifications = append(notifications[:i], notifications[i+1:]...)
-			i--
-		}
-	}
+
+	delete(NotificationList[NotificationIDToTag[id]], id)
+	delete(NotificationIDToTag, id)
 }
 
-func SendNotification(desc string, muteable bool) {
-	if muteable {
+// SendNotification 向指定的通知方式组的所有通知方式发送通知
+func SendNotification(notificationTag string, desc string, mutable bool) {
+	if mutable {
 		// 通知防骚扰策略
 		nID := hex.EncodeToString(md5.New().Sum([]byte(desc))) // #nosec
 		var flag bool
@@ -80,16 +129,17 @@ func SendNotification(desc string, muteable bool) {
 
 		if !flag {
 			if Conf.Debug {
-				log.Println("NEZHA>> 静音的重复通知：", desc, muteable)
+				log.Println("NEZHA>> 静音的重复通知：", desc, mutable)
 			}
 			return
 		}
 	}
-	// 发出通知
+	// 向该通知方式组的所有通知方式发出通知
 	notificationsLock.RLock()
 	defer notificationsLock.RUnlock()
-	for i := 0; i < len(notifications); i++ {
-		if err := notifications[i].Send(desc); err != nil {
+
+	for _, n := range NotificationList[notificationTag] {
+		if err := n.Send(desc); err != nil {
 			log.Println("NEZHA>> 发送通知失败：", err)
 		}
 	}