发布新版

coupon.pb.go

@@ -100,6 +100,7 @@ func init() {
 type CouponService interface {
 	IsAvailable(in *CouponRequest, out *CouponResponse) error
 	Use(in *CouponRequest, out *CouponResponse) error
+	GetCouponInfo(in *CouponRequest, out *CouponResponse) error
 }
 
 // AcceptCouponServiceClient accepts connections on the listener and serves requests
@@ -177,6 +178,9 @@ func (c *CouponServiceClient) IsAvailable(in *CouponRequest, out *CouponResponse
 func (c *CouponServiceClient) Use(in *CouponRequest, out *CouponResponse) error {
 	return c.Call("CouponService.Use", in, out)
 }
+func (c *CouponServiceClient) GetCouponInfo(in *CouponRequest, out *CouponResponse) error {
+	return c.Call("CouponService.GetCouponInfo", in, out)
+}
 
 // DialCouponService connects to an CouponService at the specified network address.
 func DialCouponService(network, addr string) (*CouponServiceClient, *rpc.Client, error) {

coupon.proto

@@ -20,4 +20,5 @@ message CouponResponse {
 service CouponService {
     rpc isAvailable (CouponRequest) returns (CouponResponse);    // 优惠券是否可用
     rpc use (CouponRequest) returns (CouponResponse);    // 使用优惠券
-}
+    rpc GetCouponInfo (CouponRequest) returns (CouponResponse);      //获取优惠券信息
+}

data.go

@@ -0,0 +1,106 @@
+package couponrpc
+
+import (
+	"encoding/json"
+	"errors"
+	"strconv"
+	"time"
+
+	"git.tetele.net/tgo/crypter"
+	"github.com/golang/protobuf/proto"
+)
+
+func SetResData(data interface{}, res *CouponResponse) {
+	res_data_json, err := json.Marshal(data)
+	if err == nil {
+
+		encryData := crypter.DesEn(string(res_data_json), DES_KEY)
+
+		now_str := strconv.FormatInt(time.Now().Unix(), 10)
+
+		res_sign := Sign(encryData, now_str)
+
+		res.Data = proto.String(encryData)
+		res.Time = proto.String(now_str)
+		res.Sign = proto.String(res_sign)
+	}
+
+}
+
+func SetReqData(arg interface{}) (*CouponRequest, error) {
+	data_json, err := json.Marshal(arg)
+	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)
+
+	return &CouponRequest{proto.String(encryData), proto.String(now), proto.String(sign), nil}, nil
+
+}
+
+func GetReqData(req *CouponRequest) (string, error) {
+	res_data := req.GetData()
+
+	if res_data != "" {
+
+		time_int64, err := strconv.ParseInt(req.GetTime(), 10, 64)
+		if err != nil {
+			return "", err
+		}
+
+		now_int64 := time.Now().Unix()
+
+		if now_int64-time_int64 > 10 || time_int64-now_int64 > 10 {
+			//时间误差前后10秒，返回
+			return "", errors.New("返回时间错误")
+		}
+
+		check_sign := CheckSign(req.GetSign(), res_data, req.GetTime())
+		if !check_sign {
+			return "", errors.New("返回数据签名错误")
+		}
+
+		//解密
+		return crypter.DesDe(res_data, DES_KEY), nil
+
+	}
+
+	return "", nil
+}
+
+func GetResData(res *CouponResponse) (string, error) {
+	res_data := res.GetData()
+
+	if res_data != "" {
+
+		time_int64, err := strconv.ParseInt(res.GetTime(), 10, 64)
+		if err != nil {
+			return "", err
+		}
+
+		now_int64 := time.Now().Unix()
+
+		if now_int64-time_int64 > 10 || time_int64-now_int64 > 10 {
+			//时间误差前后10秒，返回
+			return "", errors.New("返回时间错误")
+		}
+
+		check_sign := CheckSign(res.GetSign(), res_data, res.GetTime())
+		if !check_sign {
+			return "", errors.New("返回数据签名错误")
+		}
+
+		//解密
+		return crypter.DesDe(res_data, DES_KEY), nil
+
+	}
+
+	return "", nil
+}

get_coupon.go

@@ -0,0 +1,55 @@
+package couponrpc
+
+import (
+	"encoding/json"
+	"errors"
+)
+
+func GetCouponInfo(dbname,couponId string, url ...string) (map[string]string, error) {
+
+	if dbname == "" {
+		return nil, errors.New("参数错误")
+	}
+
+	conn, err := rpc_server_conn(url...)
+
+	if err != nil {
+		return nil, err
+	}
+	defer conn.Close()
+
+	arg := map[string]string{
+		"dbname": dbname,
+		"coupon_id": couponId,
+	}
+
+	req, err := SetReqData(arg)
+	if err != nil {
+		return nil, err
+	}
+	res := &CouponResponse{}
+
+	err = conn.GetCouponInfo(req, res)
+
+	if err != nil {
+		return nil, err
+	}
+
+	res_data_de, err := GetResData(res)
+	if err != nil {
+		return nil, err
+	}
+
+	var couponInfo map[string]string
+	err = json.Unmarshal([]byte(res_data_de),&couponInfo)
+
+	if err != nil {
+		return nil, err
+	}
+
+	if res_data_de == "" {
+		return nil, nil
+	}
+
+	return couponInfo, nil
+}

get_coupon_test.go

@@ -0,0 +1,13 @@
+package couponrpc
+
+import "testing"
+
+func Test_getCoupon(t *testing.T) {
+
+	dbname := "shop_v2"
+	couponId := "4"
+	res, err := GetCouponInfo(dbname, couponId)
+	t.Log(res)
+	t.Log(err)
+
+}