Author | SHA1 | Message | Date |
---|---|---|---|
guzeng | 8af968dc22 | 修改readme | 2 years ago |
guzeng | 738d76190a | 增加用户名及编号的RPC请求方式 | 3 years ago |
guzeng | ce09ffce1b | 调用方法增加用户类型 | 3 years ago |
guzeng | b7d20762aa | 增加用户类型 | 3 years ago |
guzeng | 88d36c6ad4 | 取消无用 | 3 years ago |
@ -1,3 +1,3 @@ | |||||
# rpc | |||||
# 此目录将作废 | |||||
rpc | rpc |
@ -0,0 +1,141 @@ | |||||
package pay | |||||
import ( | |||||
"crypto/md5" | |||||
"encoding/hex" | |||||
"encoding/json" | |||||
"errors" | |||||
"strconv" | |||||
"strings" | |||||
"time" | |||||
"git.tetele.net/tgo/crypter" | |||||
"github.com/golang/protobuf/proto" | |||||
) | |||||
type PayToBusinessReq struct { | |||||
SiteId string `json:"site_id"` | |||||
Dbname string `json:"dbname"` | |||||
UserId string `json:"user_id"` | |||||
BusinessId string `json:"business_id"` | |||||
BalanceTypeId string `json:"balance_type"` | |||||
Money float64 `json:"money"` | |||||
Payer string `json:"payer"` | |||||
Memo string `json:"memo"` | |||||
PayType string `json:"pay_type"` | |||||
} | |||||
type PayToBusinessRes struct { | |||||
Id string | |||||
Sn string | |||||
} | |||||
const DES_KEY string = "paypayda" | |||||
/** | |||||
* 向商家支付 | |||||
* 请求及回均加密验签 | |||||
* 2021/02/18 | |||||
* GZ | |||||
*/ | |||||
func Pay(req PayToBusinessReq, url ...string) (*PayToBusinessRes, error) { | |||||
var pay_rpc_url string = "127.0.0.1:7967" | |||||
if len(url) > 0 && url[0] != "" { | |||||
pay_rpc_url = url[0] | |||||
} | |||||
conn, _, err := DialSiteService("tcp", pay_rpc_url) | |||||
if err != nil { | |||||
return nil, err | |||||
} | |||||
defer conn.Close() | |||||
data_json, err := json.Marshal(req) | |||||
if err != nil { | |||||
return nil, err | |||||
} | |||||
now_int64 := time.Now().Unix() | |||||
encryData := crypter.DesEn(string(data_json), DES_KEY) | |||||
now := strconv.FormatInt(now_int64, 10) | |||||
sign := Sign(encryData, now) | |||||
req := &PayRequest{proto.String(encryData), proto.String(now), proto.String(sign), nil} | |||||
res := &PayResponse{} | |||||
err = conn.PayToBusiness(req, res) | |||||
if err != nil { | |||||
return nil, err | |||||
} | |||||
res_data := res.GetData() | |||||
if res_data != "" { | |||||
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) | |||||
var res_arr PayToBusinessRes | |||||
err = json.Unmarshal([]byte(res_data_de), &res_arr) | |||||
if err != nil { | |||||
return nil, err | |||||
} | |||||
return &res_arr, nil | |||||
} | |||||
return nil, nil | |||||
} | |||||
/** | |||||
* 签名 | |||||
*/ | |||||
func Sign(data string, salt string) string { | |||||
var build strings.Builder | |||||
build.WriteString(data) | |||||
build.WriteString(salt) | |||||
build.WriteString("pay66sign33") | |||||
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 | |||||
} |
@ -0,0 +1,198 @@ | |||||
// Code generated by protoc-gen-go. | |||||
// source: pay.proto | |||||
// DO NOT EDIT! | |||||
/* | |||||
Package pay is a generated protocol buffer package. | |||||
It is generated from these files: | |||||
pay.proto | |||||
It has these top-level messages: | |||||
PayRequest | |||||
PayResponse | |||||
*/ | |||||
package pay | |||||
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 PayRequest 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 *PayRequest) Reset() { *m = PayRequest{} } | |||||
func (m *PayRequest) String() string { return proto.CompactTextString(m) } | |||||
func (*PayRequest) ProtoMessage() {} | |||||
func (m *PayRequest) GetData() string { | |||||
if m != nil && m.Data != nil { | |||||
return *m.Data | |||||
} | |||||
return "" | |||||
} | |||||
func (m *PayRequest) GetTime() string { | |||||
if m != nil && m.Time != nil { | |||||
return *m.Time | |||||
} | |||||
return "" | |||||
} | |||||
func (m *PayRequest) GetSign() string { | |||||
if m != nil && m.Sign != nil { | |||||
return *m.Sign | |||||
} | |||||
return "" | |||||
} | |||||
// 配置信息响应结构 | |||||
type PayResponse 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 *PayResponse) Reset() { *m = PayResponse{} } | |||||
func (m *PayResponse) String() string { return proto.CompactTextString(m) } | |||||
func (*PayResponse) ProtoMessage() {} | |||||
func (m *PayResponse) GetData() string { | |||||
if m != nil && m.Data != nil { | |||||
return *m.Data | |||||
} | |||||
return "" | |||||
} | |||||
func (m *PayResponse) GetTime() string { | |||||
if m != nil && m.Time != nil { | |||||
return *m.Time | |||||
} | |||||
return "" | |||||
} | |||||
func (m *PayResponse) GetSign() string { | |||||
if m != nil && m.Sign != nil { | |||||
return *m.Sign | |||||
} | |||||
return "" | |||||
} | |||||
func init() { | |||||
} | |||||
type PayService interface { | |||||
PayToBusiness(in *PayRequest, out *PayResponse) error | |||||
PayToUser(in *PayRequest, out *PayResponse) error | |||||
} | |||||
// AcceptPayServiceClient accepts connections on the listener and serves requests | |||||
// for each incoming connection. Accept blocks; the caller typically | |||||
// invokes it in a go statement. | |||||
func AcceptPayServiceClient(lis net.Listener, x PayService) { | |||||
srv := rpc.NewServer() | |||||
if err := srv.RegisterName("PayService", 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)) | |||||
} | |||||
} | |||||
// RegisterPayService publish the given PayService implementation on the server. | |||||
func RegisterPayService(srv *rpc.Server, x PayService) error { | |||||
if err := srv.RegisterName("PayService", x); err != nil { | |||||
return err | |||||
} | |||||
return nil | |||||
} | |||||
// NewPayServiceServer returns a new PayService Server. | |||||
func NewPayServiceServer(x PayService) *rpc.Server { | |||||
srv := rpc.NewServer() | |||||
if err := srv.RegisterName("PayService", x); err != nil { | |||||
log.Fatal(err) | |||||
} | |||||
return srv | |||||
} | |||||
// ListenAndServePayService listen announces on the local network address laddr | |||||
// and serves the given PayService implementation. | |||||
func ListenAndServePayService(network, addr string, x PayService) error { | |||||
lis, err := net.Listen(network, addr) | |||||
if err != nil { | |||||
return err | |||||
} | |||||
defer lis.Close() | |||||
srv := rpc.NewServer() | |||||
if err := srv.RegisterName("PayService", 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 PayServiceClient struct { | |||||
*rpc.Client | |||||
} | |||||
// NewPayServiceClient returns a PayService rpc.Client and stub to handle | |||||
// requests to the set of PayService at the other end of the connection. | |||||
func NewPayServiceClient(conn io.ReadWriteCloser) (*PayServiceClient, *rpc.Client) { | |||||
c := rpc.NewClientWithCodec(protorpc.NewClientCodec(conn)) | |||||
return &PayServiceClient{c}, c | |||||
} | |||||
func (c *PayServiceClient) PayToBusiness(in *PayRequest, out *PayResponse) error { | |||||
return c.Call("PayService.PayToBusiness", in, out) | |||||
} | |||||
func (c *PayServiceClient) PayToUser(in *PayRequest, out *PayResponse) error { | |||||
return c.Call("PayService.PayToUser", in, out) | |||||
} | |||||
// DialPayService connects to an PayService at the specified network address. | |||||
func DialPayService(network, addr string) (*PayServiceClient, *rpc.Client, error) { | |||||
c, err := protorpc.Dial(network, addr) | |||||
if err != nil { | |||||
return nil, nil, err | |||||
} | |||||
return &PayServiceClient{c}, c, nil | |||||
} | |||||
// DialPayServiceTimeout connects to an PayService at the specified network address. | |||||
func DialPayServiceTimeout(network, addr string, | |||||
timeout time.Duration) (*PayServiceClient, *rpc.Client, error) { | |||||
c, err := protorpc.DialTimeout(network, addr, timeout) | |||||
if err != nil { | |||||
return nil, nil, err | |||||
} | |||||
return &PayServiceClient{c}, c, nil | |||||
} |
@ -0,0 +1,23 @@ | |||||
syntax = "proto3"; | |||||
package pay; | |||||
// 配置信息请求结构 | |||||
message PayRequest { | |||||
string data = 1; | |||||
string time = 2; | |||||
string sign = 3; | |||||
} | |||||
// 配置信息响应结构 | |||||
message PayResponse { | |||||
string data = 1; | |||||
string time = 2; | |||||
string sign = 3; | |||||
} | |||||
// rpc方法 | |||||
service PayService { | |||||
rpc payToBusiness (PayRequest) returns (PayResponse); // 用户各商家支付 | |||||
rpc payToUser (PayRequest) returns (PayResponse); // 商家向用户支付(退款) | |||||
} |
@ -1,109 +1,106 @@ | |||||
package site | package site | ||||
import ( | |||||
"crypto/md5" | |||||
"encoding/hex" | |||||
"encoding/json" | |||||
"errors" | |||||
"strconv" | |||||
"strings" | |||||
"time" | |||||
// import ( | |||||
// "encoding/json" | |||||
// "errors" | |||||
// "strconv" | |||||
// "time" | |||||
"git.tetele.net/tgo/crypter" | |||||
// "git.tetele.net/tgo/crypter" | |||||
"github.com/golang/protobuf/proto" | |||||
) | |||||
// "github.com/golang/protobuf/proto" | |||||
// ) | |||||
type SiteBalanceTypeReqArg struct { | |||||
SiteId string `json:"site_id"` | |||||
Dbname string `json:"dbname"` | |||||
} | |||||
// type SiteBalanceTypeReqArg struct { | |||||
// SiteId string `json:"site_id"` | |||||
// Dbname string `json:"dbname"` | |||||
// } | |||||
type SiteBalanceType struct { | |||||
Id string | |||||
Name string | |||||
UseLimit string | |||||
Unit string | |||||
Type string | |||||
CleanRate string | |||||
CleanTime string | |||||
} | |||||
// type SiteBalanceType struct { | |||||
// Id string | |||||
// Name string | |||||
// UseLimit string | |||||
// Unit string | |||||
// Type string | |||||
// CleanRate string | |||||
// CleanTime string | |||||
// } | |||||
/** | |||||
* 由配置key取对应value | |||||
* 请求及回均加密验签 | |||||
* 2021/01/20 | |||||
* GZ | |||||
*/ | |||||
func GetBalanceType(site_id, dbname string, url ...string) (*[]SiteBalanceType, error) { | |||||
// /** | |||||
// * 由配置key取对应value | |||||
// * 请求及回均加密验签 | |||||
// * 2021/01/20 | |||||
// * GZ | |||||
// */ | |||||
// func GetBalanceType(site_id, dbname string, url ...string) (*[]SiteBalanceType, error) { | |||||
var site_rpc_url string = "127.0.0.1:7971" | |||||
if len(url) > 0 && url[0] != "" { | |||||
site_rpc_url = url[0] | |||||
} | |||||
conn, _, err := DialSiteService("tcp", site_rpc_url) | |||||
if err != nil { | |||||
return nil, err | |||||
} | |||||
defer conn.Close() | |||||
// var site_rpc_url string = "127.0.0.1:7971" | |||||
// if len(url) > 0 && url[0] != "" { | |||||
// site_rpc_url = url[0] | |||||
// } | |||||
// conn, _, err := DialSiteService("tcp", site_rpc_url) | |||||
// if err != nil { | |||||
// return nil, err | |||||
// } | |||||
// defer conn.Close() | |||||
arg := SiteConfigItemReqArg{site_id, dbname, key} | |||||
// arg := SiteConfigItemReqArg{site_id, dbname, key} | |||||
data_json, err := json.Marshal(arg) | |||||
if err != nil { | |||||
return nil, err | |||||
} | |||||
now_int64 := time.Now().Unix() | |||||
// data_json, err := json.Marshal(arg) | |||||
// if err != nil { | |||||
// return nil, err | |||||
// } | |||||
// now_int64 := time.Now().Unix() | |||||
encryData := crypter.DesEn(string(data_json), "confdata") | |||||
// encryData := crypter.DesEn(string(data_json), "confdata") | |||||
now := strconv.FormatInt(now_int64, 10) | |||||
// now := strconv.FormatInt(now_int64, 10) | |||||
sign := Sign(encryData, now) | |||||
// sign := Sign(encryData, now) | |||||
req := &ConfigRequest{proto.String(encryData), proto.String(now), proto.String(sign), nil} | |||||
// req := &ConfigRequest{proto.String(encryData), proto.String(now), proto.String(sign), nil} | |||||
res := &ConfigResponse{} | |||||
// res := &ConfigResponse{} | |||||
err = conn.GetConfig(req, res) | |||||
// err = conn.GetConfig(req, res) | |||||
if err != nil { | |||||
return nil, err | |||||
} | |||||
// if err != nil { | |||||
// return nil, err | |||||
// } | |||||
res_data := res.GetData() | |||||
// res_data := res.GetData() | |||||
if res_data != "" { | |||||
// if res_data != "" { | |||||
time_int64, err := strconv.ParseInt(res.GetTime(), 10, 64) | |||||
if err != nil { | |||||
return nil, err | |||||
} | |||||
// time_int64, err := strconv.ParseInt(res.GetTime(), 10, 64) | |||||
// if err != nil { | |||||
// return nil, err | |||||
// } | |||||
now_int64 = time.Now().Unix() | |||||
// now_int64 = time.Now().Unix() | |||||
if now_int64-time_int64 > 10 || time_int64-now_int64 > 10 { | |||||
//时间误差前后10秒,返回 | |||||
return nil, errors.New("返回时间错误") | |||||
} | |||||
// 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("返回数据签名错误") | |||||
} | |||||
// check_sign := CheckSign(res.GetSign(), res_data, res.GetTime()) | |||||
// if !check_sign { | |||||
// return nil, errors.New("返回数据签名错误") | |||||
// } | |||||
//解密 | |||||
res_data_de := crypter.DesDe(res_data, "confdata") | |||||
// //解密 | |||||
// res_data_de := crypter.DesDe(res_data, "confdata") | |||||
var res_arr []SiteBalanceType | |||||
// var res_arr []SiteBalanceType | |||||
err = json.Unmarshal([]byte(res_data_de), &res_arr) | |||||
// err = json.Unmarshal([]byte(res_data_de), &res_arr) | |||||
if err != nil { | |||||
return nil, err | |||||
} | |||||
return &res_arr, nil | |||||
} | |||||
// if err != nil { | |||||
// return nil, err | |||||
// } | |||||
// return &res_arr, nil | |||||
// } | |||||
return nil, nil | |||||
} | |||||
// return nil, nil | |||||
// } |
@ -0,0 +1,8 @@ | |||||
module git.tetele.net/tgo/rpc/user | |||||
go 1.14 | |||||
require ( | |||||
github.com/chai2010/protorpc v1.0.0 // indirect | |||||
golang.org/x/sync v0.0.0-20210220032951-036812b2e83c // indirect | |||||
) |
@ -0,0 +1,8 @@ | |||||
github.com/chai2010/protorpc v1.0.0 h1:aJ45G9sl1utSKo35EqnBSTs5jqTpdJDJAuZMMYPAtFo= | |||||
github.com/chai2010/protorpc v1.0.0/go.mod h1:woR3WwjaQDqFjlzdVsFEKiK5Ur12QL8mYxVPjfr5z54= | |||||
github.com/golang/protobuf v1.0.0 h1:lsek0oXi8iFE9L+EXARyHIjU5rlWIhhTkjDz3vHhWWQ= | |||||
github.com/golang/protobuf v1.0.0/go.mod h1:6lQm79b+lXiMfvg/cZm0SGofjICqVBUtrP5yJMmIC1U= | |||||
github.com/golang/snappy v0.0.0-20170215233205-553a64147049 h1:K9KHZbXKpGydfDN0aZrsoHpLJlZsBrGMFWbgLDGnPZk= | |||||
github.com/golang/snappy v0.0.0-20170215233205-553a64147049/go.mod h1:/XxbfmMg8lxefKM7IXC3fBNl/7bRcc72aCRzEWrmP2Q= | |||||
golang.org/x/sync v0.0.0-20210220032951-036812b2e83c h1:5KslGYwFpkhGh+Q16bwMP3cOontH8FOep7tGV86Y7SQ= | |||||
golang.org/x/sync v0.0.0-20210220032951-036812b2e83c/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM= |