探究 Go 的高级特性之【稳定排序,原来这么简单!】

514 阅读2分钟

在 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}]

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