From d1e0308322545f99ea06a32499befb3e05561e7d Mon Sep 17 00:00:00 2001 From: guzeng Date: Sat, 13 Mar 2021 11:41:39 +0800 Subject: [PATCH] init --- client.balance.go | 106 +++++++++++++++++++++++++ client.go | 135 +++++++++++++++++++++++++++++++ client_test.go | 15 ++++ site.pb.go | 198 ++++++++++++++++++++++++++++++++++++++++++++++ site.proto | 23 ++++++ 5 files changed, 477 insertions(+) create mode 100644 client.balance.go create mode 100644 client.go create mode 100644 client_test.go create mode 100644 site.pb.go create mode 100644 site.proto diff --git a/client.balance.go b/client.balance.go new file mode 100644 index 0000000..d91a3da --- /dev/null +++ b/client.balance.go @@ -0,0 +1,106 @@ +package site + +// import ( +// "encoding/json" +// "errors" +// "strconv" +// "time" + +// "git.tetele.net/tgo/crypter" + +// "github.com/golang/protobuf/proto" +// ) + +// 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 +// } + +// /** +// * 由配置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() + +// arg := SiteConfigItemReqArg{site_id, dbname, key} + +// data_json, err := json.Marshal(arg) +// if err != nil { +// return nil, err +// } +// now_int64 := time.Now().Unix() + +// encryData := crypter.DesEn(string(data_json), "confdata") + +// now := strconv.FormatInt(now_int64, 10) + +// sign := Sign(encryData, now) + +// req := &ConfigRequest{proto.String(encryData), proto.String(now), proto.String(sign), nil} + +// res := &ConfigResponse{} + +// err = conn.GetConfig(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, "confdata") + +// var res_arr []SiteBalanceType + +// err = json.Unmarshal([]byte(res_data_de), &res_arr) + +// if err != nil { +// return nil, err +// } +// return &res_arr, nil +// } + +// return nil, nil +// } diff --git a/client.go b/client.go new file mode 100644 index 0000000..79dbd3f --- /dev/null +++ b/client.go @@ -0,0 +1,135 @@ +package site + +import ( + "crypto/md5" + "encoding/hex" + "encoding/json" + "errors" + "strconv" + "strings" + "time" + + "git.tetele.net/tgo/crypter" + + "github.com/golang/protobuf/proto" +) + +type SiteConfigItemReqArg struct { + SiteId string `json:"site_id"` + Dbname string `json:"dbname"` + Key string `json:"key"` +} + +type SiteConfigItemRes struct { + Type string + Value string +} + +/** + * 由配置key取对应value + * 请求及回均加密验签 + * 2021/01/20 + * GZ + */ +func GetConfigItem(site_id, dbname, key string, url ...string) (*SiteConfigItemRes, 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() + + arg := SiteConfigItemReqArg{site_id, dbname, key} + + data_json, err := json.Marshal(arg) + if err != nil { + return nil, err + } + now_int64 := time.Now().Unix() + + encryData := crypter.DesEn(string(data_json), "confdata") + + now := strconv.FormatInt(now_int64, 10) + + sign := Sign(encryData, now) + + req := &ConfigRequest{proto.String(encryData), proto.String(now), proto.String(sign), nil} + + res := &ConfigResponse{} + + err = conn.GetConfig(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, "confdata") + + var res_arr SiteConfigItemRes + + 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("site55sign33") + + 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 +} diff --git a/client_test.go b/client_test.go new file mode 100644 index 0000000..a20b19d --- /dev/null +++ b/client_test.go @@ -0,0 +1,15 @@ +package site + +import ( + "testing" +) + +func Test_GetConfigItem(t *testing.T) { + siteid := "100065" + dbname := "test1_tetele_com" + key := "percentage" + res, err := GetConfigItem(siteid, dbname, key) + t.Log(res) + t.Log(err) + +} diff --git a/site.pb.go b/site.pb.go new file mode 100644 index 0000000..24f2d3f --- /dev/null +++ b/site.pb.go @@ -0,0 +1,198 @@ +// Code generated by protoc-gen-go. +// source: site.proto +// DO NOT EDIT! + +/* +Package site is a generated protocol buffer package. + +It is generated from these files: + site.proto + +It has these top-level messages: + ConfigRequest + ConfigResponse +*/ +package site + +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 ConfigRequest 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 *ConfigRequest) Reset() { *m = ConfigRequest{} } +func (m *ConfigRequest) String() string { return proto.CompactTextString(m) } +func (*ConfigRequest) ProtoMessage() {} + +func (m *ConfigRequest) GetData() string { + if m != nil && m.Data != nil { + return *m.Data + } + return "" +} + +func (m *ConfigRequest) GetTime() string { + if m != nil && m.Time != nil { + return *m.Time + } + return "" +} + +func (m *ConfigRequest) GetSign() string { + if m != nil && m.Sign != nil { + return *m.Sign + } + return "" +} + +// 配置信息响应结构 +type ConfigResponse 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 *ConfigResponse) Reset() { *m = ConfigResponse{} } +func (m *ConfigResponse) String() string { return proto.CompactTextString(m) } +func (*ConfigResponse) ProtoMessage() {} + +func (m *ConfigResponse) GetData() string { + if m != nil && m.Data != nil { + return *m.Data + } + return "" +} + +func (m *ConfigResponse) GetTime() string { + if m != nil && m.Time != nil { + return *m.Time + } + return "" +} + +func (m *ConfigResponse) GetSign() string { + if m != nil && m.Sign != nil { + return *m.Sign + } + return "" +} + +func init() { +} + +type SiteService interface { + GetConfig(in *ConfigRequest, out *ConfigResponse) error + GetBalanceType(in *ConfigRequest, out *ConfigResponse) error +} + +// AcceptSiteServiceClient accepts connections on the listener and serves requests +// for each incoming connection. Accept blocks; the caller typically +// invokes it in a go statement. +func AcceptSiteServiceClient(lis net.Listener, x SiteService) { + srv := rpc.NewServer() + if err := srv.RegisterName("SiteService", 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)) + } +} + +// RegisterSiteService publish the given SiteService implementation on the server. +func RegisterSiteService(srv *rpc.Server, x SiteService) error { + if err := srv.RegisterName("SiteService", x); err != nil { + return err + } + return nil +} + +// NewSiteServiceServer returns a new SiteService Server. +func NewSiteServiceServer(x SiteService) *rpc.Server { + srv := rpc.NewServer() + if err := srv.RegisterName("SiteService", x); err != nil { + log.Fatal(err) + } + return srv +} + +// ListenAndServeSiteService listen announces on the local network address laddr +// and serves the given SiteService implementation. +func ListenAndServeSiteService(network, addr string, x SiteService) error { + lis, err := net.Listen(network, addr) + if err != nil { + return err + } + defer lis.Close() + + srv := rpc.NewServer() + if err := srv.RegisterName("SiteService", 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 SiteServiceClient struct { + *rpc.Client +} + +// NewSiteServiceClient returns a SiteService rpc.Client and stub to handle +// requests to the set of SiteService at the other end of the connection. +func NewSiteServiceClient(conn io.ReadWriteCloser) (*SiteServiceClient, *rpc.Client) { + c := rpc.NewClientWithCodec(protorpc.NewClientCodec(conn)) + return &SiteServiceClient{c}, c +} + +func (c *SiteServiceClient) GetConfig(in *ConfigRequest, out *ConfigResponse) error { + return c.Call("SiteService.GetConfig", in, out) +} +func (c *SiteServiceClient) GetBalanceType(in *ConfigRequest, out *ConfigResponse) error { + return c.Call("SiteService.GetBalanceType", in, out) +} + +// DialSiteService connects to an SiteService at the specified network address. +func DialSiteService(network, addr string) (*SiteServiceClient, *rpc.Client, error) { + c, err := protorpc.Dial(network, addr) + if err != nil { + return nil, nil, err + } + return &SiteServiceClient{c}, c, nil +} + +// DialSiteServiceTimeout connects to an SiteService at the specified network address. +func DialSiteServiceTimeout(network, addr string, + timeout time.Duration) (*SiteServiceClient, *rpc.Client, error) { + c, err := protorpc.DialTimeout(network, addr, timeout) + if err != nil { + return nil, nil, err + } + return &SiteServiceClient{c}, c, nil +} diff --git a/site.proto b/site.proto new file mode 100644 index 0000000..d08695d --- /dev/null +++ b/site.proto @@ -0,0 +1,23 @@ +syntax = "proto3"; +package site; + +// 配置信息请求结构 +message ConfigRequest { + string data = 1; + string time = 2; + string sign = 3; +} + +// 配置信息响应结构 +message ConfigResponse { + string data = 1; + string time = 2; + string sign = 3; +} + + +// rpc方法 +service SiteService { + rpc getConfig (ConfigRequest) returns (ConfigResponse); // 查询配置信息 + rpc getBalanceType (ConfigRequest) returns (ConfigResponse); // 查询账户类型信息 +} \ No newline at end of file