yuxh
84c58c3756
- 新增 AddHeaders 方法支持批量添加请求头参数 - 新增 RefererHeader 结构体用于管理 Referer 和 TraceId - 添加随机字符串生成工具包 random - 集成 MD5 加密和追踪 ID 生成功能 - 修改 IP 服务域名从 ip.gaore.com 到 ip.gaore.com.hk - 在客户端请求中自动添加 Referer 和 Traceparent 头信息
117 lines
2.6 KiB
Go
117 lines
2.6 KiB
Go
package utils
|
|
|
|
import (
|
|
"crypto/md5"
|
|
"crypto/rand"
|
|
"encoding/hex"
|
|
"fmt"
|
|
"golib.gaore.com/GaoreGo/haiwai-common-sdk-go/sdk/utils/random"
|
|
"hash"
|
|
"net/url"
|
|
"reflect"
|
|
"sort"
|
|
"strconv"
|
|
"strings"
|
|
"time"
|
|
)
|
|
|
|
type UUID [16]byte
|
|
|
|
const letterBytes = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ"
|
|
|
|
func GetUUID() (uuidHex string) {
|
|
uuid := NewUUID()
|
|
uuidHex = hex.EncodeToString(uuid[:])
|
|
return
|
|
}
|
|
|
|
func NewUUID() UUID {
|
|
ns := UUID{}
|
|
safeRandom(ns[:])
|
|
u := newFromHash(md5.New(), ns, RandStringBytes(16))
|
|
u[6] = (u[6] & 0x0f) | (byte(2) << 4)
|
|
u[8] = (u[8]&(0xff>>2) | (0x02 << 6))
|
|
|
|
return u
|
|
}
|
|
|
|
func RandStringBytes(n int) string {
|
|
return random.StrRandom(int64(n))
|
|
}
|
|
|
|
func newFromHash(h hash.Hash, ns UUID, name string) UUID {
|
|
u := UUID{}
|
|
h.Write(ns[:])
|
|
h.Write([]byte(name))
|
|
copy(u[:], h.Sum(nil))
|
|
|
|
return u
|
|
}
|
|
|
|
func safeRandom(dest []byte) {
|
|
if _, err := rand.Read(dest); err != nil {
|
|
panic(err)
|
|
}
|
|
}
|
|
|
|
func GetUrlFormedMap(source map[string]string) (urlEncoded string) {
|
|
urlEncoder := url.Values{}
|
|
for key, value := range source {
|
|
urlEncoder.Add(key, value)
|
|
}
|
|
urlEncoded = urlEncoder.Encode()
|
|
return
|
|
}
|
|
|
|
func GetUrlByKeySort(source map[string]string) (url string) {
|
|
keys := make([]string, 0, len(source))
|
|
for k := range source {
|
|
keys = append(keys, k)
|
|
}
|
|
sort.Strings(keys)
|
|
for _, key := range keys {
|
|
url += fmt.Sprintf("%s=%s&", key, source[key])
|
|
}
|
|
return strings.TrimRight(url, "&")
|
|
}
|
|
|
|
func InitStructWithDefaultTag(bean interface{}) {
|
|
beantype := reflect.TypeOf(bean)
|
|
for i := 0; i < beantype.Elem().NumField(); i++ {
|
|
field := beantype.Elem().Field(i)
|
|
defaultValue := strings.TrimSpace(field.Tag.Get("default"))
|
|
if defaultValue == "" || defaultValue == "-" {
|
|
continue
|
|
}
|
|
setter := reflect.ValueOf(bean).Elem().Field(i)
|
|
fieldTypeName := field.Type.String()
|
|
switch fieldTypeName {
|
|
case "int", "int64", "int32", "int8", "int16":
|
|
intval, _ := strconv.ParseInt(defaultValue, 10, 64)
|
|
setter.SetInt(intval)
|
|
case "uint", "uint8", "uint16", "uint32", "uint64", "uintptr":
|
|
uintval, _ := strconv.ParseUint(defaultValue, 10, 64)
|
|
setter.SetUint(uintval)
|
|
case "string":
|
|
setter.SetString(defaultValue)
|
|
case "time.Duration":
|
|
intval, _ := strconv.ParseInt(defaultValue, 10, 64)
|
|
setter.SetInt(intval * int64(time.Second))
|
|
case "bool":
|
|
boolval, _ := strconv.ParseBool(defaultValue)
|
|
setter.SetBool(boolval)
|
|
default:
|
|
fmt.Println(field.Type.String(), field.Name)
|
|
}
|
|
}
|
|
}
|
|
|
|
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))
|
|
}
|