Merge branch 'refs/heads/master' into develop/yuxh/msdk

- 新增随机字符串生成函数，支持不同模式（字母、数字、字母数字、字母十六进制）
-重构 RandStringBytes 函数，使用新的随机字符串生成逻辑
- 添加 MakeTraceId 函数，用于生成追踪ID
- 更新 Client 的 GetRefererHeader 方法，自动添加追踪ID

sdk/client.go

@@ -46,6 +46,27 @@ type Client struct {
 	httpProxy      string
 	httpsProxy     string
 	noProxy        string
+
+	header *requests.RefererHeader
+}
+
+func (c *Client) SetRefererHeader(header *requests.RefererHeader) {
+	c.header = header
+}
+
+func (c *Client) GetRefererHeader() map[string]string {
+	var header *requests.RefererHeader
+	if c.header == nil {
+		header = &requests.RefererHeader{
+			TraceId: utils.MakeTraceId(),
+		}
+	} else {
+		header = c.header
+	}
+	return map[string]string{
+		"Referer":     header.Referer,
+		"Traceparent": header.TraceId,
+	}
 }
 
 func (client *Client) GetNoProxy() string {
@@ -218,7 +239,7 @@ func (client *Client) DoAction(request requests.AcsRequest, response responses.A
 }
 
 func (client *Client) DoActionWithSigner(request requests.AcsRequest, response responses.AcsResponse, signer auth.Signer) (err error) {
-
+	request.AddHeaders(client.GetRefererHeader())
 	httpRequest, err := client.buildRequestWithSigner(request, signer)
 	if err != nil {
 		return err

sdk/requests/request.go

@@ -56,6 +56,11 @@ type Host struct {
 	Func    func(string) string
 }
 
+type RefererHeader struct {
+	Referer string
+	TraceId string
+}
+
 var debug utils.Debug
 
 func init() {
@@ -95,6 +100,7 @@ type AcsRequest interface {
 	GetBodyReader() io.Reader
 
 	AddHeaderParam(key, value string)
+	AddHeaders(headers map[string]string)
 	addQueryParam(key, value string)
 	addFormParam(key, value string)
 	addJsonParam(string, any)
@@ -226,6 +232,12 @@ func (request *baseRequest) AddHeaderParam(key, val string) {
 	request.Headers[key] = val
 }
 
+func (request *baseRequest) AddHeaders(headers map[string]string) {
+	for key, val := range headers {
+		request.Headers[key] = val
+	}
+}
+
 func (request *baseRequest) addQueryParam(key, val string) {
 	request.QueryParams[key] = val
 }

sdk/utils/random/str_random.go

@@ -0,0 +1,69 @@
+package random
+
+import (
+	"math/rand"
+	"time"
+	"unsafe"
+)
+
+type Mode int // 随机数模式
+
+const (
+	Letter Mode = iota
+	Number
+	LetterNumber
+	LetterHex
+)
+
+const (
+	letters       = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ"
+	numbers       = "0123456789"
+	lettersHex    = "0123456789abcdef"
+	letterIdBit   = 6
+	letterIdxMask = 1<<letterIdBit - 1
+	letterIdxMax  = 63 / letterIdBit
+)
+
+func StrRandom(n int64) string {
+	return Random(n, Letter)
+}
+
+func NumRandom(n int64) string {
+	return Random(n, Number)
+}
+
+func Random(n int64, mode ...Mode) string {
+	var baseStr string
+	mode = append(mode, LetterNumber)
+	switch mode[0] {
+	case LetterHex:
+		baseStr = lettersHex
+	case Letter:
+		baseStr = letters
+	case Number:
+		baseStr = numbers
+	case LetterNumber:
+		fallthrough
+	default:
+		baseStr = letters + numbers
+	}
+	return random(n, baseStr)
+}
+
+func random(n int64, baseStr string) string {
+
+	var src = rand.NewSource(time.Now().UnixNano())
+	b := make([]byte, n)
+	for i, cache, remain := n-1, src.Int63(), letterIdxMax; i >= 0; {
+		if remain == 0 {
+			cache, remain = src.Int63(), letterIdxMax
+		}
+		if idx := int(cache & letterIdxMask); idx < len(baseStr) {
+			b[i] = baseStr[idx]
+			i--
+		}
+		cache >>= letterIdBit
+		remain--
+	}
+	return *(*string)(unsafe.Pointer(&b))
+}

sdk/utils/utils.go

@@ -5,8 +5,8 @@ import (
 	"crypto/rand"
 	"encoding/hex"
 	"fmt"
+	"golib.gaore.com/GaoreGo/gaore-common-sdk-go/sdk/utils/random"
 	"hash"
-	rand2 "math/rand"
 	"net/url"
 	"reflect"
 	"sort"
@@ -36,11 +36,7 @@ func NewUUID() UUID {
 }
 
 func RandStringBytes(n int) string {
-	b := make([]byte, n)
-	for i := range b {
-		b[i] = letterBytes[rand2.Intn(len(letterBytes))]
-	}
-	return string(b)
+	return random.StrRandom(int64(n))
 }
 
 func newFromHash(h hash.Hash, ns UUID, name string) UUID {
@@ -114,3 +110,7 @@ func Md5(data string) string {
 	s := md5.Sum([]byte(data))
 	return hex.EncodeToString(s[:])
 }
+
+func MakeTraceId() string {
+	return fmt.Sprintf("00-%s-%s-01", random.Random(32, random.LetterHex), random.Random(16, random.LetterHex))
+}

services/msdk/client_test.go

@@ -1,6 +1,8 @@
 package msdk
 
 import (
+	"golib.gaore.com/GaoreGo/gaore-common-sdk-go/sdk/requests"
+	"golib.gaore.com/GaoreGo/gaore-common-sdk-go/sdk/utils"
 	"testing"
 )
 
@@ -20,11 +22,17 @@ func TestClient_GetUserAttr(t *testing.T) {
 }
 
 func TestGetIdfa(t *testing.T) {
+	header := &requests.RefererHeader{
+		Referer: "https://www.gaore.com",
+		TraceId: utils.MakeTraceId(),
+	}
+	_ = header
 	client, err := NewClient()
 	if err != nil {
 		t.Error(err)
 		return
 	}
+	client.SetRefererHeader(header)
 	req := CreateGetIdfaReq()
 	req.ChannelId = 1
 	req.GameId = 3706

services/msdk/user.go

@@ -63,7 +63,7 @@ func CreateGetIdfaReq() *GetIdfaReq {
 	req := &GetIdfaReq{
 		RpcRequest: &requests.RpcRequest{},
 	}
-	req.InitWithApiInfo(HOST, VERSION, "/getIdfa.php")
+	req.InitWithApiInfo(HOST, VERSION, "/getidfa.php")
 	req.Method = requests.GET
 	return req
 }
@@ -173,7 +173,7 @@ func CreateGetImeiReq() *GetImeiReq {
 	req := &GetImeiReq{
 		RpcRequest: &requests.RpcRequest{},
 	}
-	req.InitWithApiInfo(HOST, VERSION, "/getImei.php")
+	req.InitWithApiInfo(HOST, VERSION, "/getimei.php")
 	req.Method = requests.GET
 	return req
 }
@@ -212,7 +212,7 @@ func CreateSetImeiReq() *SetImeiReq {
 	req := &SetImeiReq{
 		RpcRequest: &requests.RpcRequest{},
 	}
-	req.InitWithApiInfo(HOST, VERSION, "/setImei.php")
+	req.InitWithApiInfo(HOST, VERSION, "/setimei.php")
 	req.Method = requests.GET
 	return req
 }