@ -0,0 +1,79 @@ | |||
package weixinrpc | |||
import ( | |||
"encoding/json" | |||
"strconv" | |||
"time" | |||
"git.tetele.net/tgo/crypter" | |||
"github.com/golang/protobuf/proto" | |||
) | |||
type AccessTokenReq struct { | |||
Appid string | |||
Secret string | |||
} | |||
func GetAccessToken(dbname, site_id string, appid, secret string, url ...string) (*AccessTokenRes, error) { | |||
conn, err := rpc_server_conn(url...) | |||
data["dbname"] = dbname | |||
data["site_id"] = site_id | |||
data["appid"] = appid | |||
data["secret"] = secret | |||
data_json, err := json.Marshal(data) | |||
if err != nil { | |||
return nil, err | |||
} | |||
encryData := crypter.DesEn(string(data_json), DES_KEY) | |||
now_int64 := time.Now().Unix() | |||
now := strconv.FormatInt(now_int64, 10) | |||
sign := Sign(encryData, now) | |||
req := &Request{ | |||
proto.String(encryData), | |||
proto.String(now), | |||
proto.String(sign), | |||
nil} | |||
res := &Response{} | |||
err = conn.GetAccessToken(req, res) | |||
if err != nil { | |||
return nil, err | |||
} | |||
return HandleAccessToken(res) | |||
} | |||
/** | |||
* 处理返回结果 | |||
*/ | |||
func HandleAccessToken(res *Response) (*AccessTokenRes, error) { | |||
//解密 | |||
res_data_de, err := GetOrgData(res) | |||
if err != nil { | |||
return nil, err | |||
} | |||
var res_arr AccessTokenRes | |||
err = json.Unmarshal([]byte(res_data_de), &res_arr) | |||
if err != nil { | |||
return nil, err | |||
} | |||
return &res_arr, nil | |||
} |
@ -0,0 +1,29 @@ | |||
package weixinrpc | |||
const DES_KEY = "wxserrpc" | |||
type WxApiRes struct { | |||
Errcode string | |||
Errmsg string | |||
} | |||
type AccessTokenRes struct { | |||
AccessToken string | |||
ExpiresIn string | |||
WxApiRes | |||
} | |||
func rpc_server_conn(url ...string) (*WeixinRpcServiceClient, error) { | |||
var wx_rpc_url string = "127.0.0.1:7969" | |||
if len(url) > 0 && url[0] != "" { | |||
wx_rpc_url = url[0] | |||
} | |||
conn, _, err := DialWeixinRpcService("tcp", wx_rpc_url) | |||
if err != nil { | |||
return nil, err | |||
} | |||
defer conn.Close() | |||
return conn, nil | |||
} |
@ -0,0 +1,82 @@ | |||
package weixinrpc | |||
import ( | |||
"encoding/json" | |||
"strconv" | |||
"time" | |||
"git.tetele.net/tgo/crypter" | |||
"github.com/golang/protobuf/proto" | |||
) | |||
//公众号模板消息 | |||
type MpTempMsg struct { | |||
Appid string | |||
TemplateId string | |||
Url string | |||
Miniprogram string | |||
Data string | |||
} | |||
type UniformMessageReq struct { | |||
MiniAppid string | |||
Touser string | |||
MpTemplateMsg MpTempMsg | |||
} | |||
/** | |||
* 发送小程序统一服务消息,默认发送公众号消息 | |||
* appid,secret 小程序ID 密钥 | |||
* touser 小程序openid | |||
* mp_appid 关联公众号appid,要求与小程序有绑定且同主体 | |||
* temp_id 公众号模板id | |||
* url 跳转的url | |||
* mini_program 公众号模板消息所要跳转的小程序,小程序的必须与公众号具有绑定关系 | |||
* data 公众号模板消息的数据 | |||
*/ | |||
func SendUniformMessage(dbname, site_id string, appid, secret, touser, mp_appid, temp_id, url, mini_program, data string, url ...string) (*WxApiRes, error) { | |||
conn, err := rpc_server_conn(url...) | |||
data["dbname"] = dbname | |||
data["site_id"] = site_id | |||
data["appid"] = appid | |||
data["secret"] = secret | |||
data["touser"] = touser | |||
data["mp_appid"] = mp_appid | |||
data["temp_id"] = temp_id | |||
data["url"] = url | |||
data["mini_program"] = mini_program | |||
data["data"] = data | |||
data_json, err := json.Marshal(data) | |||
if err != nil { | |||
return nil, err | |||
} | |||
encryData := crypter.DesEn(string(data_json), DES_KEY) | |||
now_int64 := time.Now().Unix() | |||
now := strconv.FormatInt(now_int64, 10) | |||
sign := Sign(encryData, now) | |||
req := &Request{ | |||
proto.String(encryData), | |||
proto.String(now), | |||
proto.String(sign), | |||
nil} | |||
res := &Response{} | |||
err = conn.SendUniformMessage(req, res) | |||
if err != nil { | |||
return nil, err | |||
} | |||
return HandleRes(res) | |||
} |
@ -0,0 +1,101 @@ | |||
package weixinrpc | |||
import ( | |||
"crypto/md5" | |||
"encoding/hex" | |||
"encoding/json" | |||
"errors" | |||
"strconv" | |||
"strings" | |||
"time" | |||
"git.tetele.net/tgo/crypter" | |||
) | |||
/** | |||
* 签名 | |||
*/ | |||
func Sign(data string, salt string) string { | |||
var build strings.Builder | |||
build.WriteString(data) | |||
build.WriteString(salt) | |||
build.WriteString(DES_KEY) | |||
data_str := build.String() | |||
h := md5.New() | |||
h.Write([]byte(data_str)) // 需要加密的字符串 | |||
return hex.EncodeToString(h.Sum(nil)) // 输出加密结果 | |||
} | |||
/** | |||
* 验证签名 | |||
*/ | |||
func CheckSign(sign_str, data, salt string) bool { | |||
sign := Sign(data, salt) | |||
if strings.Compare(sign_str, sign) > -1 { | |||
return true | |||
} | |||
return false | |||
} | |||
/** | |||
* 解密 | |||
*/ | |||
func GetOrgData(res *Response) (string, error) { | |||
res_data := res.GetData() | |||
if res_data == "" { | |||
return nil, errors.New("未收到收据") | |||
} | |||
time_int64, err := strconv.ParseInt(res.GetTime(), 10, 64) | |||
if err != nil { | |||
return nil, err | |||
} | |||
now_int64 := time.Now().Unix() | |||
if now_int64-time_int64 > 10 || time_int64-now_int64 > 10 { | |||
//时间误差前后10秒,返回 | |||
return nil, errors.New("返回时间错误") | |||
} | |||
check_sign := CheckSign(res.GetSign(), res_data, res.GetTime()) | |||
if !check_sign { | |||
return nil, errors.New("返回数据签名错误") | |||
} | |||
//解密 | |||
res_data_de := crypter.DesDe(res_data, DES_KEY) | |||
return res_data_de, nil | |||
} | |||
/** | |||
* 处理返回结果 | |||
*/ | |||
func HandleRes(res *Response) (*WxApiRes, error) { | |||
//解密 | |||
res_data_de, err := GetOrgData(res) | |||
if err != nil { | |||
return nil, err | |||
} | |||
var res_arr WxApiRes | |||
err = json.Unmarshal([]byte(res_data_de), &res_arr) | |||
if err != nil { | |||
return nil, err | |||
} | |||
return &res_arr, nil | |||
} |
@ -0,0 +1,198 @@ | |||
// Code generated by protoc-gen-go. | |||
// source: weixin.proto | |||
// DO NOT EDIT! | |||
/* | |||
Package weixinrpc is a generated protocol buffer package. | |||
It is generated from these files: | |||
weixin.proto | |||
It has these top-level messages: | |||
Request | |||
Response | |||
*/ | |||
package weixinrpc | |||
import proto "github.com/chai2010/protorpc/proto" | |||
import math "math" | |||
import "io" | |||
import "log" | |||
import "net" | |||
import "net/rpc" | |||
import "time" | |||
import protorpc "github.com/chai2010/protorpc" | |||
// Reference imports to suppress errors if they are not otherwise used. | |||
var _ = proto.Marshal | |||
var _ = math.Inf | |||
// 配置信息请求结构 | |||
type Request struct { | |||
Data *string `protobuf:"bytes,1,opt,name=data" json:"data,omitempty"` | |||
Time *string `protobuf:"bytes,2,opt,name=time" json:"time,omitempty"` | |||
Sign *string `protobuf:"bytes,3,opt,name=sign" json:"sign,omitempty"` | |||
XXX_unrecognized []byte `json:"-"` | |||
} | |||
func (m *Request) Reset() { *m = Request{} } | |||
func (m *Request) String() string { return proto.CompactTextString(m) } | |||
func (*Request) ProtoMessage() {} | |||
func (m *Request) GetData() string { | |||
if m != nil && m.Data != nil { | |||
return *m.Data | |||
} | |||
return "" | |||
} | |||
func (m *Request) GetTime() string { | |||
if m != nil && m.Time != nil { | |||
return *m.Time | |||
} | |||
return "" | |||
} | |||
func (m *Request) GetSign() string { | |||
if m != nil && m.Sign != nil { | |||
return *m.Sign | |||
} | |||
return "" | |||
} | |||
// 配置信息响应结构 | |||
type Response struct { | |||
Data *string `protobuf:"bytes,1,opt,name=data" json:"data,omitempty"` | |||
Time *string `protobuf:"bytes,2,opt,name=time" json:"time,omitempty"` | |||
Sign *string `protobuf:"bytes,3,opt,name=sign" json:"sign,omitempty"` | |||
XXX_unrecognized []byte `json:"-"` | |||
} | |||
func (m *Response) Reset() { *m = Response{} } | |||
func (m *Response) String() string { return proto.CompactTextString(m) } | |||
func (*Response) ProtoMessage() {} | |||
func (m *Response) GetData() string { | |||
if m != nil && m.Data != nil { | |||
return *m.Data | |||
} | |||
return "" | |||
} | |||
func (m *Response) GetTime() string { | |||
if m != nil && m.Time != nil { | |||
return *m.Time | |||
} | |||
return "" | |||
} | |||
func (m *Response) GetSign() string { | |||
if m != nil && m.Sign != nil { | |||
return *m.Sign | |||
} | |||
return "" | |||
} | |||
func init() { | |||
} | |||
type WeixinRpcService interface { | |||
GetAccessToken(in *Request, out *Response) error | |||
SendMiniprogramSubscribeMessage(in *Request, out *Response) error | |||
} | |||
// AcceptWeixinRpcServiceClient accepts connections on the listener and serves requests | |||
// for each incoming connection. Accept blocks; the caller typically | |||
// invokes it in a go statement. | |||
func AcceptWeixinRpcServiceClient(lis net.Listener, x WeixinRpcService) { | |||
srv := rpc.NewServer() | |||
if err := srv.RegisterName("WeixinRpcService", x); err != nil { | |||
log.Fatal(err) | |||
} | |||
for { | |||
conn, err := lis.Accept() | |||
if err != nil { | |||
log.Fatalf("lis.Accept(): %v\n", err) | |||
} | |||
go srv.ServeCodec(protorpc.NewServerCodec(conn)) | |||
} | |||
} | |||
// RegisterWeixinRpcService publish the given WeixinRpcService implementation on the server. | |||
func RegisterWeixinRpcService(srv *rpc.Server, x WeixinRpcService) error { | |||
if err := srv.RegisterName("WeixinRpcService", x); err != nil { | |||
return err | |||
} | |||
return nil | |||
} | |||
// NewWeixinRpcServiceServer returns a new WeixinRpcService Server. | |||
func NewWeixinRpcServiceServer(x WeixinRpcService) *rpc.Server { | |||
srv := rpc.NewServer() | |||
if err := srv.RegisterName("WeixinRpcService", x); err != nil { | |||
log.Fatal(err) | |||
} | |||
return srv | |||
} | |||
// ListenAndServeWeixinRpcService listen announces on the local network address laddr | |||
// and serves the given WeixinRpcService implementation. | |||
func ListenAndServeWeixinRpcService(network, addr string, x WeixinRpcService) error { | |||
lis, err := net.Listen(network, addr) | |||
if err != nil { | |||
return err | |||
} | |||
defer lis.Close() | |||
srv := rpc.NewServer() | |||
if err := srv.RegisterName("WeixinRpcService", x); err != nil { | |||
return err | |||
} | |||
for { | |||
conn, err := lis.Accept() | |||
if err != nil { | |||
log.Fatalf("lis.Accept(): %v\n", err) | |||
} | |||
go srv.ServeCodec(protorpc.NewServerCodec(conn)) | |||
} | |||
} | |||
type WeixinRpcServiceClient struct { | |||
*rpc.Client | |||
} | |||
// NewWeixinRpcServiceClient returns a WeixinRpcService rpc.Client and stub to handle | |||
// requests to the set of WeixinRpcService at the other end of the connection. | |||
func NewWeixinRpcServiceClient(conn io.ReadWriteCloser) (*WeixinRpcServiceClient, *rpc.Client) { | |||
c := rpc.NewClientWithCodec(protorpc.NewClientCodec(conn)) | |||
return &WeixinRpcServiceClient{c}, c | |||
} | |||
func (c *WeixinRpcServiceClient) GetAccessToken(in *Request, out *Response) error { | |||
return c.Call("WeixinRpcService.GetAccessToken", in, out) | |||
} | |||
func (c *WeixinRpcServiceClient) SendMiniprogramSubscribeMessage(in *Request, out *Response) error { | |||
return c.Call("WeixinRpcService.SendMiniprogramSubscribeMessage", in, out) | |||
} | |||
// DialWeixinRpcService connects to an WeixinRpcService at the specified network address. | |||
func DialWeixinRpcService(network, addr string) (*WeixinRpcServiceClient, *rpc.Client, error) { | |||
c, err := protorpc.Dial(network, addr) | |||
if err != nil { | |||
return nil, nil, err | |||
} | |||
return &WeixinRpcServiceClient{c}, c, nil | |||
} | |||
// DialWeixinRpcServiceTimeout connects to an WeixinRpcService at the specified network address. | |||
func DialWeixinRpcServiceTimeout(network, addr string, | |||
timeout time.Duration) (*WeixinRpcServiceClient, *rpc.Client, error) { | |||
c, err := protorpc.DialTimeout(network, addr, timeout) | |||
if err != nil { | |||
return nil, nil, err | |||
} | |||
return &WeixinRpcServiceClient{c}, c, nil | |||
} |
@ -0,0 +1,24 @@ | |||
syntax = "proto3"; | |||
package weixinrpc; | |||
// 配置信息请求结构 | |||
message Request { | |||
string data = 1; | |||
string time = 2; | |||
string sign = 3; | |||
} | |||
// 配置信息响应结构 | |||
message Response { | |||
string data = 1; | |||
string time = 2; | |||
string sign = 3; | |||
} | |||
// rpc方法 | |||
service WeixinRpcService { | |||
rpc getAccessToken (Request) returns (Response); // 查询access token | |||
rpc sendMiniprogramSubscribeMessage (Request) returns (Response); // 发送小程序订阅消息 | |||
rpc sendUniformMessage(Request) returns (Response); // 发送统一服务消息 | |||
} |