97 lines
1.7 KiB
Go
97 lines
1.7 KiB
Go
/************************************************************
|
|
** @Description: notify
|
|
** @Author: george hao
|
|
** @Date: 2018-08-09 13:05
|
|
** @Last Modified by: george hao
|
|
** @Last Modified time: 2018-08-09 13:05
|
|
*************************************************************/
|
|
package notify
|
|
|
|
import (
|
|
"github.com/astaxie/beego"
|
|
"github.com/george518/PPGo_Job/libs"
|
|
"log"
|
|
"time"
|
|
"encoding/json"
|
|
"github.com/pkg/errors"
|
|
)
|
|
|
|
type AjaxReturn struct {
|
|
Status int `json:"status"`
|
|
Message string `json:"message"`
|
|
Data interface{} `json:"data"`
|
|
}
|
|
|
|
type Sms struct {
|
|
Mobiles []string
|
|
Param map[string]string
|
|
}
|
|
|
|
var SmsChan chan *Sms
|
|
var SmsUrl string
|
|
|
|
func init() {
|
|
SmsUrl = beego.AppConfig.String("msg.url")
|
|
poolSize, _ := beego.AppConfig.Int("msg.pool")
|
|
|
|
//创建通道
|
|
SmsChan = make(chan *Sms, poolSize)
|
|
|
|
go func() {
|
|
for {
|
|
select {
|
|
case m, ok := <-SmsChan:
|
|
if !ok {
|
|
return
|
|
}
|
|
if _, err := m.SendSms(); err != nil {
|
|
beego.Error("SendSms:", err.Error())
|
|
}
|
|
}
|
|
}
|
|
}()
|
|
|
|
}
|
|
|
|
func SendSmsToChan(mobiles []string, param map[string]string) bool {
|
|
sms := &Sms{
|
|
Mobiles: mobiles,
|
|
Param: param,
|
|
}
|
|
|
|
select {
|
|
case SmsChan <- sms:
|
|
return true
|
|
case <-time.After(time.Second * 3):
|
|
return false
|
|
}
|
|
}
|
|
|
|
func (s *Sms) SendSms() (string, error) {
|
|
|
|
for _, v := range s.Mobiles {
|
|
s.Param["mobile"] = v
|
|
res, err := libs.HttpGet(SmsUrl, s.Param)
|
|
|
|
if err != nil {
|
|
log.Println(err)
|
|
return "", err
|
|
}
|
|
|
|
ajaxData := AjaxReturn{}
|
|
jsonErr := json.Unmarshal([]byte(res), &ajaxData)
|
|
|
|
if jsonErr != nil {
|
|
return "", jsonErr
|
|
}
|
|
|
|
if ajaxData.Status != 200 {
|
|
return "", errors.Errorf("msg %s", ajaxData.Message)
|
|
}
|
|
|
|
return res, nil
|
|
|
|
}
|
|
return "", nil
|
|
}
|