gocron/modules/notify/notify.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)
    }
}