wechat refund check & random

wechat/refund.go

@@ -3,6 +3,7 @@ package wechat
 import (
 	"bytes"
 	"encoding/xml"
+	"errors"
 	"fmt"
 
 	"git.chuangxin1.com/cx/util"
@@ -61,6 +62,23 @@ type WePayRefundReply struct {
 	CashFee       int      `xml:"cash_fee"`   // 现金支付金额
 }
 
+// PayRefundCheck check refund reply
+func PayRefundCheck(reply WePayRefundReply) (ok bool, err error) {
+	//  通信是否正确
+	if reply.ReturnCode == "FAIL" {
+		err = errors.New(reply.ReturnMsg)
+		return
+	}
+	// 交易是否成功
+	if reply.ResultCode == "FAIL" {
+		err = errors.New(reply.ErrCodeDes)
+		return
+	}
+
+	ok = true
+	return
+}
+
 // PayRefund 退款
 func PayRefund(config WePayConfig, order WePayRefund) (reply WePayRefundReply, err error) {
 	var data []byte

wechat/wechat.go

@@ -6,9 +6,11 @@ import (
 	"encoding/xml"
 	"errors"
 	"fmt"
+	"math/rand"
 	"sort"
 	"strings"
 	"sync"
+	"time"
 
 	"git.chuangxin1.com/cx/util"
 )
@@ -120,6 +122,20 @@ func Sign(req map[string]interface{}, key string) (sign string) {
 	return
 }
 
+// Random 指定长度的随机字符串(字母或数字)
+func Random(n int) string {
+	str := "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789"
+	bytes := []byte(str)
+	result := []byte{}
+
+	r := rand.New(rand.NewSource(time.Now().UnixNano()))
+	n1 := len(bytes)
+	for i := 0; i < n; i++ {
+		result = append(result, bytes[r.Intn(n1)])
+	}
+	return string(result)
+}
+
 func makeSignature(token, signature, timestamp, nonce string) bool {
 	sl := []string{token, timestamp, nonce}
 	sort.Strings(sl)