refactor wechat sender
This commit is contained in:
parent
8455994118
commit
c4ad9f1e88
|
@ -12,8 +12,8 @@ import (
|
||||||
|
|
||||||
"github.com/didi/nightingale/src/common/dataobj"
|
"github.com/didi/nightingale/src/common/dataobj"
|
||||||
"github.com/didi/nightingale/src/modules/rdb/config"
|
"github.com/didi/nightingale/src/modules/rdb/config"
|
||||||
"github.com/didi/nightingale/src/modules/rdb/corp"
|
|
||||||
"github.com/didi/nightingale/src/modules/rdb/redisc"
|
"github.com/didi/nightingale/src/modules/rdb/redisc"
|
||||||
|
"github.com/didi/nightingale/src/modules/rdb/wechat"
|
||||||
)
|
)
|
||||||
|
|
||||||
func ConsumeIm() {
|
func ConsumeIm() {
|
||||||
|
@ -83,13 +83,17 @@ func sendImByWeChat(message *dataobj.Message) {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
client := corp.New(corpID, agentID, secret)
|
client := wechat.New(corpID, agentID, secret)
|
||||||
var err error
|
var err error
|
||||||
for i := 0; i < cnt; i++ {
|
for i := 0; i < cnt; i++ {
|
||||||
err = client.Send(corp.Message{
|
toUser := strings.TrimSpace(message.Tos[i])
|
||||||
ToUser: message.Tos[i],
|
if toUser == "" {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
err = client.Send(wechat.Message{
|
||||||
|
ToUser: toUser,
|
||||||
MsgType: "text",
|
MsgType: "text",
|
||||||
Text: corp.Content{Content: message.Content},
|
Text: wechat.Content{Content: message.Content},
|
||||||
})
|
})
|
||||||
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
|
|
|
@ -1,4 +1,4 @@
|
||||||
package corp
|
package wechat
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"bytes"
|
"bytes"
|
||||||
|
@ -56,11 +56,11 @@ type Message struct {
|
||||||
}
|
}
|
||||||
|
|
||||||
// New 实例化微信企业号应用
|
// New 实例化微信企业号应用
|
||||||
func New(corpID string, agentID int, AgentSecret string) *Client {
|
func New(corpID string, agentID int, agentSecret string) *Client {
|
||||||
c := new(Client)
|
c := new(Client)
|
||||||
c.CorpID = corpID
|
c.CorpID = corpID
|
||||||
c.AgentID = agentID
|
c.AgentID = agentID
|
||||||
c.AgentSecret = AgentSecret
|
c.AgentSecret = agentSecret
|
||||||
return c
|
return c
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in New Issue