golang 自定义sort

在Golang中，排序是一个非常常见的操作。而Go语言中提供了sort包，可以方便地进行各种排序操作。除了sort包提供的基本排序功能外，我们还可以自定义排序方法，以满足更加特定的需求。本文将介绍如何使用Golang自定义sort。

实现sort.Interface接口

要自定义排序，首先需要实现sort.Interface接口。sort.Interface接口定义了Len()、Swap()和Less(i, j int)三个方法，分别用于获取待排序数据长度、交换两个元素的位置和判断两个元素的大小关系。

type Interface interface {
	Len() int
	Swap(i, j int)
	Less(i, j int) bool
}

在自定义类型上实现这三个方法，即可实现sort.Interface接口。比如，我们有一个自定义类型Person，包含名字和年龄两个字段：

type Person struct {
	Name string
	Age  int
}

为了实现sort.Interface接口，我们需要在Person类型上定义Len()、Swap()和Less(i, j int)三个方法。Len()方法返回待排序数据长度；Swap()方法交换两个元素的位置；Less()方法返回两个元素的大小比较结果（是否升序）。

type ByAge []Person

func (s ByAge) Len() int           { return len(s) }
func (s ByAge) Swap(i, j int)      { s[i], s[j] = s[j], s[i] }
func (s ByAge) Less(i, j int) bool { return s[i].Age < s[j].Age }

使用自定义排序方法

一旦实现了sort.Interface接口，我们就可以使用自定义的排序方法了。首先，我们需要准备待排序的数据。

people := []Person{
	{"Alice", 25},
	{"Bob", 30},
	{"John", 20},
	{"Doe", 35},
}

接下来，我们使用sort包的Sort()方法对数据进行排序：

sort.Sort(ByAge(people))

以上代码将按照Person的Age字段进行升序排序。完成排序后，people的顺序如下：

[{John 20} {Alice 25} {Bob 30} {Doe 35}]

自定义降序排序

在上面的例子中，我们实现了升序排序。如果我们希望按照Age字段进行降序排序，只需要稍作修改即可。

type ByAgeDesc []Person

func (s ByAgeDesc) Len() int           { return len(s) }
func (s ByAgeDesc) Swap(i, j int)      { s[i], s[j] = s[j], s[i] }
func (s ByAgeDesc) Less(i, j int) bool { return s[i].Age > s[j].Age }

sort.Sort(ByAgeDesc(people))

通过定义ByAgeDesc类型，并在Less()方法中采用大于号进行比较，我们实现了按照Age字段进行降序排序。完成排序后，people的顺序如下：

[{Doe 35} {Bob 30} {Alice 25} {John 20}]

自定义多字段排序

除了单个字段的排序外，有时候我们需要按照多个字段进行排序。例如，我们希望先按照Age进行排序，如果Age相同，则按照Name进行排序。

为了实现多字段排序，我们可以先按照第一个字段排序，然后在Less()方法中判断第一个字段相等的情况下，再按照第二个字段进行排序。

type ByAgeName []Person

func (s ByAgeName) Len() int      { return len(s) }
func (s ByAgeName) Swap(i, j int) { s[i], s[j] = s[j], s[i] }
func (s ByAgeName) Less(i, j int) bool {
	if s[i].Age == s[j].Age {
		return s[i].Name < s[j].Name
	}
	return s[i].Age < s[j].Age
}

sort.Sort(ByAgeName(people))

经过以上代码的处理，我们实现了先按照Age进行排序，如果Age相同，则按照Name进行排序。完成排序后，people的顺序如下：

[{John 20} {Alice 25} {Bob 30} {Doe 35}]

通过自定义sort包提供的sort.Interface接口，我们可以很方便地实现各种排序需求。无论是升序、降序还是多字段排序，Golang都提供了相应的接口和方法，方便我们进行定制化的排序操作。