✨ [dashboard v0.3.9] 手动即时触发计划任务

service/dao/notification.go

@@ -0,0 +1,81 @@
+package dao
+
+import (
+	"crypto/md5"
+	"encoding/hex"
+	"sync"
+	"time"
+
+	"github.com/naiba/nezha/model"
+)
+
+const firstNotificationDelay = time.Minute * 15
+
+// 通知方式
+var notifications []model.Notification
+var notificationsLock sync.RWMutex
+
+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 !isEdit {
+		notifications = append(notifications, n)
+	}
+}
+
+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--
+		}
+	}
+}
+
+func SendNotification(desc string, muteable bool) {
+	if muteable {
+		// 通知防骚扰策略
+		nID := hex.EncodeToString(md5.New().Sum([]byte(desc)))
+		var flag bool
+		if cacheN, has := Cache.Get(nID); has {
+			nHistory := cacheN.(NotificationHistory)
+			// 每次提醒都增加一倍等待时间，最后每天最多提醒一次
+			if time.Now().After(nHistory.Until) {
+				flag = true
+				nHistory.Duration *= 2
+				if nHistory.Duration > time.Hour*24 {
+					nHistory.Duration = time.Hour * 24
+				}
+				nHistory.Until = time.Now().Add(nHistory.Duration)
+				// 缓存有效期加 10 分钟
+				Cache.Set(nID, nHistory, nHistory.Duration+time.Minute*10)
+			}
+		} else {
+			// 新提醒直接通知
+			flag = true
+			Cache.Set(nID, NotificationHistory{
+				Duration: firstNotificationDelay,
+				Until:    time.Now().Add(firstNotificationDelay),
+			}, firstNotificationDelay+time.Minute*10)
+		}
+
+		if !flag {
+			return
+		}
+	}
+	// 发出通知
+	notificationsLock.RLock()
+	defer notificationsLock.RUnlock()
+	for i := 0; i < len(notifications); i++ {
+		notifications[i].Send(desc)
+	}
+}