73 lines
1.5 KiB
Go
73 lines
1.5 KiB
Go
|
package dao
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"net/http"
|
|||
|
|
"net/url"
|
|||
|
|
"strconv"
|
|||
|
|
"time"
|
|||
|
|
|
|||
|
|
"go-common/library/log"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
const _smsURL = "http://ops-mng.bilibili.co/api/sendsms"
|
|||
|
|
|
|||
|
|
type sms struct {
|
|||
|
|
d *Dao
|
|||
|
|
|
|||
|
|
client *http.Client
|
|||
|
|
lastTime int64
|
|||
|
|
interval int64
|
|||
|
|
params *url.Values
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func newSMS(d *Dao) (s *sms) {
|
|||
|
|
s = &sms{
|
|||
|
|
d: d,
|
|||
|
|
client: &http.Client{},
|
|||
|
|
lastTime: time.Now().Unix() - d.c.SMS.Interval, //如果不想让初始化的时候告警，把减号去掉
|
|||
|
|
interval: d.c.SMS.Interval,
|
|||
|
|
params: &url.Values{
|
|||
|
|
"phone": []string{d.c.SMS.Phone},
|
|||
|
|
"token": []string{d.c.SMS.Token},
|
|||
|
|
},
|
|||
|
|
}
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// SendSMS .
|
|||
|
|
func (d *Dao) SendSMS(msg string) (err error) {
|
|||
|
|
if !d.sms.IntervalCheck() {
|
|||
|
|
log.Error("发短信太频繁啦, msg：%s", msg)
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
if err = d.sms.Send(msg); err != nil {
|
|||
|
|
log.Error("发短信失败, msg：%s, error(%v)", msg, err)
|
|||
|
|
}
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (sms *sms) Send(msg string) (err error) {
|
|||
|
|
var req *http.Request
|
|||
|
|
sms.params.Set("message", msg)
|
|||
|
|
if req, err = http.NewRequest("GET", _smsURL+"?"+sms.params.Encode(), nil); err != nil {
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
req.Header.Set("x1-bilispy-timeout", strconv.FormatInt(int64(time.Duration(1)/time.Millisecond), 10))
|
|||
|
|
if _, err = sms.client.Do(req); err != nil {
|
|||
|
|
log.Error("ops-mng sendsms url(%s) error(%v)", _smsURL+"?"+sms.params.Encode(), err)
|
|||
|
|
}
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// IntervalCheck accessible or not to send msg at present time
|
|||
|
|
func (sms *sms) IntervalCheck() (send bool) {
|
|||
|
|
now := time.Now().Unix()
|
|||
|
|
if (now - sms.lastTime) >= sms.interval {
|
|||
|
|
send = true
|
|||
|
|
sms.lastTime = now
|
|||
|
|
} else {
|
|||
|
|
send = false
|
|||
|
|
}
|
|||
|
|
return
|
|||
|
|
}
|