@ -0,0 +1,240 @@ | |||||
package order | |||||
import ( | |||||
"crypto/md5" | |||||
"encoding/hex" | |||||
"encoding/json" | |||||
"errors" | |||||
"log" | |||||
"strconv" | |||||
"strings" | |||||
"time" | |||||
"git.tetele.net/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 | |||||
} |
@ -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) | |||||
} |
@ -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 | |||||
} |
@ -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); // 创建订单 | |||||
} |