61 lines
1.2 KiB
Go
61 lines
1.2 KiB
Go
|
package main
|
||
|
|
|
||
|
|
import (
|
||
|
|
"log"
|
||
|
|
"sync"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"github.com/levigross/grequests"
|
||
|
|
)
|
||
|
|
|
||
|
|
// AlertOver 报警相关
|
||
|
|
type AlertOver struct {
|
||
|
|
Collection sync.Map // k 是title v是时间戳 用于统计上次的发送时间, 发送题目不能重复
|
||
|
|
Setting map[string]string
|
||
|
|
}
|
||
|
|
|
||
|
|
// SetDefaultSetting 设置默认Setting
|
||
|
|
func (ao *AlertOver) SetDefaultSetting() {
|
||
|
|
ao.Setting = map[string]string{
|
||
|
|
"source": "s-577f047d-763a-4f45-a652-475595dc",
|
||
|
|
"receiver": "g-5b4ce1ea-7f6b-422a-9127-a9049c65",
|
||
|
|
"priority": "1",
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (ao *AlertOver) send(data map[string]string) {
|
||
|
|
// 合并初始化设置
|
||
|
|
for k, v := range ao.Setting {
|
||
|
|
data[k] = v
|
||
|
|
}
|
||
|
|
|
||
|
|
resp, err := grequests.Post("https://api.alertover.com/v1/alert",
|
||
|
|
&grequests.RequestOptions{Data: data})
|
||
|
|
|
||
|
|
now := time.Now().Unix()
|
||
|
|
ao.Collection.Store(data["title"], now)
|
||
|
|
|
||
|
|
ErrorLog(err)
|
||
|
|
log.Println(resp.String())
|
||
|
|
}
|
||
|
|
|
||
|
|
// Alert 报警API
|
||
|
|
func (ao *AlertOver) Alert(title, content string) {
|
||
|
|
|
||
|
|
data := map[string]string{
|
||
|
|
"title": title,
|
||
|
|
"content": content,
|
||
|
|
}
|
||
|
|
|
||
|
|
now := time.Now().Unix()
|
||
|
|
if _t, ok := ao.Collection.Load(title); ok {
|
||
|
|
lasttime := _t.(int64)
|
||
|
|
if now-lasttime >= 600 {
|
||
|
|
ao.send(data)
|
||
|
|
}
|
||
|
|
} else {
|
||
|
|
ao.Collection.Store(title, now)
|
||
|
|
ao.send(data)
|
||
|
|
}
|
||
|
|
}
|