wxgzh
parent
179c6a3afa
commit
0df3603ef2
@ -0,0 +1,84 @@
|
|||||||
|
/**
|
||||||
|
* @Author: jager
|
||||||
|
* @Email: lhj168os@gmail.com
|
||||||
|
* @File: main
|
||||||
|
* @Date: 2021/12/20 2:24 下午
|
||||||
|
* @package: ss
|
||||||
|
* @Version: v1.0.0
|
||||||
|
*
|
||||||
|
* @Description:
|
||||||
|
*
|
||||||
|
*/
|
||||||
|
|
||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"github.com/gin-gonic/gin"
|
||||||
|
"github.com/jageros/hawox/contextx"
|
||||||
|
"github.com/jageros/hawox/httpx"
|
||||||
|
"github.com/jageros/hawox/logx"
|
||||||
|
"net/http"
|
||||||
|
"stock/msg"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
func main() {
|
||||||
|
ctx, cancel := contextx.Default()
|
||||||
|
defer cancel()
|
||||||
|
|
||||||
|
logx.Init(logx.DebugLevel, logx.SetCaller(), logx.SetRequest())
|
||||||
|
defer logx.Sync()
|
||||||
|
|
||||||
|
httpx.InitializeHttpServer(ctx, func(engine *gin.Engine) {
|
||||||
|
r := engine.Group("/api")
|
||||||
|
r.GET("/sayhello", func(c *gin.Context) {
|
||||||
|
httpx.PkgMsgWrite(c, map[string]interface{}{"say": "hello world!"})
|
||||||
|
})
|
||||||
|
r.POST("/receive", func(c *gin.Context) {
|
||||||
|
rMsg := &msg.RData{}
|
||||||
|
err := c.BindXML(rMsg)
|
||||||
|
if err != nil {
|
||||||
|
logx.Error(err)
|
||||||
|
} else {
|
||||||
|
logx.Info(rMsg)
|
||||||
|
}
|
||||||
|
|
||||||
|
wMsg := &xml{
|
||||||
|
ToUserName: rMsg.FromUserName,
|
||||||
|
FromUserName: rMsg.ToUserName,
|
||||||
|
CreateTime: int(time.Now().Unix()),
|
||||||
|
MsgType: "text",
|
||||||
|
Content: "收到，谢谢！",
|
||||||
|
}
|
||||||
|
c.XML(http.StatusOK, wMsg)
|
||||||
|
|
||||||
|
//c.XML(http.StatusOK, wMsg)
|
||||||
|
|
||||||
|
//err = msg.Rcv(rMsg.FromUserName)
|
||||||
|
//if err != nil {
|
||||||
|
// logx.Error(err)
|
||||||
|
//}
|
||||||
|
|
||||||
|
//c.String(http.StatusOK, "success")
|
||||||
|
})
|
||||||
|
|
||||||
|
r.GET("/receive", func(c *gin.Context) {
|
||||||
|
echostr := c.Query("echostr")
|
||||||
|
logx.Infof("=== %s ===", echostr)
|
||||||
|
c.String(http.StatusOK, echostr)
|
||||||
|
})
|
||||||
|
}, func(s *httpx.Server) {
|
||||||
|
s.Mode = "debug"
|
||||||
|
s.Port = 8567
|
||||||
|
})
|
||||||
|
err := ctx.Wait()
|
||||||
|
logx.Infof("Stop With: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
type xml struct {
|
||||||
|
ToUserName string `xml:"ToUserName"`
|
||||||
|
FromUserName string `xml:"FromUserName"`
|
||||||
|
CreateTime int `xml:"CreateTime"`
|
||||||
|
MsgType string `xml:"MsgType"`
|
||||||
|
Content string `xml:"Content"`
|
||||||
|
}
|
||||||
@ -0,0 +1,45 @@
|
|||||||
|
/**
|
||||||
|
* @Author: jager
|
||||||
|
* @Email: lhj168os@gmail.com
|
||||||
|
* @File: cache
|
||||||
|
* @Date: 2021/12/20 6:39 下午
|
||||||
|
* @package: user
|
||||||
|
* @Version: v1.0.0
|
||||||
|
*
|
||||||
|
* @Description:
|
||||||
|
*
|
||||||
|
*/
|
||||||
|
|
||||||
|
package user
|
||||||
|
|
||||||
|
import (
|
||||||
|
"github.com/jageros/hawox/attribute"
|
||||||
|
"sync"
|
||||||
|
)
|
||||||
|
|
||||||
|
var users sync.Map
|
||||||
|
|
||||||
|
func LoadAllUserIntoCache() error {
|
||||||
|
attrs, err := attribute.LoadAll("user")
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
for _, attr := range attrs {
|
||||||
|
u := &User{attr: attr}
|
||||||
|
users.Store(attr.GetAttrID(), u)
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func GetUser(openId string) (*User, error) {
|
||||||
|
u, ok := users.Load(openId)
|
||||||
|
if ok {
|
||||||
|
return u.(*User), nil
|
||||||
|
}
|
||||||
|
us, err := newUser(openId)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
users.Store(openId, us)
|
||||||
|
return us, nil
|
||||||
|
}
|
||||||
@ -0,0 +1,114 @@
|
|||||||
|
/**
|
||||||
|
* @Author: jager
|
||||||
|
* @Email: lhj168os@gmail.com
|
||||||
|
* @File: user
|
||||||
|
* @Date: 2021/12/20 4:54 下午
|
||||||
|
* @package: user
|
||||||
|
* @Version: v1.0.0
|
||||||
|
*
|
||||||
|
* @Description:
|
||||||
|
*
|
||||||
|
*/
|
||||||
|
|
||||||
|
package user
|
||||||
|
|
||||||
|
import (
|
||||||
|
"github.com/jageros/hawox/attribute"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
type User struct {
|
||||||
|
attr *attribute.AttrMgr
|
||||||
|
}
|
||||||
|
|
||||||
|
func newUser(openId string) (*User, error) {
|
||||||
|
u := &User{
|
||||||
|
attr: attribute.NewAttrMgr("user", openId),
|
||||||
|
}
|
||||||
|
err := u.attr.Load(true)
|
||||||
|
if err == attribute.NotExistsErr {
|
||||||
|
u.attr.SetInt64("create_time", time.Now().Unix())
|
||||||
|
err = u.attr.Save(false)
|
||||||
|
}
|
||||||
|
return u, err
|
||||||
|
}
|
||||||
|
|
||||||
|
func (u *User) OpenID() string {
|
||||||
|
return u.attr.GetAttrID().(string)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (u *User) Codes(isFund bool) []string {
|
||||||
|
key := "stock"
|
||||||
|
if isFund {
|
||||||
|
key = "fund"
|
||||||
|
}
|
||||||
|
attr := u.attr.GetMapAttr(key)
|
||||||
|
if attr == nil {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
var codes []string
|
||||||
|
attr.ForEachKey(func(key string) bool {
|
||||||
|
codes = append(codes, key)
|
||||||
|
return true
|
||||||
|
})
|
||||||
|
return codes
|
||||||
|
}
|
||||||
|
|
||||||
|
// HasSubscribed 查询用户是否订阅此票
|
||||||
|
func (u *User) HasSubscribed(isFund bool, code string) bool {
|
||||||
|
key := "stock"
|
||||||
|
if isFund {
|
||||||
|
key = "fund"
|
||||||
|
}
|
||||||
|
attr := u.attr.GetMapAttr(key)
|
||||||
|
if attr == nil {
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
sAttr := attr.GetMapAttr(code)
|
||||||
|
if sAttr == nil {
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
return sAttr.GetBool("notify")
|
||||||
|
}
|
||||||
|
|
||||||
|
// Subscribe 订阅股票或基金
|
||||||
|
func (u *User) Subscribe(isFund bool, codes ...string) {
|
||||||
|
key := "stock"
|
||||||
|
if isFund {
|
||||||
|
key = "fund"
|
||||||
|
}
|
||||||
|
attr := u.attr.GetMapAttr(key)
|
||||||
|
if attr == nil {
|
||||||
|
attr = attribute.NewMapAttr()
|
||||||
|
u.attr.SetMapAttr(key, attr)
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, code := range codes {
|
||||||
|
sAttr := attr.GetMapAttr(code)
|
||||||
|
if sAttr == nil {
|
||||||
|
sAttr = attribute.NewMapAttr()
|
||||||
|
attr.SetMapAttr(code, sAttr)
|
||||||
|
}
|
||||||
|
sAttr.SetBool("notify", true)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// UnSubscribe 取消订阅股票或基金
|
||||||
|
func (u *User) UnSubscribe(isFund bool, codes ...string) {
|
||||||
|
key := "stock"
|
||||||
|
if isFund {
|
||||||
|
key = "fund"
|
||||||
|
}
|
||||||
|
attr := u.attr.GetMapAttr(key)
|
||||||
|
if attr == nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, code := range codes {
|
||||||
|
sAttr := attr.GetMapAttr(code)
|
||||||
|
if sAttr == nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
sAttr.SetBool("notify", false)
|
||||||
|
}
|
||||||
|
}
|
||||||
@ -0,0 +1,110 @@
|
|||||||
|
/**
|
||||||
|
* @Author: jager
|
||||||
|
* @Email: lhj168os@gmail.com
|
||||||
|
* @File: wxgzh
|
||||||
|
* @Date: 2021/12/20 6:53 下午
|
||||||
|
* @package: wxgzh
|
||||||
|
* @Version: v1.0.0
|
||||||
|
*
|
||||||
|
* @Description:
|
||||||
|
*
|
||||||
|
*/
|
||||||
|
|
||||||
|
package wxgzh
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"github.com/jageros/hawox/errcode"
|
||||||
|
"github.com/jageros/hawox/httpc"
|
||||||
|
"stock/stock"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
var (
|
||||||
|
appid = "wxba88e64e7342b027"
|
||||||
|
secret = "ab8130a7bf55b78992e3d17f59909e0a"
|
||||||
|
accessTokenUrl = fmt.Sprintf("https://api.weixin.qq.com/cgi-bin/token?grant_type=client_credential&appid=%s&secret=%s", appid, secret)
|
||||||
|
sendUrl = "https://api.weixin.qq.com/cgi-bin/message/template/send?access_token=%s"
|
||||||
|
|
||||||
|
accessToken = ""
|
||||||
|
expiresIn int64
|
||||||
|
)
|
||||||
|
|
||||||
|
type Resp struct {
|
||||||
|
Errcode int `json:"errcode"`
|
||||||
|
Errmsg string `json:"errmsg"`
|
||||||
|
Msgid int `json:"msgid"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type AccessToken struct {
|
||||||
|
AccessToken string `json:"access_token"`
|
||||||
|
ExpiresIn int64 `json:"expires_in"`
|
||||||
|
Errcode int `json:"errcode"`
|
||||||
|
Errmsg string `json:"errmsg"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type RData struct {
|
||||||
|
ToUserName string `xml:"ToUserName"`
|
||||||
|
FromUserName string `xml:"FromUserName"`
|
||||||
|
CreateTime int `xml:"CreateTime"`
|
||||||
|
MsgType string `xml:"MsgType"`
|
||||||
|
Content string `xml:"Content"`
|
||||||
|
MsgID int64 `xml:"MsgId"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type xml struct {
|
||||||
|
ToUserName string `xml:"ToUserName"`
|
||||||
|
FromUserName string `xml:"FromUserName"`
|
||||||
|
CreateTime int `xml:"CreateTime"`
|
||||||
|
MsgType string `xml:"MsgType"`
|
||||||
|
Content string `xml:"Content"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func getAccessToken(update bool) (string, error) {
|
||||||
|
if !update && time.Now().Unix() < expiresIn {
|
||||||
|
return accessToken, nil
|
||||||
|
}
|
||||||
|
resp := &AccessToken{}
|
||||||
|
err := httpc.RequestWithInterface(httpc.GET, accessTokenUrl, httpc.FORM, nil, nil, resp)
|
||||||
|
if err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
if resp.Errcode != 0 {
|
||||||
|
return "", errcode.New(int32(resp.Errcode), resp.Errmsg)
|
||||||
|
}
|
||||||
|
accessToken = resp.AccessToken
|
||||||
|
expiresIn = time.Now().Unix() + resp.ExpiresIn
|
||||||
|
return resp.AccessToken, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func send(openID string, stk stock.IArg, recall bool) error {
|
||||||
|
token, err := getAccessToken(false)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
url := fmt.Sprintf(sendUrl, token)
|
||||||
|
|
||||||
|
arg := stk.Arg(openID)
|
||||||
|
resp := &Resp{}
|
||||||
|
err = httpc.RequestWithInterface(httpc.POST, url, httpc.JSON, arg, nil, resp)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
if resp.Errcode != 0 {
|
||||||
|
if resp.Errcode == 40014 {
|
||||||
|
_, err = getAccessToken(true)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
if recall {
|
||||||
|
return send(openID, stk, false)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return errcode.New(int32(resp.Errcode), resp.Errmsg)
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func Send(openId string, stk stock.IArg) error {
|
||||||
|
return send(openId, stk, true)
|
||||||
|
}
|
||||||
Loading…
Reference in New Issue