-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathmain.go
221 lines (205 loc) · 7.46 KB
/
main.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
package main
//执行方式：
//alter -to=接收人 -agent=应用id -color=消息头部颜色 -corpid=corpid -corpsecret=corpsecret
//alter -to=@all -agent=29481187 -color=FFE61A1A -corpid=dingd123465865 -corpsecret=zC5Jbed9S
//CorpID和CorpSecret可以在钉钉后台找到
import (
"bytes"
"encoding/json"
"encoding/xml"
"flag"
"fmt"
"io/ioutil"
"log"
"net/http"
"net/url"
"strings"
)
type MsgInfo struct {
//消息属性和内容
To, Agentid, Corpid, Corpsecret, Msg, Url, Style string
}
var msgInfo MsgInfo
type Alter struct {
From string `json:"from" xml:"from"`
Time string `json:"time" xml:"time"`
Level string `json:"level" xml:"level"`
Name string `json:"name" xml:"name"`
Key string `json:"key" xml:"key"`
Value string `json:"value" xml:"value"`
Now string `json:"now" xml:"now"`
ID string `json:"id" xml:"id"`
IP string `json:"ip" xml:"ip"`
Color string `json:"color" xml:"color"`
Age string `json:"age" xml:"age"`
Status string `json:"status" xml:"status"`
RecoveryTime string `json:"recoveryTime" xml:"recoveryTime"`
Acknowledgement string `json:"acknowledgement" xml:"acknowledgement"`
Acknowledgementhistory string `json:"acknowledgementhistory" xml:"acknowledgementhistory"`
}
type DingMsg struct {
Touser string `json:"touser"`
Agentid string `json:"agentid"`
Msgtype string `json:"msgtype"`
Oa struct {
MessageURL string `json:"message_url"`
Head struct {
Bgcolor string `json:"bgcolor"`
} `json:"head"`
Body struct {
Title string `json:"title"`
Form [5]struct {
Key string `json:"key"`
Value string `json:"value"`
} `json:"form"`
Rich struct {
Num string `json:"num"`
} `json:"rich"`
Content string `json:"content"`
Author string `json:"author"`
} `json:"body"`
} `json:"oa"`
}
func init() {
// log.SetFlags(log.Lshortfile | log.LstdFlags)
flag.StringVar(&msgInfo.To, "to", "@all", "消息的接收人，可以在钉钉后台查看，可空。")
flag.StringVar(&msgInfo.Agentid, "agentid", "", "AgentID，可以在钉钉后台查看，不可空。")
flag.StringVar(&msgInfo.Corpid, "corpid", "", "CorpID，可以在钉钉后台查看，不可空。")
flag.StringVar(&msgInfo.Corpsecret, "corpsecret", "", "CorpSecret，可以在钉钉后台查看，不可空。")
flag.StringVar(&msgInfo.Msg, "msg", `{ "from": "千思网", "time": "2016.07.28 17:00:05", "level": "Warning", "name": "这是一个千思网（qiansw.com）提供的ZABBIX钉钉报警插件。", "key": "icmpping", "value": "30ms", "now": "56ms", "id": "1637", "ip": "8.8.8.8", "color":"FF4A934A", "age":"3m", "recoveryTime":"2016.07.28 17:03:05", "status":"OK" }`, "Json格式的文本消息内容，不可空。")
flag.StringVar(&msgInfo.Url, "url", "http://www.qiansw.com/golang-zabbix-alter-to-dingding.html", "消息内容点击后跳转到的URL，可空。")
flag.StringVar(&msgInfo.Style, "style", "json", "Msg的格式，可选json和xml，推荐使用xml（支持消息中含双引号），可空。")
flag.Parse()
log.Println("[Init] 初始化完成。")
}
func makeMsg(msg string) string {
// 根据json或xml文本创建消息体
log.Println("[makeMsg] 开始创建消息。")
var alter Alter
if msgInfo.Style == "xml" {
log.Println("[makeMsg] 来源消息格式为XML。")
err := xml.Unmarshal([]byte(msg), &alter)
if err != nil {
log.Fatal(err)
}
} else if msgInfo.Style == "json" {
log.Println("[makeMsg] 来源消息格式为Json。")
err := json.Unmarshal([]byte(msg), &alter)
if err != nil {
log.Fatal(err)
}
} else {
log.Println("[makeMsg] 未指定来源消息格式，默认使用Json解析。")
err := json.Unmarshal([]byte(msg), &alter)
if err != nil {
log.Fatal(err)
}
}
log.Printf("[makeMsg] 来源消息为：%s。\r\n", msg)
var dingMsg DingMsg
//给dingMsg各元素赋值
dingMsg.Touser = msgInfo.To
dingMsg.Agentid = msgInfo.Agentid
dingMsg.Msgtype = "oa"
dingMsg.Oa.MessageURL = msgInfo.Url
dingMsg.Oa.Head.Bgcolor = alter.Color
dingMsg.Oa.Body.Title = alter.Name
dingMsg.Oa.Body.Form[0].Key = "告警级别："
dingMsg.Oa.Body.Form[1].Key = "故障时间："
dingMsg.Oa.Body.Form[2].Key = "故障时长："
dingMsg.Oa.Body.Form[3].Key = "IP地址："
dingMsg.Oa.Body.Form[4].Key = "检测项："
dingMsg.Oa.Body.Form[0].Value = alter.Level
dingMsg.Oa.Body.Form[1].Value = alter.Time
dingMsg.Oa.Body.Form[2].Value = alter.Age
dingMsg.Oa.Body.Form[3].Value = alter.IP
dingMsg.Oa.Body.Form[4].Value = alter.Key
dingMsg.Oa.Body.Rich.Num = alter.Now
if alter.Status == "PROBLEM" {
// 故障处理
dingMsg.Oa.Body.Author = fmt.Sprintf("[%s·%s(%s)]", alter.From, "故障", alter.ID)
if strings.Replace(alter.Acknowledgement, " ", "", -1) == "Yes" {
dingMsg.Oa.Body.Content = "故障已经被确认，" + alter.Acknowledgementhistory
}
} else if alter.Status == "OK" {
// 恢复处理
dingMsg.Oa.Body.Form[0].Key = "故障时间："
dingMsg.Oa.Body.Form[1].Key = "恢复时间："
dingMsg.Oa.Body.Form[0].Value = alter.Time
dingMsg.Oa.Body.Form[1].Value = alter.RecoveryTime
dingMsg.Oa.Body.Author = fmt.Sprintf("[%s·%s(%s)]", alter.From, "恢复", alter.ID)
} else {
// 其他status状况处理
dingMsg.Oa.MessageURL = "http://www.qiansw.com/golang-zabbix-alter-to-dingding.html"
dingMsg.Oa.Body.Content = "ZABBIX动作配置有误，请至千思网[qiansw.com]或直接[点击此消息]查看具体配置文档。"
dingMsg.Oa.Body.Author = fmt.Sprintf("[%s·%s(%s)]", alter.From, alter.Status, alter.ID)
if strings.Replace(alter.Acknowledgement, " ", "", -1) == "Yes" {
dingMsg.Oa.Body.Content = "故障已经被确认，" + alter.Acknowledgementhistory
}
}
// 创建post给钉钉的Json文本
JsonMsg, err := json.Marshal(dingMsg)
if err != nil {
log.Fatal(err)
}
log.Printf("[makeMsg] 消息创建完成：%s\r\n", string(JsonMsg))
return string(JsonMsg)
}
func getToken(corpid, corpsecret string) (token string) { //根据id和secret获取AccessToken
type ResToken struct {
Access_token string
Errcode int
Errmsg string
}
urlstr := fmt.Sprintf("https://oapi.dingtalk.com/gettoken?corpid=%s&corpsecret=%s", corpid, corpsecret)
u, _ := url.Parse(urlstr)
q := u.Query()
u.RawQuery = q.Encode()
res, err := http.Get(u.String())
if err != nil {
log.Fatal(err)
return
}
result, err := ioutil.ReadAll(res.Body)
res.Body.Close()
if err != nil {
log.Fatal(err)
return
}
var m ResToken
err1 := json.Unmarshal(result, &m)
if err1 == nil {
if m.Errcode == 0 {
return m.Access_token
} else {
log.Fatal("AccessToken获取失败，", m.Errmsg)
return
}
return
} else {
log.Fatal("Token解析失败！")
return
}
}
func sendMsg(token, msg string) (status bool) { //发送OA消息，,返回成功或失败
log.Printf("[sendMsg] 需要POST的内容：%s\r\n", msg)
body := bytes.NewBuffer([]byte(msg))
url := fmt.Sprintf("https://oapi.dingtalk.com/message/send?access_token=%s", token)
// fmt.Println(url)
res, err := http.Post(url, "application/json;charset=utf-8", body)
if err != nil {
log.Fatal(err)
return
}
result, err := ioutil.ReadAll(res.Body)
res.Body.Close()
if err != nil {
log.Fatal(err)
return
}
log.Printf("[sendMsg] 钉钉接口返回消息：%s\r\n", result)
return
}
func main() {
sendMsg(getToken(msgInfo.Corpid, msgInfo.Corpsecret), makeMsg(msgInfo.Msg))
}