Create & Init Project...

app/job/main/ugcpay/service/pay/BUILD

@@ -0,0 +1,45 @@
+package(default_visibility = ["//visibility:public"])
+
+load(
+    "@io_bazel_rules_go//go:def.bzl",
+    "go_test",
+    "go_library",
+)
+
+go_test(
+    name = "go_default_test",
+    srcs = ["pay_test.go"],
+    embed = [":go_default_library"],
+    rundir = ".",
+    tags = ["automanaged"],
+    deps = [
+        "//app/service/main/ugcpay/conf:go_default_library",
+        "//vendor/github.com/smartystreets/goconvey/convey:go_default_library",
+    ],
+)
+
+go_library(
+    name = "go_default_library",
+    srcs = ["pay.go"],
+    importpath = "go-common/app/job/main/ugcpay/service/pay",
+    tags = ["automanaged"],
+    visibility = ["//visibility:public"],
+    deps = [
+        "//library/log:go_default_library",
+        "//vendor/github.com/pkg/errors:go_default_library",
+    ],
+)
+
+filegroup(
+    name = "package-srcs",
+    srcs = glob(["**"]),
+    tags = ["automanaged"],
+    visibility = ["//visibility:private"],
+)
+
+filegroup(
+    name = "all-srcs",
+    srcs = [":package-srcs"],
+    tags = ["automanaged"],
+    visibility = ["//visibility:public"],
+)

app/job/main/ugcpay/service/pay/pay.go

@@ -0,0 +1,196 @@
+package pay
+
+import (
+	"bytes"
+	"crypto/md5"
+	"encoding/hex"
+	"encoding/json"
+	"fmt"
+	"net/url"
+	"strconv"
+	"time"
+
+	"github.com/pkg/errors"
+	"go-common/library/log"
+)
+
+// Pay is.
+type Pay struct {
+	ID                     string
+	Token                  string
+	RechargeShellNotifyURL string
+}
+
+// TraceID .
+func (p *Pay) TraceID() string {
+	return strconv.FormatInt(time.Now().UnixNano(), 10)
+}
+
+// RechargeShellReq .
+type RechargeShellReq struct {
+	CustomerID  string              `json:"customerId"`
+	ProductName string              `json:"productName"`
+	Rate        string              `json:"rate"`
+	NotifyURL   string              `json:"notifyUrl"`
+	Timestamp   int64               `json:"timestamp"`
+	SignType    string              `json:"signType"`
+	Sign        string              `json:"sign"`
+	Data        []RechargeShellData `json:"data"`
+}
+
+// RechargeShellData .
+type RechargeShellData struct {
+	ThirdOrderNo string `json:"thirdOrderNo"`
+	MID          int64  `json:"mid"`
+	ThirdCoin    string `json:"thirdCoin"`
+	Brokerage    string `json:"brokerage"`
+	ThirdCtime   int64  `json:"thirdCtime"`
+}
+
+// RechargeShell 转入贝壳
+func (p *Pay) RechargeShell(orderID string, mid int64, assetBP int64, shell int64) (params url.Values, jsonData string, err error) {
+	var (
+		productName = "UGC付费"
+		rate        = "1.00"
+		timestamp   = time.Now().Unix() * 1000
+		thirdCoin   = float64(assetBP) / 100
+		brokerage   = float64(shell) / 100
+	)
+
+	params = make(url.Values)
+	params.Set("customerId", p.ID)
+	params.Set("productName", productName)
+	params.Set("rate", rate)
+	params.Set("notifyUrl", p.RechargeShellNotifyURL)
+	params.Set("timestamp", strconv.FormatInt(timestamp, 10))
+	params.Set("data", fmt.Sprintf("[{brokerage=%.2f&mid=%d&thirdCoin=%.2f&thirdCtime=%d&thirdOrderNo=%s}]", brokerage, mid, thirdCoin, timestamp, orderID))
+	p.Sign(params)
+
+	data := RechargeShellData{
+		ThirdOrderNo: orderID,
+		MID:          mid,
+		ThirdCoin:    fmt.Sprintf("%.2f", thirdCoin),
+		Brokerage:    fmt.Sprintf("%.2f", brokerage),
+		ThirdCtime:   timestamp,
+	}
+	req := RechargeShellReq{
+		CustomerID:  p.ID,
+		ProductName: productName,
+		Rate:        rate,
+		NotifyURL:   p.RechargeShellNotifyURL,
+		Timestamp:   timestamp,
+		SignType:    params.Get("signType"),
+		Sign:        params.Get("sign"),
+		Data:        []RechargeShellData{data},
+	}
+
+	payBytes, err := json.Marshal(req)
+	if err != nil {
+		err = errors.Wrapf(err, "pay.RechargeShell.ToJSON : %s", params.Encode())
+		return
+	}
+	jsonData = string(payBytes)
+	return
+}
+
+// CheckOrder 对账param
+func (p Pay) CheckOrder(txID string) (params url.Values) {
+	params = make(url.Values)
+	params.Set("customerId", p.ID)
+	params.Set("txIds", txID)
+	return
+}
+
+// CheckRefundOrder 退款对账param
+func (p Pay) CheckRefundOrder(txID string) (params url.Values) {
+	params = make(url.Values)
+	params.Set("customerId", p.ID)
+	params.Set("txIds", txID)
+	return
+}
+
+// Query 返回订单查询param
+func (p *Pay) Query(orderID string) (params url.Values) {
+	params = make(url.Values)
+	params.Set("customerId", p.ID)
+	params.Set("orderIds", orderID)
+	params.Set("timestamp", strconv.FormatInt(time.Now().Unix()*1000, 10))
+	params.Set("traceId", p.TraceID())
+	params.Set("version", "1.0")
+	return
+}
+
+// ToJSON param to json
+func (p *Pay) ToJSON(params url.Values) (j string, err error) {
+	var (
+		payBytes []byte
+		pmap     = make(map[string]string)
+	)
+	for k, v := range params {
+		if len(v) > 0 {
+			pmap[k] = v[0]
+		}
+	}
+	if payBytes, err = json.Marshal(pmap); err != nil {
+		err = errors.Wrapf(err, "pay.ToJSON : %s", params.Encode())
+		return
+	}
+	j = string(payBytes)
+	return
+}
+
+// DeviceType 支付平台DeviceType
+func (p *Pay) DeviceType(platform string) (t int64) {
+	// 支付设备渠道类型，  1 pc 2 webapp 3 app 4jsapi 5 server 6小程序支付 7聚合二维码支付
+	switch platform {
+	case "ios", "android":
+		return 3
+	default:
+		return 1
+	}
+}
+
+// Sign 支付平台接口签名
+func (p *Pay) Sign(params url.Values) (err error) {
+	params.Set("signType", "MD5")
+	sortedStr := params.Encode()
+	if sortedStr, err = url.QueryUnescape(sortedStr); err != nil {
+		return
+	}
+	b := bytes.Buffer{}
+	b.WriteString(sortedStr)
+	b.WriteString("&token=" + p.Token)
+	signMD5 := md5.Sum(b.Bytes())
+	sign := hex.EncodeToString(signMD5[:])
+	params.Set("sign", sign)
+	return
+}
+
+// Verify 支付平台返回param校验
+func (p *Pay) Verify(params url.Values) (ok bool) {
+	var (
+		rs = params.Get("sign")
+		s  string
+	)
+	ok = false
+	defer func() {
+		if !ok {
+			params.Set("sign", rs)
+			log.Error("Verify pay sign error, expect : %s, actual : %s, params : %s", s, rs, params.Encode())
+		}
+	}()
+	if rs == "" {
+		return
+	}
+	params.Del("sign")
+	if err := p.Sign(params); err != nil {
+		log.Error("Verify pay sign error : %+v", err)
+		return
+	}
+	s = params.Get("sign")
+	if rs == s {
+		ok = true
+		return
+	}
+	return
+}

app/job/main/ugcpay/service/pay/pay_test.go

@@ -0,0 +1,131 @@
+package pay
+
+import (
+	"encoding/json"
+	"flag"
+	"net/url"
+	"os"
+	"testing"
+
+	"go-common/app/service/main/ugcpay/conf"
+
+	. "github.com/smartystreets/goconvey/convey"
+)
+
+var (
+	p *Pay
+)
+
+func TestMain(m *testing.M) {
+	flag.Set("conf", "../../cmd/test.toml")
+	if err := conf.Init(); err != nil {
+		panic(err)
+	}
+	p = &Pay{
+		ID:                     conf.Conf.Biz.Pay.ID,
+		Token:                  conf.Conf.Biz.Pay.Token,
+		RechargeShellNotifyURL: "http://api.bilibili.co/x/internal/ugcpay/trade/recharge/callback",
+	}
+
+	m.Run()
+	os.Exit(0)
+}
+
+func TestCheckOrder(t *testing.T) {
+	Convey("", t, func() {
+		param := p.CheckOrder("3059753508505497600")
+		p.Sign(param)
+		t.Log(p.ToJSON(param))
+	})
+}
+
+func TestCheckRefundOrder(t *testing.T) {
+	Convey("", t, func() {
+		param := p.CheckRefundOrder("3059753508505497600")
+		p.Sign(param)
+		t.Log(p.ToJSON(param))
+	})
+}
+
+func TestRechargeShell(t *testing.T) {
+	var (
+		orderID = "123"
+		mid     = int64(46333)
+		assetBP = int64(1)
+		shell   = int64(1)
+	)
+	Convey("", t, func() {
+		_, json, err := p.RechargeShell(orderID, mid, assetBP, shell)
+		So(err, ShouldBeNil)
+		t.Log(json)
+	})
+}
+
+func TestSign(t *testing.T) {
+	Convey("", t, func() {
+		var (
+			param = url.Values{
+				"customerId":      []string{"10017"},
+				"deviceType":      []string{"3"},
+				"notifyUrl":       []string{"http://api.bilibili.co/x/internal/ugcpay/trade/pay/callback"},
+				"orderCreateTime": []string{"1539935981000"},
+				"orderExpire":     []string{"1800"},
+				"orderId":         []string{"224"},
+				"originalAmount":  []string{"2000"},
+				"payAmount":       []string{"2000"},
+				"productId":       []string{"10110688"},
+				"serviceType":     []string{"99"},
+				"showTitle":       []string{"传点什么好呢？"},
+				"timestamp":       []string{"1539935981000"},
+				"traceId":         []string{"1539935981967342977"},
+				"uid":             []string{"27515244"},
+				"version":         []string{"1.0"},
+				"feeType":         []string{"CNY"},
+			}
+		)
+		err := p.Sign(param)
+		So(err, ShouldBeNil)
+
+		pmap := make(map[string]string)
+		var payBytes []byte
+		for k, v := range param {
+			if len(v) > 0 {
+				pmap[k] = v[0]
+			}
+		}
+		if payBytes, err = json.Marshal(pmap); err != nil {
+			return
+		}
+		t.Log(string(payBytes))
+	})
+}
+
+func TestSignVerify(t *testing.T) {
+	Convey("", t, func() {
+		var (
+			param = url.Values{
+				"customerId":      []string{"10017"},
+				"deviceType":      []string{"3"},
+				"notifyUrl":       []string{"http://api.bilibili.co/x/internal/ugcpay/trade/pay/callback"},
+				"orderCreateTime": []string{"1539935981000"},
+				"orderExpire":     []string{"1800"},
+				"orderId":         []string{"15"},
+				"originalAmount":  []string{"2000"},
+				"payAmount":       []string{"2000"},
+				"productId":       []string{"10110688"},
+				"serviceType":     []string{"99"},
+				"showTitle":       []string{"传点什么好呢？"},
+				"timestamp":       []string{"1539935981000"},
+				"traceId":         []string{"1539935981967342977"},
+				"uid":             []string{"27515244"},
+				"version":         []string{"1.0"},
+				"feeType":         []string{"CNY"},
+			}
+		)
+		err := p.Sign(param)
+		So(err, ShouldBeNil)
+
+		ok := p.Verify(param)
+		So(ok, ShouldBeTrue)
+	})
+}