新增出错短信提醒

notify/sms.go

@@ -0,0 +1,71 @@
+/************************************************************
+** @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"
+)
+
+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() error {
+	for _, v := range s.Mobiles {
+		s.Param["mobile"] = v
+		err := libs.HttpGet(SmsUrl, s.Param)
+		if err != nil {
+			log.Println(err)
+		}
+	}
+	return nil
+}