Go高级特性探究之稳定排序详解

在 IT 开发中，有时我们需要对结构体数组进行排序。Go 语言提供了 sort 包，其中最常用的一种是 sort.Slice() 函数。但是，当我们需要保持相同元素之间的顺序稳定时，该如何实现呢？

本篇文章将为大家介绍如何使用 sort.SliceStable() 对结构体数组的某个字段进行稳定排序。同时，我们将为你展示如何使用反射和结构体标签，让排序更加优雅和通用。

给结构体字段打上“排序标签”

如果我们有一个名为 Student 的结构体，其中包含了一个 Name 字符串字段，我们希望对 Student 数组按照 Name 字段进行排序，该怎么办呢？我们可以为 Name 字段打上一个“排序标签”，表示排序时使用的顺序：

type Student struct {
   Id    int     sort:"id"
   Name  string  sort:"name"
   Score float64 sort:"score"
}

使用 sort.SliceStable() 进行稳定排序

接下来，我们将展示一个可适用于任何类型的排序函数。该函数将通过反射和标签获取结构体的排序字段，并使用 sort.SliceStable() 进行排序。

func SortSliceStable(slice interface{}, sortField string) {
   rv := reflect.ValueOf(slice)
   if rv.Kind() != reflect.Slice {
       panic("SortSliceStable called with non-slice type")
   }
   if rv.Len() == 0 {
       return
   }
   // 获取结构体的元素类型
   elemType := rv.Type().Elem()
   // 获取排序字段
   field, ok := elemType.FieldByName(sortField)
   if !ok {
       panic("SortSliceStable called with unknown or unexported struct field name: " + sortField)
   }
   // 获取 less 函数
   less := func(i, j int) bool {
       v1 := rv.Index(i).FieldByName(field.Name)
       v2 := rv.Index(j).FieldByName(field.Name)
       switch v1.Kind() {
       case reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32, reflect.Int64:
           return v1.Int() < v2.Int()
       case reflect.Float32, reflect.Float64:
           return v1.Float() < v2.Float()
       case reflect.String:
           return v1.String() < v2.String()
   }
   panic("unsupported type")
}
// 使用 sort.SliceStable 进行排序
sort.SliceStable(slice, less)
}

现在，我们可以按照以下方法使用该排序函数进行排序：

func main() {
   students := []Student{
       {1, "zhangsan", 90.0},
       {2, "lisi", 80.0},
       {3, "wangwu", 70.0},
   }
   // 按照 Name 字段进行排序
   SortSliceStable(students, "name")
   fmt.Println(students)
}

运行以上代码，即可得到按照 Name 字段进行稳定排序的结果：

[{2 lisi 80} {3 wangwu 70} {1 zhangsan 90}]

通过添加标签和使用反射，我们让排序过程更加通用和优雅。这个方法能够有效地提高我们的工作效率和代码质量，值得我们推广和应用。稳定排序，原来这么简单！

来源：https://juejin.cn/post/7238570834342805563