Created

mapx/map.go

@@ -0,0 +1,52 @@
+package mapx
+
+import (
+	"net/url"
+	"reflect"
+	"sort"
+	"strconv"
+)
+
+// Keys 获取 map 键值（默认按照升序排列）
+func Keys(m interface{}) []string {
+	var keys []string
+	vo := reflect.ValueOf(m)
+	if vo.Kind() == reflect.Map {
+		mapKeys := vo.MapKeys()
+		keys = make([]string, len(mapKeys))
+		for k, v := range mapKeys {
+			var vString string
+			switch v.Type().Kind() {
+			case reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32, reflect.Int64:
+				vString = strconv.FormatInt(v.Int(), 10)
+			case reflect.Float32, reflect.Float64:
+				vString = strconv.FormatFloat(v.Float(), 'f', -1, 64)
+			case reflect.Bool:
+				if v.Bool() {
+					vString = "1"
+				} else {
+					vString = "0"
+				}
+			default:
+				vString = v.String()
+			}
+			keys[k] = vString
+		}
+		if len(keys) > 0 {
+			sort.Strings(keys)
+		}
+	}
+	return keys
+}
+
+func StringMapStringEncode(params map[string]string) string {
+	if len(params) == 0 {
+		return ""
+	}
+
+	values := url.Values{}
+	for k, v := range params {
+		values.Add(k, v)
+	}
+	return values.Encode()
+}

mapx/map_test.go

@@ -0,0 +1,60 @@
+package mapx
+
+import (
+	"fmt"
+	"github.com/stretchr/testify/assert"
+	"testing"
+)
+
+func TestKeys(t *testing.T) {
+	tests := []struct {
+		tag   string
+		value interface{}
+		keys  []string
+	}{
+		{"t0", nil, nil},
+		{"t1", map[string]interface{}{"a": 1, "b": 2}, []string{"a", "b"}},
+		{"t2", map[string]interface{}{"b": 1, "a": 2}, []string{"a", "b"}},
+		{"t3", map[string]interface{}{"a": 1, "b": 2, "": 3}, []string{"", "a", "b"}},
+		{"t4", map[string]string{"a": "1", "b": "2", "": "3"}, []string{"", "a", "b"}},
+		{"t4", map[int]string{1: "1", 3: "3", 2: "2"}, []string{"1", "2", "3"}},
+		{"t4", map[float64]string{1.1: "1", 3: "3", 2: "2"}, []string{"1.1", "2", "3"}},
+		{"t4", map[bool]string{true: "1", false: "3"}, []string{"0", "1"}},
+	}
+
+	for _, test := range tests {
+		keys := Keys(test.value)
+		v := assert.Equal(t, test.keys, keys, test.tag)
+		if v {
+			for k, value := range test.keys {
+				assert.Equal(t, value, keys[k], fmt.Sprintf("keys[%d]", k))
+			}
+		}
+	}
+}
+
+func BenchmarkKeys(b *testing.B) {
+	for i := 0; i < b.N; i++ {
+		Keys(map[interface{}]interface{}{"a": 1, "b": 2, "c": "cValue", "d": "dValue", 1: 1, 2: 2})
+	}
+}
+
+func TestStringMapStringEncode(t *testing.T) {
+	tests := []struct {
+		tag      string
+		value    map[string]string
+		expected string
+	}{
+		{"t0", nil, ""},
+		{"t1", map[string]string{"a": "1", "b": "2"}, "a=1&b=2"},
+		{"t2", map[string]string{"b": "1", "a": "2"}, "a=2&b=1"},
+		{"t3", map[string]string{"a": "1", "b": "2", "c": "3"}, "a=1&b=2&c=3"},
+		{"t4", map[string]string{"a": "1", "b": "2", "": "3"}, "=3&a=1&b=2"},
+		{"t4", map[string]string{"1": "1", "3": "3", "2": "2"}, "1=1&2=2&3=3"},
+	}
+
+	for _, test := range tests {
+		s := StringMapStringEncode(test.value)
+		assert.Equal(t, test.expected, s, test.tag)
+	}
+}