go语言简单快速的按顺序遍历kv结构(map)

文章目录

需求描述
- 用map实现
- 按照map的key排序
- 用二维切片实现
- 用结构体实现

需求描述

在go语言中，如果需要对map遍历，每次输出的顺序是不固定的，可以考虑存储为二维切片或结构体。
假设现在需要在页面的下拉菜单中展示一些基础的选项，需要服务端输出如下的结构：

{"code": 200000,"message": "请求成功","data": [{"id": 1,"name": "中国"},{"id": 2,"name": "美国"},{"id": 3,"name": "韩国"},{"id": 4,"name": "新加坡"}]
}

用map实现

首先，常规思维，考虑用map存储，实现方式如下：

先定义好常量值和映射关系，代码路径：gozero/internal/constants/commonConstants.go

package constantsconst (// 国家常量值CountryChina     = 1CountryAmerica   = 2CountryKorea     = 3CountrySingapore = 4
)var (//国家常量映射关系CountryMap = map[int]string{CountryChina:     "中国",CountryAmerica:   "美国",CountryKorea:     "韩国",CountrySingapore: "新加坡",}
)

然后在逻辑层使用上面定义的map：

代码路径：gozero/internal/logic/common/simpleselectlogic.go

func (l *SimpleSelectLogic) SimpleSelect(req *types.SimpleSelectRequest) (resp *utils.Response, err error) {paramType := req.Typevar data interface{}switch paramType {case "country":data = mapToSlice(constants.CountryMap)case "others":data = nil}//成功返回return utils.SuccessResponse(data), nil
}// 定义结构体
type Item struct {ID   int    `json:"id"`Name string `json:"name"`
}func mapToSlice(m map[int]string) []Item {// 提取所有的keykeys := make([]int, 0, len(m))for k := range m {keys = append(keys, k)}// 将key和value组合成结构体切片sortedSlice := make([]Item, 0, len(m))for _, k := range keys {sortedSlice = append(sortedSlice, Item{ID: k, Name: m[k]})}return sortedSlice
}

通过上面的方式已经可以输出我们想要的结构了：

但是，多运行几次就会发现，每次运行后的顺序并不一致，因为go中的map的遍历是不保证顺序的。

那么，直接在遍历的时候对key进行排序再按照key的顺序输出是否可行？尝试如下：

按照map的key排序

在 mapToSlice方法中加上对key的排序：

func mapToSortedSlice(m map[int]string) []Item {// 提取所有的keykeys := make([]int, 0, len(m))for k := range m {keys = append(keys, k)}// 对key进行排序sort.Ints(keys)// 根据排序后的key顺序，将key和value组合成结构体切片sortedSlice := make([]Item, 0, len(m))for _, k := range keys {sortedSlice = append(sortedSlice, Item{ID: k, Name: m[k]})}return sortedSlice
}

这样修改后，可以保证输出的都是按照key由小到大排序的结果。

但是，假如现在有一种情况，产品要求在已有的下拉选项中插入一个新的选项值，并且顺序在中间。比如加入一个“英国”的选项，在“美国”后面，我们修改常量枚举值如下：

package constantsconst (// 国家常量值CountryChina     = 1CountryAmerica   = 2CountryEngland   = 5 //增加的"英国"的枚举值CountryKorea     = 3CountrySingapore = 4
)var (//国家常量映射关系CountryMap = map[int]string{CountryChina:     "中国",CountryAmerica:   "美国",CountryEngland:   "英国", //增加的"英国"排在"美国"下面CountryKorea:     "韩国",CountrySingapore: "新加坡",}
)

按照上面的方法，运行后会发现，新增加的枚举值排在了最后面：

这是因为在mapToSortedSlice方法中根据map的key排序后，后来新增的key是5，所以会排在最后面。由于已有的key（1-4）不能修改，那么只能考虑再定义一个排序的切片来自定义需要排序的数据：

CountrySort = []int{CountryChina,CountryAmerica,CountryEngland,CountryKorea,CountrySingapore,}

然后在 mapToSortedSlice 方法中对上面的CountrySort进行排序后遍历。这种方法可以实现需求，但是会比较麻烦。因此，我决定改为使用二维切片来存储数据。

用二维切片实现

package constantsconst (// 国家常量值--改为存储字符串CountryStrChina     = "1"CountryStrAmerica   = "2"CountryStrEngland   = "5"CountryStrKorea     = "3"CountryStrSingapore = "4"
)var (//国家常量映射关系--切片存储CountrySlice = [][]string{{CountryStrChina, "中国"},{CountryStrAmerica, "美国"},{CountryStrEngland, "英国"},{CountryStrKorea, "韩国"},{CountryStrSingapore, "新加坡"},}
)

然后，处理切片的函数如下：

func formatSlice(m [][]string) []Item {// 根据排序后的key顺序，将key和value组合成结构体切片sortedSlice := make([]Item, 0, len(m))for _, v := range m {id, _ := strconv.Atoi(v[0])sortedSlice = append(sortedSlice, Item{ID: id, Name: v[1]})}return sortedSlice
}

调用一下：

func (l *SimpleSelectLogic) SimpleSelect(req *types.SimpleSelectRequest) (resp *utils.Response, err error) {paramType := req.Typevar data interface{}switch paramType {case "country"://data = mapToSortedSlice(constants.CountryMap)data = formatSlice(constants.CountrySlice)case "others":data = nil}//成功返回return utils.SuccessResponse(data), nil
}

输出结果验证一下，会发现按照我们定义二维切片的顺序输出了：

至此，问题解决。

如果你也需要在go中按顺序遍历kv结构，并且想偷个懒，那么不妨试试这样的方式。如果还想再偷懒，那么可以连常量都省去了，直接这么写：

//国家常量映射关系--切片存储-省去常量
CountrySliceSimple = [][]string{{"1", "中国"},{"2", "美国"},{"5", "英国"},{"3", "韩国"},{"4", "新加坡"},
}

用结构体实现

当然，也可以直接在定义的时候存储为结构体：

package constantsconst (// 国家常量值CountryChina     = 1CountryAmerica   = 2CountryEngland   = 5 //增加的"英国"的枚举值CountryKorea     = 3CountrySingapore = 4//国家常量映射关系--结构体存储CountryStruct = []struct {Id   int    `json:"id"`Name string `json:"name"`}{{CountryChina, "中国"},{CountryAmerica, "美国"},{CountryEngland, "英国"},{999, "法国"},{CountryKorea, "韩国"},{CountrySingapore, "新加坡"},}
)

使用的时候也不需要转换了：

func (l *SimpleSelectLogic) SimpleSelect(req *types.SimpleSelectRequest) (resp *utils.Response, err error) {paramType := req.Typevar data interface{}switch paramType {case "country"://data = mapToSortedSlice(constants.CountryMap) //使用map//data = formatSlice(constants.CountrySlice) //使用切片data = constants.CountryStruct //使用结构体case "others":data = nil}//成功返回return utils.SuccessResponse(data), nil
}