refactor: Wrapper metrics

lingcoder · lingcoder · commit bd1ad44d663d · 2023-10-23T07:23:56.000+08:00
diff --git a/README.md b/README.md
@@ -8,4 +8,16 @@
 
 ### 已支持的通知方式
 
-- [x] feishu 飞书群组机器人
+- [x] feishu 飞书群机器人
+- [ ] feishu_app 飞书应用机器人
+- [ ] dingtalk 钉钉
+- [ ] wechat 企业微信
+- [ ] email 邮箱
+- [ ] telegram 电报
+
+
+### 快速启动
+
+```bash
+docker run -d -p 8080:8080 lingcoder/prometheus-notifier:latest
+```
diff --git a/api/v1/base.go b/api/v1/base.go
@@ -9,8 +9,8 @@ type BaseReq struct {
 	URL  string `json:"url"`
 }
 
-// Message defines the JSON object send to webhook endpoints.
-type Message struct {
+// WebhookBody defines the JSON object send to webhook endpoints.
+type WebhookBody struct {
 	*Data
 
 	// The protocol version.
diff --git a/api/v1/custom.go b/api/v1/custom.go
@@ -2,5 +2,5 @@ package v1
 
 type CustomReq struct {
 	BaseReq
-	Body Message
+	Body WebhookBody
 }
diff --git a/api/v1/feishu.go b/api/v1/feishu.go
@@ -2,5 +2,5 @@ package v1
 
 type FeishuReq struct {
 	BaseReq
-	Body Message
+	WebhookBody
 }
diff --git a/config/config.yaml b/config/config.yaml
@@ -1,5 +1,5 @@
 server:
-  address:     ":8082"
+  address:     ":8080"
 #  openapiPath: "/api.json"
 #  swaggerPath: "/swagger"
 
diff --git a/docs/feishu.md b/docs/feishu.md
@@ -0,0 +1,15 @@
+```yml
+route:
+group_by: [ ... ]
+group_wait: 60s
+group_interval: 5m
+repeat_interval: 1h
+receiver: 'webhook'
+
+receivers:
+  - name: 'webhook'
+    webhook_configs:
+      - url: 'http://IP:8080/notifier/feishu?lang=zh-CN&url=https://open.feishu.cn/open-apis/bot/v2/hook/{your-token}'
+        send_resolved: true
+
+```
diff --git a/i18n/en/alert_template.json b/i18n/en/alert_template.json
diff --git a/i18n/en/alert_template.toml b/i18n/en/alert_template.toml
@@ -0,0 +1,11 @@
+alert-notification = "Alert Notification"
+alert-level-status = "Level Status"
+alert-rule-name = "Rule Name"
+alert-rule-notes = "Rule Notes"
+alert-monitoring-metrics = "Monitoring Metrics"
+alert-trigger-time = "Trigger Time"
+alert-trigger-value = "Trigger Value"
+alert-sending-time = "Sending Time"
+alert-recovery-time = "Recovery Time"
+alert-status-triggered = "Triggered"
+alert-status-recovered = "Recovered"
diff --git a/i18n/zh-CN/alert_template.json b/i18n/zh-CN/alert_template.json
diff --git a/i18n/zh-CN/alert_template.toml b/i18n/zh-CN/alert_template.toml
@@ -0,0 +1,11 @@
+alert-notification = "告警通知"
+alert-level-status = "级别状态"
+alert-rule-name = "规则名称"
+alert-rule-notes = "规则备注"
+alert-monitoring-metrics = "监控指标"
+alert-trigger-time = "触发时间"
+alert-trigger-value = "触发时值"
+alert-sending-time = "发送时间"
+alert-recovery-time = "恢复时间"
+alert-status-triggered = "已触发"
+alert-status-recovered = "已恢复"
diff --git a/i18n/zh-TW/alert_template.json b/i18n/zh-TW/alert_template.json
diff --git a/i18n/zh-TW/alert_template.toml b/i18n/zh-TW/alert_template.toml
@@ -0,0 +1,11 @@
+alert-notification = "警報通知"
+alert-level-status = "等級狀態"
+alert-rule-name = "規則名稱"
+alert-rule-notes = "規則備註"
+alert-monitoring-metrics = "監控指標"
+alert-trigger-time = "觸發時間"
+alert-trigger-value = "觸發時值"
+alert-sending-time = "發送時間"
+alert-recovery-time = "恢復時間"
+alert-status-triggered = "已觸發"
+alert-status-recovered = "已恢復"
diff --git a/internal/consts/alert_template.go b/internal/consts/alert_template.go
@@ -0,0 +1,10 @@
+package consts
+
+const DefaultFeishuTemplate = `
+**{#alert-level-status}**: {{ .Severity }} {{ if .IsRecovered }}{#alert-status-recovered}{{else }}{#alert-status-triggered}{{ end }}
+**{#alert-rule-name}**: {{ .RuleName }}
+{{ if .RuleNotes }}**{#alert-rule-notes}**: {{ .RuleNotes }}{{end }}
+**{#alert-monitoring-metrics}**: {{.TagsJSON }}
+{{ if .IsRecovered }}**{#alert-recovery-time}**: {{ .LastEvalTime }}{{ else }}**{#alert-trigger-time}**: {{  .TriggerTime }}
+**{#alert-trigger-value}**: {{ .TriggerValue }}{{ end }}
+**{#alert-sending-time}**: {{ .SendingTime }}`
diff --git a/internal/logic/feishu/feishu.go b/internal/logic/feishu/feishu.go
@@ -2,11 +2,16 @@ package feishu
 
 import (
 	"context"
+	"fmt"
+	"github.com/gogf/gf/v2/encoding/gjson"
 	"github.com/gogf/gf/v2/frame/g"
 	"github.com/gogf/gf/v2/util/gconv"
 	larkcard "github.com/larksuite/oapi-sdk-go/v3/card"
 	v1 "github.com/lingcoder/prometheus-notifier/api/v1"
+	"github.com/lingcoder/prometheus-notifier/internal/consts"
+	"github.com/lingcoder/prometheus-notifier/internal/model"
 	"github.com/lingcoder/prometheus-notifier/internal/service"
+	"time"
 )
 
 type sFeishu struct {
@@ -26,25 +31,58 @@ func (u *sFeishu) init(ctx context.Context) {
 
 func (u *sFeishu) Send(ctx context.Context, req *v1.FeishuReq) error {
 
-	// 根据语言选择合适的模板
-	template := g.I18n().T(ctx, "feishu-template")
+	// 重新解析模板
+	alerts := req.WebhookBody.Alerts
+	webhookAlert := model.WebhookAlert{
+		IsRecovered: req.WebhookBody.Status == "resolved",
+		SendingTime: time.Now().Format(time.RFC3339),
+	}
 
-	// 解析模板
-	content, err := g.View().ParseContent(ctx, template, gconv.Map(req.Body))
-	// 创建卡片配置，设置宽度和固定显示
-
-	card := larkcard.NewMessageCard().
-		Config(larkcard.NewMessageCardConfig().
-			WideScreenMode(true).EnableForward(true)).
-		Header(larkcard.NewMessageCardHeader().
-			Title(larkcard.NewMessageCardPlainText().Content(g.I18n().T(ctx, "feishu-title")))).
-		Elements([]larkcard.MessageCardElement{
-			larkcard.NewMessageCardDiv().Text(larkcard.NewMessageCardPlainText().Content(content)),
+	encodeString := gjson.MustEncodeString(req)
+	fmt.Println(encodeString)
+
+	for i := 0; i < len(alerts); i++ {
+		alert := alerts[i]
+
+		webhookAlert.Severity = alert.Labels["severity"]
+		webhookAlert.RuleName = alert.Labels["alertname"]
+		webhookAlert.RuleNotes = alert.Annotations["summary"]
+		webhookAlert.TriggerTime = alert.StartsAt.Format(time.RFC3339)
+		webhookAlert.LastEvalTime = alert.EndsAt.Format(time.RFC3339)
+		webhookAlert.TriggerValue = alert.Annotations["value"]
+		webhookAlert.TagsJSON = alert.Labels["tags"]
+
+		content, err := g.View().ParseContent(ctx, consts.DefaultFeishuTemplate, gconv.Map(webhookAlert))
+
+		if err != nil {
+
+		}
+		// 创建卡片配置，设置宽度和固定显示
+
+		card := larkcard.NewMessageCard().
+			Config(larkcard.NewMessageCardConfig().
+				WideScreenMode(true).EnableForward(true)).
+			Header(larkcard.NewMessageCardHeader().
+				Template(func() string {
+					if webhookAlert.IsRecovered {
+						return larkcard.TemplateGreen
+					} else {
+						return larkcard.TemplateRed
+					}
+				}()).
+				Title(larkcard.NewMessageCardPlainText().Content(g.I18n().T(ctx, "alert-notification")))).
+			Elements([]larkcard.MessageCardElement{
+				larkcard.NewMessageCardDiv().Text(larkcard.NewMessageCardLarkMd().Content(content)),
+			})
+
+		_, _ = g.Client().Post(ctx, req.URL, g.Map{
+			"msg_type": "interactive",
+			"card":     card,
 		})
 
-	_, err = g.Client().Post(ctx, req.URL, g.Map{
-		"msg_type": "interactive",
-		"card":     card,
-	})
-	return err
+	}
+
+	// 解析模板
+
+	return nil
 }
diff --git a/internal/model/.gitkeep b/internal/model/.gitkeep
diff --git a/internal/model/template.go b/internal/model/template.go
@@ -0,0 +1,13 @@
+package model
+
+type WebhookAlert struct {
+	Severity     string
+	IsRecovered  bool
+	RuleName     string
+	RuleNotes    string
+	TagsJSON     string
+	LastEvalTime string
+	TriggerTime  string
+	TriggerValue string
+	SendingTime  string
+}
diff --git a/main.go b/main.go
@@ -1,13 +1,10 @@
 package main
 
 import (
-	_ "github.com/lingcoder/prometheus-notifier/internal/packed"
-
-	_ "github.com/lingcoder/prometheus-notifier/internal/logic"
-
 	"github.com/gogf/gf/v2/os/gctx"
-
 	"github.com/lingcoder/prometheus-notifier/internal/cmd"
+	_ "github.com/lingcoder/prometheus-notifier/internal/logic"
+	_ "github.com/lingcoder/prometheus-notifier/internal/packed"
 )
 
 func main() {

Original file line number	Diff line number	Diff line change
`@@ -9,8 +9,8 @@ type BaseReq struct {`
`9`	`9`	URL string `json:"url"`
`10`	`10`	`}`
`11`	`11`
`12`		`-// Message defines the JSON object send to webhook endpoints.`
`13`		`-type Message struct {`
	`12`	`+// WebhookBody defines the JSON object send to webhook endpoints.`
	`13`	`+type WebhookBody struct {`
`14`	`14`	`*Data`
`15`	`15`
`16`	`16`	`// The protocol version.`
Original file line number	Diff line number	Diff line change
`@@ -2,5 +2,5 @@ package v1`
`2`	`2`
`3`	`3`	`type CustomReq struct {`
`4`	`4`	`BaseReq`
`5`		`- Body Message`
	`5`	`+ Body WebhookBody`
`6`	`6`	`}`
Original file line number	Diff line number	Diff line change
`@@ -2,5 +2,5 @@ package v1`
`2`	`2`
`3`	`3`	`type FeishuReq struct {`
`4`	`4`	`BaseReq`
`5`		`- Body Message`
	`5`	`+ WebhookBody`
`6`	`6`	`}`