开发微信报警API

chanyipiaomiao · chanyipiaomiao · commit 0bbc230533b0 · 2018-03-21T15:17:00.000+08:00
diff --git a/common/weixin.go b/common/weixin.go
@@ -0,0 +1,52 @@
+package common
+
+import (
+	"github.com/astaxie/beego"
+	weixin "github.com/chanyipiaomiao/weixin-kit"
+)
+
+var (
+	corpID            = beego.AppConfig.String("weixin::corpID")
+	appSecret         = beego.AppConfig.String("weixin::appSecret")
+	accessTokenAPI    = beego.AppConfig.String("weixin::accessTokenAPI")
+	sendMessageAPIURL = beego.AppConfig.String("weixin::sendMessageAPIURL")
+)
+
+// SendWeixinMessage 发送消息
+func SendWeixinMessage(msgType, text, toTag, toUser, toParty string, agentIDFromReq int64) (bool, error) {
+	var agentID int64
+	if agentIDFromReq == 0 {
+		agentIDFromConf, err := beego.AppConfig.Int64("weixin::agentID")
+		if err != nil {
+			return false, err
+		}
+		agentID = agentIDFromConf
+	} else {
+		agentID = agentIDFromReq
+	}
+
+	message := &weixin.Message{
+		MsgType: msgType, // 目前只支持发送文本消息
+		ToTag:   toTag,   // ToTag 是在企业微信后台定义的标签ID，标签里面可以包含很多人,多个请用|分开
+		ToUser:  toUser,  // ToUser 是企业微信后台看到的用户的ID，多个请用|分开
+		ToParty: toParty, // ToParty 是企业微信后台看到的部门的ID，多个请用|分开
+		AgentID: agentID, // 企业应用的id，整型。可在应用的设置页面查看
+		Safe:    0,       // 表示是否是保密消息，0表示否，1表示是，默认0
+		Text: &weixin.Text{
+			Content: text,
+		},
+	}
+
+	client := &weixin.Client{
+		AccessTokenAPI: accessTokenAPI,
+		APIURL:         sendMessageAPIURL,
+		CorpID:         corpID,
+		CorpSecret:     appSecret,
+		Message:        message,
+	}
+	_, err := client.SendMessage()
+	if err != nil {
+		return false, err
+	}
+	return true, nil
+}
diff --git a/conf/app.conf b/conf/app.conf
@@ -78,5 +78,23 @@ jwtokenSignString = TshTGv9QHZbGLT7nqmqfUavjazrArcN8
 # 生成的二维码图片目录
 qrImageDir = static/download/qr
 
+[weixin]
+# 微信相关配置
+
+# 获取accessTokenAPI
+accessTokenAPI = https://qyapi.weixin.qq.com/cgi-bin/gettoken
+
+# 发送消息API
+sendMessageAPIURL = https://qyapi.weixin.qq.com/cgi-bin/message/send
+
+# corpID 公司的ID
+corpID = wwf6f647f2c4fe1f62
+
+# 应用的密钥
+appSecret = SQEP1EWJXtdcBijE9898V5GGvos13L0fYJ1rFofXnGo
+
+# 报警应用的ID
+agentID = 1000002
+
 include "dev.conf"
 include "prod.conf"
diff --git a/controllers/common.go b/controllers/common.go
@@ -119,3 +119,8 @@ type TwoStepAuthController struct {
 type StorePasswordController struct {
 	BaseController
 }
+
+// WeixinController 发送微信消息管理
+type WeixinController struct {
+	BaseController
+}
diff --git a/controllers/weixin.go b/controllers/weixin.go
@@ -0,0 +1,41 @@
+package controllers
+
+import (
+	"devops-api/common"
+	"fmt"
+)
+
+// SendMessage 发送消息
+func (w *WeixinController) SendMessage() {
+	msgType := w.GetString("msgType")
+	toTag := w.GetString("toTag")
+	toUser := w.GetString("toUser")
+	toParty := w.GetString("toParty")
+	text := w.GetString("text")
+	agentID, err := w.GetInt64("agentID")
+	if err != nil {
+		agentID = 0
+	}
+
+	requestID := w.Data["RequestID"].(string)
+	sendWeixinMessageLog := map[string]interface{}{
+		"entryType": "SendWeixinMessage",
+		"requestId": requestID,
+	}
+	_, err = common.SendWeixinMessage(msgType, text, toTag, toUser, toParty, agentID)
+	if err != nil {
+		sendWeixinMessageLog["statuscode"] = 1
+		sendWeixinMessageLog["errmsg"] = fmt.Sprintf("%s", err)
+		common.GetLogger().Error(sendWeixinMessageLog, "发送微信消息")
+		w.Data["json"] = sendWeixinMessageLog
+		w.ServeJSON()
+		return
+	}
+	sendWeixinMessageLog["statuscode"] = 0
+	sendWeixinMessageLog["errmsg"] = ""
+	sendWeixinMessageLog["result"] = "发送成功"
+	common.GetLogger().Info(sendWeixinMessageLog, text)
+	w.Data["json"] = sendWeixinMessageLog
+	w.ServeJSON()
+	return
+}

Original file line number	Diff line number	Diff line change
`@@ -119,3 +119,8 @@ type TwoStepAuthController struct {`
`119`	`119`	`type StorePasswordController struct {`
`120`	`120`	`BaseController`
`121`	`121`	`}`
	`122`	`+`
	`123`	`+// WeixinController 发送微信消息管理`
	`124`	`+type WeixinController struct {`
	`125`	`+ BaseController`
	`126`	`+}`