sorting - golang排序 slice 升序或降序

Question

我需要对来自 3rdparty 包的类型 slice 进行排序。根据某些条件，顺序必须是升序或降序。

我想出的解决办法是:

type fooAscending []foo

func (v fooAscending) Len() int           { return len(v) }
func (v fooAscending) Swap(i, j int)      { v[i], v[j] = v[j], v[i] }
func (v fooAscending) Less(i, j int) bool { return v[i].Amount < v[j].Amount }

type fooDescending []foo

func (v fooDescending) Len() int           { return len(v) }
func (v fooDescending) Swap(i, j int)      { v[i], v[j] = v[j], v[i] }
func (v fooDescending) Less(i, j int) bool { return v[i].Amount > v[j].Amount }

if someCondition {
    sort.Sort(fooAscending(array))
} else {
    sort.Sort(fooDescending(array))
}

有没有更好的方法来做到这一点。这个任务13行代码，大部分都是重复的，似乎有点太多了。

Answer 1

从 Go 1.8 开始，有一种更简单的方法可以对 slice 进行排序，而无需您定义新类型。您只需将匿名函数传递给 sort.Slice 函数。

a := []int{5, 3, 4, 7, 8, 9}
sort.Slice(a, func(i, j int) bool {
    return a[i] < a[j]
})
for _, v := range a {
    fmt.Println(v)
}

这里会按升序排序，如果你想相反，只需在匿名函数中写a[i] > a[j]即可。