tcjlj/nezha
1
0
Fork 0
mirror of https://github.com/nezhahq/nezha.git synced 2025-01-22 20:58:14 -05:00
Code Issues Actions Packages Projects Releases Wiki Activity

update: 调度失败的计划任务仅通知其所属的通知组

Browse Source
This commit is contained in:
Akkia 2022-04-15 10:55:21 +08:00
parent 615e337b42
commit ab033f3b2e
No known key found for this signature in database
GPG Key ID: DABE9A4AB2DD7EF3
1 changed files with 11 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
service/singleton/crontask.go
View File

@ -28,8 +28,8 @@ func LoadCronTasks() {
var crons []model.Cron
DB.Find(&crons)
var err error
errMsg := new(bytes.Buffer)
var notificationTagList []string
notificationMsgMap := make(map[string]*bytes.Buffer)
for i := range crons {
// 旧版本计划任务可能不存在通知组 为其添加默认通知组
if crons[i].NotificationTag == "" {
@ -40,19 +40,19 @@ func LoadCronTasks() {
if err == nil {
Crons[crons[i].ID] = &crons[i]
} else {
if errMsg.Len() == 0 {
errMsg.WriteString("调度失败的计划任务：[")
}
errMsg.WriteString(fmt.Sprintf("%d,", crons[i].ID))
// 当前通知组首次出现 将其加入通知组列表并初始化通知组消息缓存
if _, ok := notificationMsgMap[crons[i].NotificationTag]; !ok {
notificationTagList = append(notificationTagList, crons[i].NotificationTag)
notificationMsgMap[crons[i].NotificationTag] = bytes.NewBufferString("")
notificationMsgMap[crons[i].NotificationTag].WriteString("调度失败的计划任务：[")
}
notificationMsgMap[crons[i].NotificationTag].WriteString(fmt.Sprintf("%d,", crons[i].ID))
}
}
if errMsg.Len() > 0 {
msg := errMsg.String() + "] 这些任务将无法正常执行,请进入后点重新修改保存。"
// 向注册错误的计划任务所在通知组发送通知
for _, tag := range notificationTagList {
// 向调度错误的计划任务所包含的所有通知组发送通知
SendNotification(tag, msg, false)
}
notificationMsgMap[tag].WriteString("] 这些任务将无法正常执行,请进入后点重新修改保存。")
SendNotification(tag, notificationMsgMap[tag].String(), false)
}
Cron.Start()
}