如何在Golang中实现Prometheus告警推送

答案:在Golang中实现Prometheus告警推送需通过Alertmanager将告警以JSON格式POST到Golang编写的Web服务,该服务监听/webhook路径,解析WebhookRequest结构体中的告警信息,并可扩展钉钉通知等功能。

在Golang中实现Prometheus告警推送,核心是借助Alertmanager接收来自Prometheus的告警,并通过自定义HTTP服务接收Alertmanager的推送通知。你可以用Golang编写一个Web服务来接收、解析并处理这些告警信息,比如发送到钉钉、企业微信或写入日志。

1. 理解告警推送流程

Prometheus本身不直接发送通知,而是将触发的告警发送给Alertmanager。Alertmanager负责去重、分组和路由,然后调用你配置的Webhook接收端。

你的Golang程序需要:

  • 启动一个HTTP服务监听特定路径(如/webhook
  • 接收Alertmanager POST过来的JSON格式告警数据
  • 解析并处理告警内容(如发消息、记录日志)

2. 编写Golang Webhook接收服务

下面是一个简单的Golang服务示例,用于接收Alertmanager的告警推送:

package main

import ( "encoding/json" "io" "log" "net/http" )

// Alert 表示告警的基本结构 type Alert struct { Status string json:"status" Labels map[string]string json:"labels" Annotations map[string]string json:"annotations" StartsAt string json:"startsAt" EndsAt string json:"endsAt" }

// WebhookRequest 是Alertmanager发送的整体请求结构 type WebhookRequest struct { Receiver string json:"receiver" Status string json:"status" Alerts []Alert json:"alerts" }

func webhookHandler(w http.ResponseWriter, r *http.Request) { if r.Method != http.MethodPost { http.Error(w, "method not allowed", http.StatusMethodNotAllowed) return }

body, err := io.ReadAll(r.Body)
if err != nil {
    http.Error(w, "can't read body", http.StatusBadRequest)
    return
}

var req WebhookRequest
if err := json.Unmarshal(body, &req); err != nil {
    http.Error(w, "invalid JSON", http.StatusBadRequest)
    log.Printf("Unmarshal error: %v\nBody: %s", err, body)
    return
}

// 打印收到的告警
log.Printf("Received alert from receiver: %s, status: %s", req.Receiver, req.Status)
for _, alert := range req.Alerts {
    log.Printf("Alert: status=%s, labels=%v, annotations=%v", 
        alert.Status, alert.Labels, alert.Annotations)

    // 在这里可以添加发送钉钉、企业微信、邮件等逻辑
    sendToDingTalk(alert)
}

w.WriteHeader(http.StatusOK)
w.Write([]byte("OK"))

}

func sendToDingTalk(alert Alert) { // 实现钉钉机器人发送逻辑(使用webhook) // 示例:构造消息并用http.Post发送 // 注意:需配置钉钉机器人的access_token }

启动服务:

func main() {
    http.HandleFunc("/webhook", webhookHandler)
    log.Println("Server starting on :8080")
    log.Fatal(http.ListenAndServe(":8080", nil))
}

3. 配置Alertmanager发送到Golang服务

修改alertmanager.yml,添加一个webhook接收器:

route:
  receiver: 'golang-webhook'

receivers:

  • name: 'golang-webhook' webhook_configs:
    • url: 'https://www./link/ec43638b66e16a5bbede9b710b12b0c7'

重启Alertmanager后,当有告警触发时,它会POST到你的Golang服务。

4. 扩展告警处理逻辑

你可以根据alert.Labels["alertname"]或其它字段判断告警类型,并执行不同操作:

  • 调用钉钉机器人API发送消息
  • 通过SMTP发送邮件
  • 写入数据库或日志系统
  • 触发自动化运维脚本

例如发送钉钉消息:

import "bytes"

func sendToDingTalk(alert Alert) { url := "https://www./link/fddb49982f360e8a94aa8642ed545a16" msg := map[string]interface{}{ "msgtype": "text", "text": map[string]string{ "content": "告警:" + alert.Labels["alertname"] + ",状态:" + alert.Status, }, } payload, := json.Marshal(msg) resp, := http.Post(url, "application/json", bytes.NewBuffer(payload)) resp.Body.Close() }

基本上就这些。不复杂但容易忽略细节,比如Alertmanager的网络可达性、JSON解析容错、服务稳定性等。