添加map排序函数

maputils/maputils.go

@@ -101,3 +101,48 @@ func GetValueByKey[K comparable, V any](m map[K]any, key K) (V, bool) {
 
 	return v, true
 }
+
+// SortMap 可以排序的map
+// 模板参数:
+// K是map的key的类型，V是map值的类型
+type SortMap[K comparable, V any] struct {
+	sortedKeys []K
+	m          map[K]V
+}
+
+// Range 遍历排序后的map
+// 模板参数:
+// K是map的key的类型，V是map值的类型
+// 参数：
+// callback: 遍历回调
+func (sortMap *SortMap[K, V]) Range(callback func(index int, key K, value V)) {
+	if callback != nil {
+		for i, key := range sortMap.sortedKeys {
+			callback(i, key, sortMap.m[key])
+		}
+	}
+}
+
+// Sort 排序map
+// 模板参数:
+// K是map的key的类型，V是map值的类型
+// 参数：
+// m: 需要排序的map
+// sortKeysFunc: 对key进行排序的回调函数
+// 返回值：
+// SortMap指针,之后可以调用Range遍历排序后的map
+func Sort[K comparable, V any](m map[K]V, sortKeysFunc func(keys []K)) *SortMap[K, V] {
+	keys := make([]K, 0)
+	for key := range m {
+		keys = append(keys, key)
+	}
+
+	if sortKeysFunc != nil {
+		sortKeysFunc(keys)
+	}
+
+	return &SortMap[K, V]{
+		sortedKeys: keys,
+		m:          m,
+	}
+}

maputils/maputils_test.go

@@ -1,6 +1,10 @@
 package maputils
 
-import "testing"
+import (
+	"sort"
+	"strconv"
+	"testing"
+)
 
 func TestGetValueByPath(t *testing.T) {
 	testMap := map[string]any{
@@ -57,3 +61,29 @@ func TestGetValueByKey(t *testing.T) {
 		t.Fatal("值不一致")
 	}
 }
+
+func TestSort(t *testing.T) {
+	testMap := map[int]any{
+		1: "test1",
+		2: "test2",
+		3: "test3",
+		4: "test4",
+		5: "test5",
+		6: "test6",
+		7: "test7",
+		8: "test8",
+		9: "test9",
+	}
+
+	Sort(testMap, func(keys []int) {
+		sort.Sort(sort.IntSlice(keys))
+	}).Range(func(index int, key int, value any) {
+		if index != key-1 {
+			t.Fatal("key顺序错误")
+		}
+
+		if value != "test"+strconv.Itoa(key) {
+			t.Fatal("key对应的value不正确")
+		}
+	})
+}