prometheus 自定义 webhook 发送告警到钉钉

prometheus 自定义 webhook 发送告警到钉钉

package main

import (
    "bytes"
    "encoding/json"
    "fmt"
    "log"
    "strings"
    "time"
    "io/ioutil"
    "net/http"
)

const (
    DingDingUrl = "https://oapi.dingtalk.com/robot/send?access_token=access_token"
)

type Text struct {
    Content string `json:"content"`
}

type Msg struct {
    MsgType string `json:"msgtype"`
    Text    Text   `json:"text"`
}

type Alert struct {
    Status      string            `json:"status"`
    Labels      map[string]string `json:"labels"`
    Annotations map[string]string `json:"annotations"`
    StartsAt    time.Time         `json:"startsAt"`
    EndsAt      time.Time         `json:"endsAt"`
}

type Notification struct {
    Version           string            `json:"version"`
    GroupKey          string            `json:"groupKey"`
    Status            string            `json:"status"`
    Receiver          string            `json:"receiver"`
    GroupLabels       map[string]string `json:"groupLabels"`
    CommonLabels      map[string]string `json:"commonLabels"`
    CommonAnnotations map[string]string `json:"commonAnnotations"`
    ExternalURL       string            `json:"externalURL"`
    Alerts            []Alert           `json:"alerts"`
}

func Dingtalk(w http.ResponseWriter, r *http.Request) {
    b, _ := ioutil.ReadAll(r.Body)
    defer r.Body.Close()
    var notification Notification
    log.Println(string(b))
    json.Unmarshal(b, &notification)
    log.Println(notification)
    contents := []string{}
    headers := fmt.Sprintf("group: %s  status:%s", notification.CommonLabels["group"], notification.Status)
    log.Println(headers)
    contents = append(contents, headers)
    for _, each := range notification.Alerts {
        body := fmt.Sprintf("status:%s %s", each.Status, each.Annotations["summary"])
        contents = append(contents, body)
    }
    strings.Join(contents, "\n")
    msg := Msg{
        MsgType: "text",
        Text: Text{
            Content: strings.Join(contents, "\n"),
        },
    }
    msgJson, _ := json.Marshal(msg)
    req, _ := http.NewRequest("POST", DingDingUrl, bytes.NewBuffer(msgJson))
    req.Header.Add("Content-Type", "application/json")
    client := http.Client{}
    res,_ := client.Do(req)
    defer res.Body.Close()
    body, _ := ioutil.ReadAll(res.Body)
    log.Printf("%s\n", body)
    fmt.Fprint(w, "hello world\n")
}

func main() {
    http.HandleFunc("/send", Dingtalk)
    http.ListenAndServe(":8090", nil)
}