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 |
@ -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 | |||
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= |