init

src/util/sign/sign.go

@@ -0,0 +1,117 @@
+package sign
+
+import (
+	"errors"
+	"github.com/assimon/luuu/util/json"
+	"github.com/gookit/goutil/strutil"
+	"reflect"
+	"sort"
+	"strconv"
+)
+
+// Get 获取签名
+func Get(data interface{}, bizKey string) (string, error) {
+	var err error
+	signStr := ""
+	switch v := reflect.ValueOf(data); v.Kind() {
+	case reflect.Map:
+		signStr, err = MapToParams(data.(map[string]interface{}))
+		if err != nil {
+			return "", err
+		}
+	case reflect.Struct:
+		signStr, err = Struct2map(v.Interface())
+		if err != nil {
+			return "", err
+		}
+	default:
+		return "", errors.New("type err")
+	}
+	sign := strutil.Md5(signStr + bizKey)
+	return sign, nil
+}
+
+func Struct2map(content interface{}) (string, error) {
+	var params map[string]interface{}
+	marshal, err := json.Cjson.Marshal(content)
+	if err != nil {
+		return "", err
+	}
+	if err = json.Cjson.Unmarshal(marshal, &params); err != nil {
+		return "", err
+	}
+	paramsUrl, err := MapToParams(params)
+	return paramsUrl, err
+}
+
+func MapToParams(params map[string]interface{}) (string, error) {
+	var tempArr []string
+	temString := ""
+	for k, v := range params {
+		if k == "signature" {
+			continue
+		}
+		if v == nil {
+			continue
+		}
+		fv := ""
+		switch v.(type) {
+		case float64:
+			ft := v.(float64)
+			fv = strconv.FormatFloat(ft, 'f', -1, 64)
+		case float32:
+			ft := v.(float32)
+			fv = strconv.FormatFloat(float64(ft), 'f', -1, 64)
+		case int:
+			it := v.(int)
+			fv = strconv.Itoa(it)
+		case uint:
+			it := v.(uint)
+			fv = strconv.Itoa(int(it))
+		case int8:
+			it := v.(int8)
+			fv = strconv.Itoa(int(it))
+		case uint8:
+			it := v.(uint8)
+			fv = strconv.Itoa(int(it))
+		case int16:
+			it := v.(int16)
+			fv = strconv.Itoa(int(it))
+		case uint16:
+			it := v.(uint16)
+			fv = strconv.Itoa(int(it))
+		case int32:
+			it := v.(int32)
+			fv = strconv.Itoa(int(it))
+		case uint32:
+			it := v.(uint32)
+			fv = strconv.Itoa(int(it))
+		case int64:
+			it := v.(int64)
+			fv = strconv.FormatInt(it, 10)
+		case uint64:
+			it := v.(uint64)
+			fv = strconv.FormatUint(it, 10)
+		case string:
+			fv = v.(string)
+		case []byte:
+			fv = string(v.([]byte))
+		default:
+			return "", errors.New("signature marshal error")
+		}
+		// 空值不参与签名
+		if fv == "" {
+			continue
+		}
+		tempArr = append(tempArr, k+"="+fv)
+	}
+	sort.Strings(tempArr)
+	for n, v := range tempArr {
+		if n+1 < len(tempArr) {
+			temString = temString + v + "&"
+		} else {
+			temString = temString + v
+		}
+	}
+	return temString, nil
+}