Go高級特性探究之穩(wěn)定排序詳解

更新時間：2023年06月02日 14:01:10 作者：tracy小貓

Go 語言提供了 sort 包，其中最常用的一種是 sort.Slice() 函數,本篇文章將為大家介紹如何使用 sort.SliceStable() 對結構體數組的某個字段進行穩(wěn)定排序,感興趣的可以了解一下

給結構體字段打上“排序標簽”

如果我們有一個名為 Student 的結構體，其中包含了一個 Name 字符串字段，我們希望對 Student 數組按照 Name 字段進行排序，該怎么辦呢？我們可以為 Name 字段打上一個“排序標簽”，表示排序時使用的順序：

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

使用 sort.SliceStable() 進行穩(wěn)定排序

接下來，我們將展示一個可適用于任何類型的排序函數。該函數將通過反射和標簽獲取結構體的排序字段，并使用 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 字段進行穩(wěn)定排序的結果：