Go by Example: Sorting by Functions

英文源地址
有时, 我们想要按自然顺序以外的方式对集合进行排序. 例如, 假设我们希望按字符串长度而不是字母顺序排序. 下面是一个Go中自定义排序的例子.

package main

import (
	"fmt"
	"sort"
)

// 为了在Go中按自定义函数排序, 我们需要一个相应的类型
// 这里我们创建了一个byLength类型, 它只是内置的[]string类型的别名
type byLength []string

// 我们实现sort.Interface接口中需要的方法Len, Less和Swap,在我们自己的类型上
// 以便我们可以使用sort包的泛型Sort函数.
// Len和Swap在不同的类型之间通常是相似的, Less将保留实际的自定义排序逻辑.
// 在我们的例子中, 我们希望按照字符串长度递增的顺序排序, 所以我们在这里使用len(s[i])和len(s[j])

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

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

func (s byLength) Less(i, j int) bool {
	return len(s[i]) < len(s[j])
}

func main() {
	// 所有这些就绪后, 我们现在可以通过将原始的fruits切片转换为byLength来实现自定义排序
	// 然后使用sort.Sort函数对输入的切片进行排序
	fruits := []string{"peach", "banana", "kiwi"}
	sort.Sort(byLength(fruits))
	fmt.Println(fruits)
}

运行我们的程序会显示一个按字符串长度排序的列表

$ go run sorting-by-functions.go 
[kiwi peach banana]

通过遵循创建自定义类型的相同模式, 在该类型上实现三个Interface方法, 然后调用sort.Sort对自定义类型的集合进行排序, 我们可以通过任意函数对Go切片进行排序.
下一节将介绍: Panic.