go切片排序

前言

有時候我們需要根據切片中的某個字段進行切片排序，但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}]

看到結果已經根據年齡字段排序啦