mirror of https://github.com/ouqiang/gocron
52 lines
1.3 KiB
Go
52 lines
1.3 KiB
Go
package notify
|
|
|
|
import (
|
|
"time"
|
|
"github.com/ouqiang/gocron/modules/logger"
|
|
"fmt"
|
|
)
|
|
|
|
type Message map[string]interface{}
|
|
|
|
type Notifiable interface {
|
|
Send(msg Message)
|
|
}
|
|
|
|
var queue chan Message = make(chan Message, 100)
|
|
|
|
func init() {
|
|
go run()
|
|
}
|
|
|
|
// 把消息推入队列
|
|
func Push(msg Message) {
|
|
queue <- msg
|
|
}
|
|
|
|
func run() {
|
|
for msg := range queue {
|
|
// 根据任务配置发送通知
|
|
taskType, taskTypeOk := msg["task_type"]
|
|
_, taskReceiverIdOk := msg["task_receiver_id"]
|
|
_, nameOk := msg["name"]
|
|
_, outputOk := msg["output"]
|
|
_, statusOk := msg["status"]
|
|
if !taskTypeOk || !taskReceiverIdOk || !nameOk || !outputOk || !statusOk {
|
|
logger.Errorf("#notify#参数不完整#%+v", msg)
|
|
continue
|
|
}
|
|
msg["content"] = fmt.Sprintf("============\n============\n============\n任务名称: %s\n状态: %s\n输出:\n %s\n", msg["name"], msg["status"], msg["output"])
|
|
logger.Debugf("%+v", msg)
|
|
switch(taskType.(int8)) {
|
|
case 1:
|
|
// 邮件
|
|
mail := Mail{}
|
|
go mail.Send(msg)
|
|
case 2:
|
|
// Slack
|
|
slack := Slack{}
|
|
go slack.Send(msg)
|
|
}
|
|
time.Sleep(1 * time.Second)
|
|
}
|
|
} |