gocron/modules/notify/slack.go

package notify
// 发送消息到slack

import (
    "fmt"
    "github.com/ouqiang/gocron/modules/httpclient"
    "github.com/ouqiang/gocron/modules/logger"
    "github.com/ouqiang/gocron/modules/utils"
    "strings"
    "github.com/ouqiang/gocron/models"
    "strconv"
    "time"
)

type Slack struct {}

func (slack *Slack) Send(msg Message)  {
    model := new(models.Setting)
    slackSetting, err := model.Slack()
    if err != nil {
        logger.Error("#slack#从数据库获取slack配置失败", err)
        return
    }
    if slackSetting.Url == "" {
        logger.Error("#slack#webhook-url为空")
        return
    }
    if len(slackSetting.Channels) == 0 {
        logger.Error("#slack#channels配置为空")
        return
    }
    logger.Debugf("%+v", slackSetting)
    channels := slack.getActiveSlackChannels(slackSetting, msg)
    logger.Debugf("%+v", channels)
    for _, channel := range(channels) {
        slack.send(msg, slackSetting.Url, channel)
    }
}

func (slack *Slack) send(msg Message, slackUrl string, channel string)  {
    formatBody := slack.format(msg["content"].(string), channel)
    timeout := 30
    maxTimes := 3
    i := 0
    for i < maxTimes {
        resp := httpclient.PostBody(slackUrl, formatBody, timeout)
        if resp.StatusCode == 200 {
            break;
        }
        i += 1
        time.Sleep(2 * time.Second)
        if i < maxTimes {
            logger.Error("slack#发送消息失败#%s#消息内容-%s", resp.Body, msg["content"])
        }
    }
}

func (slack *Slack) getActiveSlackChannels(slackSetting models.Slack, msg Message) []string  {
    taskReceiverIds := strings.Split(msg["task_receiver_id"].(string), ",")
    channels := []string{}
    for _, v := range(slackSetting.Channels) {
        if utils.InStringSlice(taskReceiverIds, strconv.Itoa(v.Id)) {
            channels = append(channels, v.Name)
        }
    }

    return channels
}

// 格式化消息内容
func (slack *Slack) format(content string, channel string) string  {
    content = utils.EscapeJson(content)
    specialChars := []string{"&", "<", ">"}
    replaceChars := []string{"&amp;", "&lt;", "&gt;"}
    content = utils.ReplaceStrings(content, specialChars, replaceChars)

    return fmt.Sprintf(`{"text":"%s","username":"监控", "channel":"%s"}`, content, channel)
}