有时候我们需要根据切片中的某个字段进行切片排序,但sort 包中只有默认基本类型 int 、 float64 和 string 的排序,所以我们可以手动实现sort 包的 sort.Interface 接口,来完成自定义排序。这个接口有三个方法 Len() 、 Less(i,j) 和 Swap(i,j) 。
切片排序
package main
import (
"fmt"
"sort"
)
type Person struct {
Name string // 姓名
Age int // 年龄
CreatedAt int64 // 记录时间
}
// 按照 Person.Age 从大到小排序
type PersonSlice []Person
// 重写 Len() 方法
func (p PersonSlice) Len() int {
return len(p)
}
// 重写 Swap() 方法
func (p PersonSlice) Swap(i, j int) {
p[i], p[j] = p[j], p[i]
}
// 重写 Less() 方法
// 使用不同字段进行对比,并依此排序,这里用了Age,可以更换成CreatedAt进行排序的
// 使用 < 从大到小排序,使用 > 从小到大排序
func (p PersonSlice) Less(i, j int) bool {
return p[j].Age < p[i].Age
}
func main() {
people1 := Person{
Name: "牛奔1",
Age: 18,
CreatedAt: 1618667000,
}
people2 := Person{
Name: "牛奔2",
Age: 19,
CreatedAt: 1618667100,
}
people3 := Person{
Name: "牛奔3",
Age: 20,
CreatedAt: 1618667200,
}
peopleSlice := PersonSlice{}
peopleSlice = append(peopleSlice, people2, people1, people3)
fmt.Println(peopleSlice)
sort.Sort(PersonSlice(peopleSlice)) // 按照Age的逆序排序
fmt.Println(peopleSlice)
sort.Sort(sort.Reverse(PersonSlice(peopleSlice))) // 按照Age的升序排序
fmt.Println(peopleSlice)
}
输出:
[{牛奔2 19 1618667100} {牛奔1 18 1618667000} {牛奔3 20 1618667200}]
[{牛奔3 20 1618667200} {牛奔2 19 1618667100} {牛奔1 18 1618667000}]
[{牛奔1 18 1618667000} {牛奔2 19 1618667100} {牛奔3 20 1618667200}]
看到结果已经根据年龄字段排序啦
|
请发表评论