func combineMail() {
dtos := popAllMailDto()
count := len(dtos)
if count == 0 {
return
}
dtoMap := make(map[string][]*MailDto)
for i := 0; i < count; i++ {
key := fmt.Sprintf("%d%s%s%s", dtos[i].Priority, dtos[i].Status, dtos[i].Email, dtos[i].Metric)
if _, ok := dtoMap[key]; ok {
dtoMap[key] = append(dtoMap[key], dtos[i])
} else {
dtoMap[key] = []*MailDto{dtos[i]}
}
}
// 不要在这处理，继续写回redis，否则重启alarm很容易丢数据
for _, arr := range dtoMap {
size := len(arr)
if size == 1 {
redi.WriteMail([]string{arr[0].Email}, arr[0].Subject, arr[0].Content)
continue
}
subject := fmt.Sprintf("[P%d][%s] %d %s", arr[0].Priority, arr[0].Status, size, arr[0].Metric)
contentArr := make([]string, size)
for i := 0; i < size; i++ {
contentArr[i] = arr[i].Content
}
content := strings.Join(contentArr, "\r\n")
redi.WriteMail([]string{arr[0].Email}, subject, content)
}
}
func HandleCallback(event *model.Event, action *api.Action) {
teams := action.Uic
phones := []string{}
mails := []string{}
if teams != "" {
phones, mails = api.ParseTeams(teams)
smsContent := GenerateSmsContent(event)
mailContent := GenerateMailContent(event)
if action.BeforeCallbackSms == 1 {
redis.WriteSms(phones, smsContent)
}
if action.BeforeCallbackMail == 1 {
redis.WriteMail(mails, smsContent, mailContent)
}
}
message := Callback(event, action)
if teams != "" {
if action.AfterCallbackSms == 1 {
redis.WriteSms(phones, message)
}
if action.AfterCallbackMail == 1 {
redis.WriteMail(mails, message, message)
}
}
}
// 高优先级的不做报警合并
func consumeHighEvents(event *model.Event, action *api.Action) {
if action.Uic == "" {
return
}
phones, mails := api.ParseTeams(action.Uic)
smsContent := GenerateSmsContent(event)
mailContent := GenerateMailContent(event)
if event.Priority() < 3 {
redis.WriteSms(phones, smsContent)
}
redis.WriteMail(mails, smsContent, mailContent)
}