121 lines
2.7 KiB
Go
121 lines
2.7 KiB
Go
|
package monitor
|
||
|
|
|
||
|
|
import (
|
||
|
|
"bytes"
|
||
|
|
"context"
|
||
|
|
"crypto/md5"
|
||
|
|
"encoding/json"
|
||
|
|
"fmt"
|
||
|
|
"io"
|
||
|
|
"io/ioutil"
|
||
|
|
"net/http"
|
||
|
|
"net/url"
|
||
|
|
"strconv"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"go-common/app/job/main/creative/conf"
|
||
|
|
"go-common/app/job/main/creative/model"
|
||
|
|
"go-common/library/log"
|
||
|
|
)
|
||
|
|
|
||
|
|
const (
|
||
|
|
_uri = "/api/v1/message/add"
|
||
|
|
_method = "POST"
|
||
|
|
_fileType = "application/json"
|
||
|
|
)
|
||
|
|
|
||
|
|
// Dao is message dao.
|
||
|
|
type Dao struct {
|
||
|
|
c *conf.Config
|
||
|
|
client *http.Client
|
||
|
|
url string
|
||
|
|
}
|
||
|
|
|
||
|
|
// New new a message dao.
|
||
|
|
func New(c *conf.Config) (d *Dao) {
|
||
|
|
d = &Dao{
|
||
|
|
c: c,
|
||
|
|
client: &http.Client{
|
||
|
|
Timeout: time.Duration(time.Second * 1),
|
||
|
|
},
|
||
|
|
url: c.Monitor.Host + _uri,
|
||
|
|
}
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
// Send send exception message to owner.
|
||
|
|
func (d *Dao) Send(c context.Context, username, msg string) (err error) {
|
||
|
|
params := url.Values{}
|
||
|
|
now := time.Now().Unix()
|
||
|
|
params.Set("username", username)
|
||
|
|
params.Set("content", msg)
|
||
|
|
params.Set("title", "test")
|
||
|
|
params.Set("url", "")
|
||
|
|
params.Set("type", "wechat")
|
||
|
|
params.Set("token", d.c.Monitor.AppToken)
|
||
|
|
params.Set("timestamp", strconv.FormatInt(now, 10))
|
||
|
|
bap := &model.BAP{
|
||
|
|
UserName: params.Get("username"),
|
||
|
|
Content: params.Get("content"),
|
||
|
|
Title: params.Get("title"),
|
||
|
|
URL: params.Get("url"),
|
||
|
|
Ty: params.Get("type"),
|
||
|
|
Token: params.Get("token"),
|
||
|
|
TimeStamp: now,
|
||
|
|
Signature: d.getSign(params),
|
||
|
|
}
|
||
|
|
jsonStr, err := json.Marshal(bap)
|
||
|
|
if err != nil {
|
||
|
|
log.Error("monitor json.Marshal error (%v)", err)
|
||
|
|
return
|
||
|
|
}
|
||
|
|
req, err := http.NewRequest(_method, d.url, bytes.NewBuffer(jsonStr))
|
||
|
|
if err != nil {
|
||
|
|
log.Error("monitor http.NewRequest error (%v)", err)
|
||
|
|
return
|
||
|
|
}
|
||
|
|
req.Header.Add("Content-Type", _fileType)
|
||
|
|
// timeout
|
||
|
|
ctx, cancel := context.WithTimeout(c, 800*time.Millisecond)
|
||
|
|
req = req.WithContext(ctx)
|
||
|
|
defer cancel()
|
||
|
|
response, err := d.client.Do(req)
|
||
|
|
if err != nil {
|
||
|
|
log.Error("monitor d.client.Post error(%v)", err)
|
||
|
|
return
|
||
|
|
}
|
||
|
|
defer response.Body.Close()
|
||
|
|
if response.StatusCode != http.StatusOK {
|
||
|
|
log.Error("monitor http.StatusCode nq http.StatusOK (%d) | url(%s)", response.StatusCode, d.url)
|
||
|
|
return
|
||
|
|
}
|
||
|
|
body, err := ioutil.ReadAll(response.Body)
|
||
|
|
if err != nil {
|
||
|
|
log.Error("monitor ioutil.ReadAll error(%v)", err)
|
||
|
|
return
|
||
|
|
}
|
||
|
|
var result struct {
|
||
|
|
Status int `json:"status"`
|
||
|
|
Msg string `json:"msg"`
|
||
|
|
}
|
||
|
|
if err = json.Unmarshal(body, &result); err != nil {
|
||
|
|
log.Error("monitor json.Unmarshal status(%d) error(%v)", result.Status, err)
|
||
|
|
}
|
||
|
|
if result.Status != 0 {
|
||
|
|
log.Error("monitor get status(%d) msg(%s)", result.Status, result.Msg)
|
||
|
|
}
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
func (d *Dao) getSign(params url.Values) (sign string) {
|
||
|
|
for k, v := range params {
|
||
|
|
if len(v) == 0 {
|
||
|
|
params.Del(k)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
h := md5.New()
|
||
|
|
io.WriteString(h, params.Encode()+d.c.Monitor.AppSecret)
|
||
|
|
sign = fmt.Sprintf("%x", h.Sum(nil))
|
||
|
|
return
|
||
|
|
}
|