Golang中的 map 默认是 无序的 。

起因

最近项目中有这样一个需求:

根据用户当前的坐标点,获取该用户附近的预设城市名称。

这里有一个注意点是,假设这些支持的城市名称是预设的,所以就不能直接通过地图类api根据坐标点获取所在城市名称了。

想到的解决思路是:

  1. 获取这几个预设城市的坐标点
  2. App端获取用户当前坐标点
  3. 分别计算得到该用户坐标点距离各个预设城市的坐标点距离
  4. 然后计算得到其中距离最小的一项
  5. 这个坐标点对应的城市就是所求

探索

经过前期计算,在上面的第 3 步操作后我得到了下面的结果:

result := map[string]float64{  
   "城市A": 2334.20,  
 "城市B": 1992.33,  
 "城市C": 500.26,  
 "城市D": 10.39,  
 "城市E": 333.33,  
}

我们知道,Golang中 Map 是 无序的 。所以当我们使用 for-range 循环时:

for k, v := range result {  
   fmt.Printf("key: %v value: %v \n", k, v)  
}

结果可能是:

// 第一种可能结果:
key: 城市B value: 1992.33 
key: 城市C value: 500.26 
key: 城市D value: 10.39 
key: 城市E value: 333.33 
key: 城市A value: 2334.2 

// 第二种可能结果:
key: 城市E value: 333.33 
key: 城市A value: 2334.2 
key: 城市B value: 1992.33 
key: 城市C value: 500.26 
key: 城市D value: 10.39 

// 第三种可能结果:
key: 城市E value: 333.33 
key: 城市A value: 2334.2 
key: 城市B value: 1992.33 
key: 城市C value: 500.26 
key: 城市D value: 10.39

所以,我们不能按照 key 或者 value 来进行排序。

实现

但Golang中切片 Slice 是 有序的。 我们可以结果使用 Slice 来实现对 Map 的排序。

第一步

我们先将上面的 map 转换成一个 slice :

type KVPair struct {  
   Key string  
 Val float64  
}  
  
tmpList := make([]KVPair, 0)  
  
for k, v := range result {  
   tmpList = append(tmpList, KVPair{Key: k, Val: v})  
}

上面创建了一个 结构体切片 ,然后将 map 的值添加到切片中。

第二步

在 go1.8 之后,引入了 sort.Slice() 方法,可以实现对 slice 进行排序,我们只需要传入一个比较函数即可:

sort.Slice(tmpList, func(i, j int) bool {  
   return tmpList[i].Val < tmpList[j].Val // 升序  
})

第三步

然后,我们对排序后的 slice 进行 for-range 遍历:

for _, pair := range tmpList {  
   fmt.Printf("key: %v value: %v \n", pair.Key, pair.Val)  
}

// 结果:
key: 城市D value: 10.39 
key: 城市E value: 333.33 
key: 城市C value: 500.26 
key: 城市B value: 1992.33 
key: 城市A value: 2334.2

可以看到,排序后的 slice 第一项就是我们想要的结果。

如果我们想要获取其中 value 值最大的一项,只需要更改 sort.Slice 中的比较方法接口:

sort.Slice(tmpList, func(i, j int) bool {  
   return tmpList[i].Val > tmpList[j].Val // 降序  
 //return tmpList[i].Val < tmpList[j].Val // 升序})

总结

上面测试的完整代码如下:

package main  
  
import (  
   "fmt"  
 "sort")  
  
var result = map[string]float64{  
   "城市A": 2334.20,  
 "城市B": 1992.33,  
 "城市C": 500.26,  
 "城市D": 10.39,  
 "城市E": 333.33,  
}

func main() {

   type KVPair struct {  
      Key string  
      Val float64  
   }  
  
   tmpList := make([]KVPair, 0)  
  
   for k, v := range result {  
      tmpList = append(tmpList, KVPair{Key: k, Val: v})  
   }  
  
   sort.Slice(tmpList, func(i, j int) bool {  
      //return tmpList[i].Val > tmpList[j].Val // 降序  
      return tmpList[i].Val < tmpList[j].Val // 升序 
   })  
  
   for _, pair := range tmpList {  
      fmt.Printf("key: %v value: %v \n", pair.Key, pair.Val)  
   }  
}

总结