diff --git a/coupon/client.go b/coupon/client.go new file mode 100644 index 0000000..a742466 --- /dev/null +++ b/coupon/client.go @@ -0,0 +1,227 @@ +package coupon + +import ( + "crypto/md5" + "encoding/hex" + "encoding/json" + "errors" + "log" + "strconv" + "strings" + "time" + + "git.tetele.net/ttlpkg/tgo/crypter" + + "github.com/golang/protobuf/proto" +) + +type AvailableReqArg struct { + Dbname string `json:"dbname"` + UserCouponId string `json:"user_coupon_id"` + UserId string `json:"user_id"` + ProductId string `json:"product_id"` +} + +type AvailableRes struct { + Available bool `json:"available"` + Money string `json:"money"` + Name string `json:"name"` +} + +type UseReqArg struct { + Dbname string `json:"dbname"` + UserCouponId string `json:"user_coupon_id"` + OrderSn string `json:"order_sn"` +} + +type UseRes struct { + Success bool +} + +/** + * 优惠券是否可用 + * @param data {"user_id":"","product_id":"","coupon_id":""} + * @param return is_available,name,money,error + */ +func IsAvailable(dbname, user_id, product_id, user_coupon_id string, url ...string) (*AvailableRes, error) { + + var coupon_rpc_url string = "127.0.0.1:7972" + if len(url) > 0 && url[0] != "" { + coupon_rpc_url = url[0] + } + conn, _, err := DialCouponService("tcp", coupon_rpc_url) + if err != nil { + return nil, err + } + defer conn.Close() + + arg := AvailableReqArg{dbname, user_coupon_id, user_id, product_id} + + data_json, err := json.Marshal(arg) + if err != nil { + return nil, err + } + now_int64 := time.Now().Unix() + + encryData := crypter.DesEn(string(data_json), "conaapon") + + now := strconv.FormatInt(now_int64, 10) + + sign := Sign(encryData, now) + + req := &CouponRequest{proto.String(encryData), proto.String(now), proto.String(sign), nil} + + res := &CouponResponse{} + + err = conn.IsAvailable(req, res) + + if err != nil { + log.Println("coupon rpc error:", err) + 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, "conaapon") + + var res_arr AvailableRes + + err = json.Unmarshal([]byte(res_data_de), &res_arr) + + if err != nil { + return nil, err + } + return &res_arr, nil + } + + return nil, nil +} + +/** + * 使用优惠券 + * @param data {"user_coupon_id":"","order_sn":""} + * @param return is_available,name,money,error + */ +func Use(dbname, user_coupon_id, order_sn string, url ...string) (*UseRes, error) { + + var coupon_rpc_url string = "127.0.0.1:7972" + if len(url) > 0 && url[0] != "" { + coupon_rpc_url = url[0] + } + conn, _, err := DialCouponService("tcp", coupon_rpc_url) + if err != nil { + return nil, err + } + defer conn.Close() + + arg := UseReqArg{dbname, user_coupon_id, order_sn} + + data_json, err := json.Marshal(arg) + if err != nil { + return nil, err + } + now_int64 := time.Now().Unix() + + encryData := crypter.DesEn(string(data_json), "conaapon") + + now := strconv.FormatInt(now_int64, 10) + + sign := Sign(encryData, now) + + req := &CouponRequest{proto.String(encryData), proto.String(now), proto.String(sign), nil} + + res := &CouponResponse{} + + err = conn.Use(req, res) + + if err != nil { + log.Println("coupon rpc error:", err) + 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, "conaapon") + + var res_arr UseRes + + 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("cou$%po87n") + + 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/coupon/client_test.go b/coupon/client_test.go new file mode 100644 index 0000000..277d735 --- /dev/null +++ b/coupon/client_test.go @@ -0,0 +1,17 @@ +package coupon + +import ( + "testing" +) + +func Test_IsAvailable(t *testing.T) { + + dbname := "test1_tetele_com" + user_id := "2" + product_id := "2" + user_coupon_id := "3" + res, err := IsAvailable(dbname, user_id, product_id, user_coupon_id) + t.Log(res) + t.Log(err) + +} diff --git a/coupon/coupon.pb.go b/coupon/coupon.pb.go new file mode 100644 index 0000000..640c6b3 --- /dev/null +++ b/coupon/coupon.pb.go @@ -0,0 +1,198 @@ +// Code generated by protoc-gen-go. +// source: coupon.proto +// DO NOT EDIT! + +/* +Package coupon is a generated protocol buffer package. + +It is generated from these files: + coupon.proto + +It has these top-level messages: + CouponRequest + CouponResponse +*/ +package coupon + +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 CouponRequest 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 *CouponRequest) Reset() { *m = CouponRequest{} } +func (m *CouponRequest) String() string { return proto.CompactTextString(m) } +func (*CouponRequest) ProtoMessage() {} + +func (m *CouponRequest) GetData() string { + if m != nil && m.Data != nil { + return *m.Data + } + return "" +} + +func (m *CouponRequest) GetTime() string { + if m != nil && m.Time != nil { + return *m.Time + } + return "" +} + +func (m *CouponRequest) GetSign() string { + if m != nil && m.Sign != nil { + return *m.Sign + } + return "" +} + +// 是否可用响应结构 +type CouponResponse 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 *CouponResponse) Reset() { *m = CouponResponse{} } +func (m *CouponResponse) String() string { return proto.CompactTextString(m) } +func (*CouponResponse) ProtoMessage() {} + +func (m *CouponResponse) GetData() string { + if m != nil && m.Data != nil { + return *m.Data + } + return "" +} + +func (m *CouponResponse) GetTime() string { + if m != nil && m.Time != nil { + return *m.Time + } + return "" +} + +func (m *CouponResponse) GetSign() string { + if m != nil && m.Sign != nil { + return *m.Sign + } + return "" +} + +func init() { +} + +type CouponService interface { + IsAvailable(in *CouponRequest, out *CouponResponse) error + Use(in *CouponRequest, out *CouponResponse) error +} + +// AcceptCouponServiceClient accepts connections on the listener and serves requests +// for each incoming connection. Accept blocks; the caller typically +// invokes it in a go statement. +func AcceptCouponServiceClient(lis net.Listener, x CouponService) { + srv := rpc.NewServer() + if err := srv.RegisterName("CouponService", 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)) + } +} + +// RegisterCouponService publish the given CouponService implementation on the server. +func RegisterCouponService(srv *rpc.Server, x CouponService) error { + if err := srv.RegisterName("CouponService", x); err != nil { + return err + } + return nil +} + +// NewCouponServiceServer returns a new CouponService Server. +func NewCouponServiceServer(x CouponService) *rpc.Server { + srv := rpc.NewServer() + if err := srv.RegisterName("CouponService", x); err != nil { + log.Fatal(err) + } + return srv +} + +// ListenAndServeCouponService listen announces on the local network address laddr +// and serves the given CouponService implementation. +func ListenAndServeCouponService(network, addr string, x CouponService) error { + lis, err := net.Listen(network, addr) + if err != nil { + return err + } + defer lis.Close() + + srv := rpc.NewServer() + if err := srv.RegisterName("CouponService", 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 CouponServiceClient struct { + *rpc.Client +} + +// NewCouponServiceClient returns a CouponService rpc.Client and stub to handle +// requests to the set of CouponService at the other end of the connection. +func NewCouponServiceClient(conn io.ReadWriteCloser) (*CouponServiceClient, *rpc.Client) { + c := rpc.NewClientWithCodec(protorpc.NewClientCodec(conn)) + return &CouponServiceClient{c}, c +} + +func (c *CouponServiceClient) IsAvailable(in *CouponRequest, out *CouponResponse) error { + return c.Call("CouponService.IsAvailable", in, out) +} +func (c *CouponServiceClient) Use(in *CouponRequest, out *CouponResponse) error { + return c.Call("CouponService.Use", in, out) +} + +// DialCouponService connects to an CouponService at the specified network address. +func DialCouponService(network, addr string) (*CouponServiceClient, *rpc.Client, error) { + c, err := protorpc.Dial(network, addr) + if err != nil { + return nil, nil, err + } + return &CouponServiceClient{c}, c, nil +} + +// DialCouponServiceTimeout connects to an CouponService at the specified network address. +func DialCouponServiceTimeout(network, addr string, + timeout time.Duration) (*CouponServiceClient, *rpc.Client, error) { + c, err := protorpc.DialTimeout(network, addr, timeout) + if err != nil { + return nil, nil, err + } + return &CouponServiceClient{c}, c, nil +} diff --git a/coupon/coupon.proto b/coupon/coupon.proto new file mode 100644 index 0000000..c640725 --- /dev/null +++ b/coupon/coupon.proto @@ -0,0 +1,23 @@ +syntax = "proto3"; +package coupon; + +// 是否可用请求结构 +message CouponRequest { + string data = 1; + string time = 2; + string sign = 3; +} + +// 是否可用响应结构 +message CouponResponse { + string data = 1; + string time = 2; + string sign = 3; +} + + +// rpc方法 +service CouponService { + rpc isAvailable (CouponRequest) returns (CouponResponse); // 优惠券是否可用 + rpc use (CouponRequest) returns (CouponResponse); // 使用优惠券 +} \ No newline at end of file diff --git a/mastermsg/client.go b/mastermsg/client.go new file mode 100644 index 0000000..855b6ad --- /dev/null +++ b/mastermsg/client.go @@ -0,0 +1,42 @@ +package mastermsg + +import ( + "encoding/json" + + "github.com/golang/protobuf/proto" +) + +func SendEventMsg(site_id, domainname, _type, memo string, data map[string]string, url ...string) (string, error) { + + var msg_rpc_url string = "127.0.0.1:7974" + if len(url) > 0 && url[0] != "" { + msg_rpc_url = url[0] + } + conn, _, err := DialMsgService("tcp", msg_rpc_url) + if err != nil { + return "", err + } + defer conn.Close() + + _data, err := json.Marshal(data) + if err != nil { + return "", err + } + + req := &EventRequest{proto.String(site_id), proto.String(domainname), proto.String(_type), proto.String(memo), proto.String(string(_data)), nil} + + res := &EventResponse{} + + err = conn.SendEvent(req, res) + + if err != nil { + return "", err + } + + event_id := res.GetEventId() + if event_id != "" { + return event_id, nil + } + + return "", nil +} diff --git a/mastermsg/client_test.go b/mastermsg/client_test.go new file mode 100644 index 0000000..4a05d1d --- /dev/null +++ b/mastermsg/client_test.go @@ -0,0 +1,18 @@ +package mastermsg + +import ( + "testing" +) + +func Test_SendEventMsg(t *testing.T) { + site_id := "1000064" + domainname := "http://dev.tetele.net" + _type := "warning" + memo := "余额不足" + data := map[string]string{"title": "sss顶替"} + + res, err := SendEventMsg(site_id, domainname, _type, memo, data) + t.Log(res) + t.Log(err) + +} diff --git a/mastermsg/mastermsg.pb.go b/mastermsg/mastermsg.pb.go new file mode 100644 index 0000000..f0b8586 --- /dev/null +++ b/mastermsg/mastermsg.pb.go @@ -0,0 +1,194 @@ +// Code generated by protoc-gen-go. +// source: mastermsg.proto +// DO NOT EDIT! + +/* +Package mastermsg is a generated protocol buffer package. + +It is generated from these files: + mastermsg.proto + +It has these top-level messages: + EventRequest + EventResponse +*/ +package mastermsg + +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 EventRequest struct { + SiteId *string `protobuf:"bytes,1,opt,name=site_id" json:"site_id,omitempty"` + Domainname *string `protobuf:"bytes,2,opt,name=domainname" json:"domainname,omitempty"` + Type *string `protobuf:"bytes,3,opt,name=type" json:"type,omitempty"` + Memo *string `protobuf:"bytes,4,opt,name=memo" json:"memo,omitempty"` + Data *string `protobuf:"bytes,5,opt,name=data" json:"data,omitempty"` + XXX_unrecognized []byte `json:"-"` +} + +func (m *EventRequest) Reset() { *m = EventRequest{} } +func (m *EventRequest) String() string { return proto.CompactTextString(m) } +func (*EventRequest) ProtoMessage() {} + +func (m *EventRequest) GetSiteId() string { + if m != nil && m.SiteId != nil { + return *m.SiteId + } + return "" +} + +func (m *EventRequest) GetDomainname() string { + if m != nil && m.Domainname != nil { + return *m.Domainname + } + return "" +} + +func (m *EventRequest) GetType() string { + if m != nil && m.Type != nil { + return *m.Type + } + return "" +} + +func (m *EventRequest) GetMemo() string { + if m != nil && m.Memo != nil { + return *m.Memo + } + return "" +} + +func (m *EventRequest) GetData() string { + if m != nil && m.Data != nil { + return *m.Data + } + return "" +} + +// 算术运算响应结构 +type EventResponse struct { + EventId *string `protobuf:"bytes,1,opt,name=event_id" json:"event_id,omitempty"` + XXX_unrecognized []byte `json:"-"` +} + +func (m *EventResponse) Reset() { *m = EventResponse{} } +func (m *EventResponse) String() string { return proto.CompactTextString(m) } +func (*EventResponse) ProtoMessage() {} + +func (m *EventResponse) GetEventId() string { + if m != nil && m.EventId != nil { + return *m.EventId + } + return "" +} + +func init() { +} + +type MsgService interface { + SendEvent(in *EventRequest, out *EventResponse) error +} + +// AcceptMsgServiceClient accepts connections on the listener and serves requests +// for each incoming connection. Accept blocks; the caller typically +// invokes it in a go statement. +func AcceptMsgServiceClient(lis net.Listener, x MsgService) { + srv := rpc.NewServer() + if err := srv.RegisterName("MsgService", 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)) + } +} + +// RegisterMsgService publish the given MsgService implementation on the server. +func RegisterMsgService(srv *rpc.Server, x MsgService) error { + if err := srv.RegisterName("MsgService", x); err != nil { + return err + } + return nil +} + +// NewMsgServiceServer returns a new MsgService Server. +func NewMsgServiceServer(x MsgService) *rpc.Server { + srv := rpc.NewServer() + if err := srv.RegisterName("MsgService", x); err != nil { + log.Fatal(err) + } + return srv +} + +// ListenAndServeMsgService listen announces on the local network address laddr +// and serves the given MsgService implementation. +func ListenAndServeMsgService(network, addr string, x MsgService) error { + lis, err := net.Listen(network, addr) + if err != nil { + return err + } + defer lis.Close() + + srv := rpc.NewServer() + if err := srv.RegisterName("MsgService", 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 MsgServiceClient struct { + *rpc.Client +} + +// NewMsgServiceClient returns a MsgService rpc.Client and stub to handle +// requests to the set of MsgService at the other end of the connection. +func NewMsgServiceClient(conn io.ReadWriteCloser) (*MsgServiceClient, *rpc.Client) { + c := rpc.NewClientWithCodec(protorpc.NewClientCodec(conn)) + return &MsgServiceClient{c}, c +} + +func (c *MsgServiceClient) SendEvent(in *EventRequest, out *EventResponse) error { + return c.Call("MsgService.SendEvent", in, out) +} + +// DialMsgService connects to an MsgService at the specified network address. +func DialMsgService(network, addr string) (*MsgServiceClient, *rpc.Client, error) { + c, err := protorpc.Dial(network, addr) + if err != nil { + return nil, nil, err + } + return &MsgServiceClient{c}, c, nil +} + +// DialMsgServiceTimeout connects to an MsgService at the specified network address. +func DialMsgServiceTimeout(network, addr string, + timeout time.Duration) (*MsgServiceClient, *rpc.Client, error) { + c, err := protorpc.DialTimeout(network, addr, timeout) + if err != nil { + return nil, nil, err + } + return &MsgServiceClient{c}, c, nil +} diff --git a/mastermsg/mastermsg.proto b/mastermsg/mastermsg.proto new file mode 100644 index 0000000..14a493e --- /dev/null +++ b/mastermsg/mastermsg.proto @@ -0,0 +1,21 @@ +syntax = "proto3"; +package mastermsg; + +// 算术运算请求结构 +message EventRequest { + string site_id = 1; + string domainname = 2; + string type = 3; + string memo = 4; + string data = 5; +} + +// 算术运算响应结构 +message EventResponse { + string event_id = 1; //ID +} + +// rpc方法 +service MsgService { + rpc sendEvent (EventRequest) returns (EventResponse); // 发送事件消息 +} \ No newline at end of file diff --git a/order/client.go b/order/client.go new file mode 100644 index 0000000..86b06f5 --- /dev/null +++ b/order/client.go @@ -0,0 +1,240 @@ +package order + +import ( + "crypto/md5" + "encoding/hex" + "encoding/json" + "errors" + "log" + "strconv" + "strings" + "time" + + "git.tetele.net/ttlpkg/tgo/crypter" + + "github.com/golang/protobuf/proto" +) + +type OrderCreateRes struct { + OrderSn string +} + +func Create(dbname, site_id string, data map[string]string, url ...string) (*OrderCreateRes, error) { + + var order_rpc_url string = "127.0.0.1:7973" + if len(url) > 0 && url[0] != "" { + order_rpc_url = url[0] + } + conn, _, err := DialOrderService("tcp", order_rpc_url) + if err != nil { + return nil, err + } + defer conn.Close() + + data["dbname"] = dbname + data["site_id"] = site_id + + data_json, err := json.Marshal(data) + if err != nil { + return nil, err + } + + encryData := crypter.DesEn(string(data_json), "ordernew") + + now_int64 := time.Now().Unix() + + now := strconv.FormatInt(now_int64, 10) + + sign := Sign(encryData, now) + + req := &CreateRequest{ + proto.String(encryData), + proto.String(now), + proto.String(sign), + nil} + + res := &CreateResponse{} + + err = conn.Create(req, res) + + if err != nil { + return nil, err + } + + return HandleRes(res) + + // return map[string]interface{}{ + // "code": res.GetCode(), + // "msg": res.GetMsg(), + // "order_sn": res.GetOrderSn()}, err +} + +func CreateByChannel(dbname, site_id string, data map[string]string, url ...string) (*OrderCreateRes, error) { + + var order_rpc_url string = "127.0.0.1:7973" + if len(url) > 0 && url[0] != "" { + order_rpc_url = url[0] + } + conn, _, err := DialOrderService("tcp", order_rpc_url) + if err != nil { + return nil, err + } + defer conn.Close() + + data["dbname"] = dbname + data["site_id"] = site_id + data["source"] = "channel" + + data_json, err := json.Marshal(data) + if err != nil { + return nil, err + } + + encryData := crypter.DesEn(string(data_json), "ordernew") + + now := strconv.FormatInt(time.Now().Unix(), 10) + + sign := Sign(encryData, now) + + req := &CreateRequest{ + proto.String(encryData), + proto.String(now), + proto.String(sign), + nil} + + res := &CreateResponse{} + + err = conn.Create(req, res) + + if err != nil { + return nil, err + } + + return HandleRes(res) + // return map[string]interface{}{ + // "code": res.GetCode(), + // "msg": res.GetMsg(), + // "order_sn": res.GetOrderSn()}, err +} + +func CreateByCart(dbname, site_id string, data map[string]interface{}, url ...string) (*OrderCreateRes, error) { + + var order_rpc_url string = "127.0.0.1:7973" + if len(url) > 0 && url[0] != "" { + order_rpc_url = url[0] + } + conn, _, err := DialOrderService("tcp", order_rpc_url) + if err != nil { + return nil, err + } + defer conn.Close() + + data["dbname"] = dbname + data["site_id"] = site_id + + data_json, err := json.Marshal(data) + if err != nil { + return nil, err + } + + log.Println(string(data_json)) + + encryData := crypter.DesEn(string(data_json), "cartsnew") + + now := strconv.FormatInt(time.Now().Unix(), 10) + + sign := Sign(encryData, now) + + req := &CreateRequest{ + proto.String(encryData), + proto.String(now), + proto.String(sign), + nil} + + res := &CreateResponse{} + + err = conn.CreateByCart(req, res) + + if err != nil { + return nil, err + } + + return HandleRes(res) + // return map[string]interface{}{ + // "code": res.GetCode(), + // "msg": res.GetMsg(), + // "order_sn": res.GetOrderSn()}, err +} + +/** + * 签名 + */ +func Sign(data string, salt string) string { + + var build strings.Builder + + build.WriteString(data) + build.WriteString(salt) + build.WriteString("neworder") + + 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 HandleRes(res *CreateResponse) (*OrderCreateRes, 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, "ordernew") + + var res_arr OrderCreateRes + + err = json.Unmarshal([]byte(res_data_de), &res_arr) + + if err != nil { + return nil, err + } + + return &res_arr, nil +} diff --git a/order/client_test.go b/order/client_test.go new file mode 100644 index 0000000..aa57085 --- /dev/null +++ b/order/client_test.go @@ -0,0 +1,31 @@ +package order + +import ( + "testing" +) + +func Test_CreateByCart(t *testing.T) { + + products := []map[string]string{ + map[string]string{ + "product_id": "18", + "sku_id": "6", + "quantity": "3", + }, + map[string]string{ + "product_id": "6", + "sku_id": "7", + "quantity": "2", + }, + } + data := map[string]interface{}{ + "product": products, + "name": "5BC02AB31C", + "mobile": "326598744", + } + + ret, err := CreateByCart("test1_tetele_com", "100064", data) + + t.Log(ret) + t.Log(err) +} diff --git a/order/order.pb.go b/order/order.pb.go new file mode 100644 index 0000000..b03b3aa --- /dev/null +++ b/order/order.pb.go @@ -0,0 +1,198 @@ +// Code generated by protoc-gen-go. +// source: order.proto +// DO NOT EDIT! + +/* +Package order is a generated protocol buffer package. + +It is generated from these files: + order.proto + +It has these top-level messages: + CreateRequest + CreateResponse +*/ +package order + +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 CreateRequest 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 *CreateRequest) Reset() { *m = CreateRequest{} } +func (m *CreateRequest) String() string { return proto.CompactTextString(m) } +func (*CreateRequest) ProtoMessage() {} + +func (m *CreateRequest) GetData() string { + if m != nil && m.Data != nil { + return *m.Data + } + return "" +} + +func (m *CreateRequest) GetTime() string { + if m != nil && m.Time != nil { + return *m.Time + } + return "" +} + +func (m *CreateRequest) GetSign() string { + if m != nil && m.Sign != nil { + return *m.Sign + } + return "" +} + +// 下单响应结构 +type CreateResponse 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 *CreateResponse) Reset() { *m = CreateResponse{} } +func (m *CreateResponse) String() string { return proto.CompactTextString(m) } +func (*CreateResponse) ProtoMessage() {} + +func (m *CreateResponse) GetData() string { + if m != nil && m.Data != nil { + return *m.Data + } + return "" +} + +func (m *CreateResponse) GetTime() string { + if m != nil && m.Time != nil { + return *m.Time + } + return "" +} + +func (m *CreateResponse) GetSign() string { + if m != nil && m.Sign != nil { + return *m.Sign + } + return "" +} + +func init() { +} + +type OrderService interface { + Create(in *CreateRequest, out *CreateResponse) error + CreateByCart(in *CreateRequest, out *CreateResponse) error +} + +// AcceptOrderServiceClient accepts connections on the listener and serves requests +// for each incoming connection. Accept blocks; the caller typically +// invokes it in a go statement. +func AcceptOrderServiceClient(lis net.Listener, x OrderService) { + srv := rpc.NewServer() + if err := srv.RegisterName("OrderService", 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)) + } +} + +// RegisterOrderService publish the given OrderService implementation on the server. +func RegisterOrderService(srv *rpc.Server, x OrderService) error { + if err := srv.RegisterName("OrderService", x); err != nil { + return err + } + return nil +} + +// NewOrderServiceServer returns a new OrderService Server. +func NewOrderServiceServer(x OrderService) *rpc.Server { + srv := rpc.NewServer() + if err := srv.RegisterName("OrderService", x); err != nil { + log.Fatal(err) + } + return srv +} + +// ListenAndServeOrderService listen announces on the local network address laddr +// and serves the given OrderService implementation. +func ListenAndServeOrderService(network, addr string, x OrderService) error { + lis, err := net.Listen(network, addr) + if err != nil { + return err + } + defer lis.Close() + + srv := rpc.NewServer() + if err := srv.RegisterName("OrderService", 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 OrderServiceClient struct { + *rpc.Client +} + +// NewOrderServiceClient returns a OrderService rpc.Client and stub to handle +// requests to the set of OrderService at the other end of the connection. +func NewOrderServiceClient(conn io.ReadWriteCloser) (*OrderServiceClient, *rpc.Client) { + c := rpc.NewClientWithCodec(protorpc.NewClientCodec(conn)) + return &OrderServiceClient{c}, c +} + +func (c *OrderServiceClient) Create(in *CreateRequest, out *CreateResponse) error { + return c.Call("OrderService.Create", in, out) +} +func (c *OrderServiceClient) CreateByCart(in *CreateRequest, out *CreateResponse) error { + return c.Call("OrderService.CreateByCart", in, out) +} + +// DialOrderService connects to an OrderService at the specified network address. +func DialOrderService(network, addr string) (*OrderServiceClient, *rpc.Client, error) { + c, err := protorpc.Dial(network, addr) + if err != nil { + return nil, nil, err + } + return &OrderServiceClient{c}, c, nil +} + +// DialOrderServiceTimeout connects to an OrderService at the specified network address. +func DialOrderServiceTimeout(network, addr string, + timeout time.Duration) (*OrderServiceClient, *rpc.Client, error) { + c, err := protorpc.DialTimeout(network, addr, timeout) + if err != nil { + return nil, nil, err + } + return &OrderServiceClient{c}, c, nil +} diff --git a/order/order.proto b/order/order.proto new file mode 100644 index 0000000..68f0a1d --- /dev/null +++ b/order/order.proto @@ -0,0 +1,22 @@ +syntax = "proto3"; +package order; + +// 下单请求结构 +message CreateRequest { + string data = 1; + string time = 2; + string sign = 3; +} + +// 下单响应结构 +message CreateResponse { + string data = 1; //订单号 + string time = 2; + string sign = 3; +} + +// rpc方法 +service OrderService { + rpc create (CreateRequest) returns (CreateResponse); // 创建订单 + rpc createByCart (CreateRequest) returns (CreateResponse); // 创建订单 +} \ No newline at end of file diff --git a/site/client.balance.go b/site/client.balance.go new file mode 100644 index 0000000..d43a50a --- /dev/null +++ b/site/client.balance.go @@ -0,0 +1,109 @@ +package site + +import ( + "crypto/md5" + "encoding/hex" + "encoding/json" + "errors" + "strconv" + "strings" + "time" + + "git.tetele.net/ttlpkg/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, 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/site/client.go b/site/client.go new file mode 100644 index 0000000..ad8d23f --- /dev/null +++ b/site/client.go @@ -0,0 +1,135 @@ +package site + +import ( + "crypto/md5" + "encoding/hex" + "encoding/json" + "errors" + "strconv" + "strings" + "time" + + "git.tetele.net/ttlpkg/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/site/client_test.go b/site/client_test.go new file mode 100644 index 0000000..a20b19d --- /dev/null +++ b/site/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/site.pb.go b/site/site.pb.go new file mode 100644 index 0000000..24f2d3f --- /dev/null +++ b/site/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/site.proto b/site/site.proto new file mode 100644 index 0000000..d08695d --- /dev/null +++ b/site/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 diff --git a/user/user.client.go b/user/user.client.go new file mode 100644 index 0000000..e616e24 --- /dev/null +++ b/user/user.client.go @@ -0,0 +1,91 @@ +package user + +import ( + "github.com/golang/protobuf/proto" +) + +func GetUserByToken(dbname, token string, url ...string) (map[string]string, error) { + + var user_rpc_url string = "127.0.0.1:7976" + if len(url) > 0 && url[0] != "" { + user_rpc_url = url[0] + } + conn, _, err := DialUserService("tcp", user_rpc_url) + if err != nil { + return map[string]string{}, err + } + defer conn.Close() + + req := &UserRequest{proto.String(dbname), proto.String(token), nil} + + res := &UserResponse{} + + err = conn.GetByToken(req, res) + + if err != nil { + return map[string]string{}, err + } + + if res.GetUserId() != "" { + return map[string]string{ + "UserId": res.GetUserId(), + "Username": res.GetUsername(), + "Nickname": res.GetNickname(), + "Mobile": res.GetMobile(), + "Email": res.GetEmail(), + "Status": res.GetStatus(), + "BusinessId": res.GetBusinessId(), + "StoreId": res.GetStoreId(), + "FansTo": res.GetFansTo(), + "IsVip": res.GetIsVip(), + "Usercode": res.GetUsercode(), + "GroupId": res.GetGroupId(), + }, nil + } + + return map[string]string{}, nil +} + +func Login(dbname, account, password string, url ...string) (map[string]string, error) { + + var user_rpc_url string = "127.0.0.1:7976" + if len(url) > 0 && url[0] != "" { + user_rpc_url = url[0] + } + + conn, _, err := DialUserService("tcp", user_rpc_url) + if err != nil { + return map[string]string{}, err + } + defer conn.Close() + + req := &LoginRequest{proto.String(dbname), proto.String(account), proto.String(password), nil} + + res := &LoginResponse{} + + err = conn.Login(req, res) + + if err != nil { + return map[string]string{}, err + } + + if res.GetUserId() != "" { + return map[string]string{ + "UserId": res.GetUserId(), + "Username": res.GetUsername(), + "Nickname": res.GetNickname(), + "Mobile": res.GetMobile(), + "Email": res.GetEmail(), + "Status": res.GetStatus(), + "BusinessId": res.GetBusinessId(), + "StoreId": res.GetStoreId(), + "FansTo": res.GetFansTo(), + "IsVip": res.GetIsVip(), + "Token": res.GetToken(), + "Usercode": res.GetUsercode(), + "GroupId": res.GetGroupId(), + }, nil + } + + return map[string]string{}, nil +} diff --git a/user/user.client_test.go b/user/user.client_test.go new file mode 100644 index 0000000..9136c05 --- /dev/null +++ b/user/user.client_test.go @@ -0,0 +1,30 @@ +package user + +import ( + "testing" +) + +func Test_GetUserByToken(t *testing.T) { + dbname := "dev_tetele_net" + token := "5536827c-36a5-4ec7-946d-49e4380c5103" + res, err := GetUserByToken(dbname, token, "111.229.34.252:7976") + t.Log(res) + t.Log(res["UserId"]) + t.Log(err) + + res, err = GetUserById(dbname, "114") + t.Log(res) + t.Log(res["UserId"]) + t.Log(err) +} + +func Test_Login(t *testing.T) { + dbname := "dev_tetele_net" + account := "adminw" + password := "e10adc3949ba59abbe56e057f20f883e" + res, err := Login(dbname, account, password) + t.Log(res) + t.Log(res["UserId"]) + t.Log(err) + +} diff --git a/user/user.pb.go b/user/user.pb.go new file mode 100644 index 0000000..c45fd23 --- /dev/null +++ b/user/user.pb.go @@ -0,0 +1,410 @@ +// Code generated by protoc-gen-go. +// source: user.proto +// DO NOT EDIT! + +/* +Package user is a generated protocol buffer package. + +It is generated from these files: + user.proto + +It has these top-level messages: + UserRequest + UserResponse + LoginRequest + LoginResponse +*/ +package user + +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 + +// 使用token查询用户信息请求结构 +type UserRequest struct { + Dbname *string `protobuf:"bytes,1,opt,name=dbname" json:"dbname,omitempty"` + Token *string `protobuf:"bytes,2,opt,name=token" json:"token,omitempty"` + XXX_unrecognized []byte `json:"-"` +} + +func (m *UserRequest) Reset() { *m = UserRequest{} } +func (m *UserRequest) String() string { return proto.CompactTextString(m) } +func (*UserRequest) ProtoMessage() {} + +func (m *UserRequest) GetDbname() string { + if m != nil && m.Dbname != nil { + return *m.Dbname + } + return "" +} + +func (m *UserRequest) GetToken() string { + if m != nil && m.Token != nil { + return *m.Token + } + return "" +} + +// 使用token查询用户信息响应结构 +type UserResponse struct { + UserId *string `protobuf:"bytes,1,opt,name=user_id" json:"user_id,omitempty"` + Username *string `protobuf:"bytes,2,opt,name=username" json:"username,omitempty"` + Nickname *string `protobuf:"bytes,3,opt,name=nickname" json:"nickname,omitempty"` + Mobile *string `protobuf:"bytes,4,opt,name=mobile" json:"mobile,omitempty"` + Email *string `protobuf:"bytes,5,opt,name=email" json:"email,omitempty"` + Status *string `protobuf:"bytes,6,opt,name=status" json:"status,omitempty"` + BusinessId *string `protobuf:"bytes,7,opt,name=business_id" json:"business_id,omitempty"` + StoreId *string `protobuf:"bytes,8,opt,name=store_id" json:"store_id,omitempty"` + FansTo *string `protobuf:"bytes,9,opt,name=fans_to" json:"fans_to,omitempty"` + IsVip *string `protobuf:"bytes,10,opt,name=is_vip" json:"is_vip,omitempty"` + Usercode *string `protobuf:"bytes,11,opt,name=usercode" json:"usercode,omitempty"` + GroupId *string `protobuf:"bytes,12,opt,name=group_id" json:"group_id,omitempty"` + XXX_unrecognized []byte `json:"-"` +} + +func (m *UserResponse) Reset() { *m = UserResponse{} } +func (m *UserResponse) String() string { return proto.CompactTextString(m) } +func (*UserResponse) ProtoMessage() {} + +func (m *UserResponse) GetUserId() string { + if m != nil && m.UserId != nil { + return *m.UserId + } + return "" +} + +func (m *UserResponse) GetUsername() string { + if m != nil && m.Username != nil { + return *m.Username + } + return "" +} + +func (m *UserResponse) GetNickname() string { + if m != nil && m.Nickname != nil { + return *m.Nickname + } + return "" +} + +func (m *UserResponse) GetMobile() string { + if m != nil && m.Mobile != nil { + return *m.Mobile + } + return "" +} + +func (m *UserResponse) GetEmail() string { + if m != nil && m.Email != nil { + return *m.Email + } + return "" +} + +func (m *UserResponse) GetStatus() string { + if m != nil && m.Status != nil { + return *m.Status + } + return "" +} + +func (m *UserResponse) GetBusinessId() string { + if m != nil && m.BusinessId != nil { + return *m.BusinessId + } + return "" +} + +func (m *UserResponse) GetStoreId() string { + if m != nil && m.StoreId != nil { + return *m.StoreId + } + return "" +} + +func (m *UserResponse) GetFansTo() string { + if m != nil && m.FansTo != nil { + return *m.FansTo + } + return "" +} + +func (m *UserResponse) GetIsVip() string { + if m != nil && m.IsVip != nil { + return *m.IsVip + } + return "" +} + +func (m *UserResponse) GetUsercode() string { + if m != nil && m.Usercode != nil { + return *m.Usercode + } + return "" +} + +func (m *UserResponse) GetGroupId() string { + if m != nil && m.GroupId != nil { + return *m.GroupId + } + return "" +} + +// 用户登录请求结构 +type LoginRequest struct { + Dbname *string `protobuf:"bytes,1,opt,name=dbname" json:"dbname,omitempty"` + Account *string `protobuf:"bytes,2,opt,name=account" json:"account,omitempty"` + Password *string `protobuf:"bytes,3,opt,name=password" json:"password,omitempty"` + XXX_unrecognized []byte `json:"-"` +} + +func (m *LoginRequest) Reset() { *m = LoginRequest{} } +func (m *LoginRequest) String() string { return proto.CompactTextString(m) } +func (*LoginRequest) ProtoMessage() {} + +func (m *LoginRequest) GetDbname() string { + if m != nil && m.Dbname != nil { + return *m.Dbname + } + return "" +} + +func (m *LoginRequest) GetAccount() string { + if m != nil && m.Account != nil { + return *m.Account + } + return "" +} + +func (m *LoginRequest) GetPassword() string { + if m != nil && m.Password != nil { + return *m.Password + } + return "" +} + +// 使用token查询用户信息响应结构 +type LoginResponse struct { + UserId *string `protobuf:"bytes,1,opt,name=user_id" json:"user_id,omitempty"` + Username *string `protobuf:"bytes,2,opt,name=username" json:"username,omitempty"` + Nickname *string `protobuf:"bytes,3,opt,name=nickname" json:"nickname,omitempty"` + Mobile *string `protobuf:"bytes,4,opt,name=mobile" json:"mobile,omitempty"` + Email *string `protobuf:"bytes,5,opt,name=email" json:"email,omitempty"` + Status *string `protobuf:"bytes,6,opt,name=status" json:"status,omitempty"` + BusinessId *string `protobuf:"bytes,7,opt,name=business_id" json:"business_id,omitempty"` + StoreId *string `protobuf:"bytes,8,opt,name=store_id" json:"store_id,omitempty"` + FansTo *string `protobuf:"bytes,9,opt,name=fans_to" json:"fans_to,omitempty"` + IsVip *string `protobuf:"bytes,10,opt,name=is_vip" json:"is_vip,omitempty"` + Token *string `protobuf:"bytes,11,opt,name=token" json:"token,omitempty"` + Usercode *string `protobuf:"bytes,12,opt,name=usercode" json:"usercode,omitempty"` + GroupId *string `protobuf:"bytes,13,opt,name=group_id" json:"group_id,omitempty"` + XXX_unrecognized []byte `json:"-"` +} + +func (m *LoginResponse) Reset() { *m = LoginResponse{} } +func (m *LoginResponse) String() string { return proto.CompactTextString(m) } +func (*LoginResponse) ProtoMessage() {} + +func (m *LoginResponse) GetUserId() string { + if m != nil && m.UserId != nil { + return *m.UserId + } + return "" +} + +func (m *LoginResponse) GetUsername() string { + if m != nil && m.Username != nil { + return *m.Username + } + return "" +} + +func (m *LoginResponse) GetNickname() string { + if m != nil && m.Nickname != nil { + return *m.Nickname + } + return "" +} + +func (m *LoginResponse) GetMobile() string { + if m != nil && m.Mobile != nil { + return *m.Mobile + } + return "" +} + +func (m *LoginResponse) GetEmail() string { + if m != nil && m.Email != nil { + return *m.Email + } + return "" +} + +func (m *LoginResponse) GetStatus() string { + if m != nil && m.Status != nil { + return *m.Status + } + return "" +} + +func (m *LoginResponse) GetBusinessId() string { + if m != nil && m.BusinessId != nil { + return *m.BusinessId + } + return "" +} + +func (m *LoginResponse) GetStoreId() string { + if m != nil && m.StoreId != nil { + return *m.StoreId + } + return "" +} + +func (m *LoginResponse) GetFansTo() string { + if m != nil && m.FansTo != nil { + return *m.FansTo + } + return "" +} + +func (m *LoginResponse) GetIsVip() string { + if m != nil && m.IsVip != nil { + return *m.IsVip + } + return "" +} + +func (m *LoginResponse) GetToken() string { + if m != nil && m.Token != nil { + return *m.Token + } + return "" +} + +func (m *LoginResponse) GetUsercode() string { + if m != nil && m.Usercode != nil { + return *m.Usercode + } + return "" +} + +func (m *LoginResponse) GetGroupId() string { + if m != nil && m.GroupId != nil { + return *m.GroupId + } + return "" +} + +func init() { +} + +type UserService interface { + GetByToken(in *UserRequest, out *UserResponse) error + Login(in *LoginRequest, out *LoginResponse) error +} + +// AcceptUserServiceClient accepts connections on the listener and serves requests +// for each incoming connection. Accept blocks; the caller typically +// invokes it in a go statement. +func AcceptUserServiceClient(lis net.Listener, x UserService) { + srv := rpc.NewServer() + if err := srv.RegisterName("UserService", 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)) + } +} + +// RegisterUserService publish the given UserService implementation on the server. +func RegisterUserService(srv *rpc.Server, x UserService) error { + if err := srv.RegisterName("UserService", x); err != nil { + return err + } + return nil +} + +// NewUserServiceServer returns a new UserService Server. +func NewUserServiceServer(x UserService) *rpc.Server { + srv := rpc.NewServer() + if err := srv.RegisterName("UserService", x); err != nil { + log.Fatal(err) + } + return srv +} + +// ListenAndServeUserService listen announces on the local network address laddr +// and serves the given UserService implementation. +func ListenAndServeUserService(network, addr string, x UserService) error { + lis, err := net.Listen(network, addr) + if err != nil { + return err + } + defer lis.Close() + + srv := rpc.NewServer() + if err := srv.RegisterName("UserService", 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 UserServiceClient struct { + *rpc.Client +} + +// NewUserServiceClient returns a UserService rpc.Client and stub to handle +// requests to the set of UserService at the other end of the connection. +func NewUserServiceClient(conn io.ReadWriteCloser) (*UserServiceClient, *rpc.Client) { + c := rpc.NewClientWithCodec(protorpc.NewClientCodec(conn)) + return &UserServiceClient{c}, c +} + +func (c *UserServiceClient) GetByToken(in *UserRequest, out *UserResponse) error { + return c.Call("UserService.GetByToken", in, out) +} +func (c *UserServiceClient) Login(in *LoginRequest, out *LoginResponse) error { + return c.Call("UserService.Login", in, out) +} + +// DialUserService connects to an UserService at the specified network address. +func DialUserService(network, addr string) (*UserServiceClient, *rpc.Client, error) { + c, err := protorpc.Dial(network, addr) + if err != nil { + return nil, nil, err + } + return &UserServiceClient{c}, c, nil +} + +// DialUserServiceTimeout connects to an UserService at the specified network address. +func DialUserServiceTimeout(network, addr string, + timeout time.Duration) (*UserServiceClient, *rpc.Client, error) { + c, err := protorpc.DialTimeout(network, addr, timeout) + if err != nil { + return nil, nil, err + } + return &UserServiceClient{c}, c, nil +} diff --git a/user/user.proto b/user/user.proto new file mode 100644 index 0000000..04bc44a --- /dev/null +++ b/user/user.proto @@ -0,0 +1,56 @@ +syntax = "proto3"; +package user; + +// 使用token查询用户信息请求结构 +message UserRequest { + string dbname = 1; + string token = 2; +} + +// 使用token查询用户信息响应结构 +message UserResponse { + string user_id = 1; //ID + string username = 2; //用户名 + string nickname = 3; //昵称 + string mobile = 4; //手机 + string email = 5; //邮箱 + string status = 6; //状态 + string business_id = 7; //所属商家 + string store_id = 8; //门店 + string fans_to = 9;//被谁锁粉 + string is_vip = 10;//是否VIP + string usercode = 11;//工号 + string group_id = 12;//分组ID +} + +// 用户登录请求结构 +message LoginRequest { + string dbname = 1; + string account = 2; + string password = 3; +} + + +// 使用token查询用户信息响应结构 +message LoginResponse { + string user_id = 1; //ID + string username = 2; //用户名 + string nickname = 3; //昵称 + string mobile = 4; //手机 + string email = 5; //邮箱 + string status = 6; //状态 + string business_id = 7; //所属商家 + string store_id = 8; //门店 + string fans_to = 9;//被谁锁粉 + string is_vip = 10;//是否VIP + string token = 11;//token + string usercode = 12;//工号 + string group_id = 13;//分组ID +} + + +// rpc方法 +service UserService { + rpc getByToken (UserRequest) returns (UserResponse); // 使用token查询用户 + rpc login (LoginRequest) returns (LoginResponse); // 登录 +} \ No newline at end of file