使用 Go 语言接口嵌入简化结构体切片排序-Golang-PHP中文网

使用 Go 语言接口嵌入简化结构体切片排序

花韻仙語

发布： 2025-07-28 18:00:22

原创

397人浏览过

使用 go 语言接口嵌入简化结构体切片排序

在 Go 语言中，使用 sort 包对结构体切片进行排序，通常需要实现 sort.Interface 接口，该接口包含 Len、Swap 和 Less 三个方法。对于不同的结构体类型，Len 和 Swap 的实现往往是相同的，只是 Less 方法的比较逻辑不同。为了避免重复编写 Len 和 Swap 方法，可以使用接口嵌入的方式来简化代码。

首先，定义一个通用的切片类型，并实现 Len 和 Swap 方法：

type MySlice []MyStruct

func (s MySlice) Len() int {
    return len(s)
}

func (s MySlice) Swap(i, j int) {
    s[i], s[j] = s[j], s[i]
}

登录后复制

然后，定义具体的结构体类型 MyStruct：

type MyStruct struct {
    Field1 int
    Field2 string
}

登录后复制

接下来，可以创建不同的排序器类型，这些类型嵌入 MySlice，并仅重写 Less 方法，以实现不同的排序逻辑。例如，按 Field1 排序的排序器：

type SortByField1 struct {
    MySlice
}

func (s SortByField1) Less(i, j int) bool {
    return s.MySlice[i].Field1 < s.MySlice[j].Field1
}

登录后复制

再例如，按 Field2 排序的排序器：

SpeakingPass-打造你的专属雅思口语语料

使用chatGPT帮你快速备考雅思口语，提升分数

查看详情

type SortByField2 struct {
    MySlice
}

func (s SortByField2) Less(i, j int) bool {
    return s.MySlice[i].Field2 < s.MySlice[j].Field2
}

登录后复制

最后，可以使用 sort.Sort 函数对切片进行排序：

package main

import (
    "fmt"
    "sort"
)

type MyStruct struct {
    Field1 int
    Field2 string
}

type MySlice []MyStruct

func (s MySlice) Len() int {
    return len(s)
}

func (s MySlice) Swap(i, j int) {
    s[i], s[j] = s[j], s[i]
}

type SortByField1 struct {
    MySlice
}

func (s SortByField1) Less(i, j int) bool {
    return s.MySlice[i].Field1 < s.MySlice[j].Field1
}

type SortByField2 struct {
    MySlice
}

func (s SortByField2) Less(i, j int) bool {
    return s.MySlice[i].Field2 < s.MySlice[j].Field2
}

func main() {
    data := []MyStruct{
        {Field1: 3, Field2: "c"},
        {Field1: 1, Field2: "a"},
        {Field1: 2, Field2: "b"},
    }

    // 按 Field1 排序
    sort.Sort(SortByField1{data})
    fmt.Println("按 Field1 排序:", data)

    // 恢复原始数据
    data = []MyStruct{
        {Field1: 3, Field2: "c"},
        {Field1: 1, Field2: "a"},
        {Field1: 2, Field2: "b"},
    }

    // 按 Field2 排序
    sort.Sort(SortByField2{data})
    fmt.Println("按 Field2 排序:", data)
}

登录后复制

注意事项：