Retro68/gcc/libgo/go/sort/example_interface_test.go

59 lines
1.5 KiB
Go
Raw Normal View History

2012-03-27 23:13:14 +00:00
// Copyright 2011 The Go Authors. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.
package sort_test
import (
"fmt"
"sort"
)
2014-09-21 17:33:12 +00:00
type Person struct {
Name string
Age int
2012-03-27 23:13:14 +00:00
}
2014-09-21 17:33:12 +00:00
func (p Person) String() string {
return fmt.Sprintf("%s: %d", p.Name, p.Age)
}
2012-03-27 23:13:14 +00:00
2014-09-21 17:33:12 +00:00
// ByAge implements sort.Interface for []Person based on
// the Age field.
type ByAge []Person
2012-03-27 23:13:14 +00:00
2014-09-21 17:33:12 +00:00
func (a ByAge) Len() int { return len(a) }
func (a ByAge) Swap(i, j int) { a[i], a[j] = a[j], a[i] }
func (a ByAge) Less(i, j int) bool { return a[i].Age < a[j].Age }
2012-03-27 23:13:14 +00:00
2014-09-21 17:33:12 +00:00
func Example() {
people := []Person{
{"Bob", 31},
{"John", 42},
{"Michael", 17},
{"Jenny", 26},
2012-03-27 23:13:14 +00:00
}
2014-09-21 17:33:12 +00:00
fmt.Println(people)
2018-12-28 15:30:48 +00:00
// There are two ways to sort a slice. First, one can define
// a set of methods for the slice type, as with ByAge, and
// call sort.Sort. In this first example we use that technique.
2014-09-21 17:33:12 +00:00
sort.Sort(ByAge(people))
fmt.Println(people)
2012-03-27 23:13:14 +00:00
2018-12-28 15:30:48 +00:00
// The other way is to use sort.Slice with a custom Less
// function, which can be provided as a closure. In this
// case no methods are needed. (And if they exist, they
// are ignored.) Here we re-sort in reverse order: compare
// the closure with ByAge.Less.
sort.Slice(people, func(i, j int) bool {
return people[i].Age > people[j].Age
})
fmt.Println(people)
2012-03-27 23:13:14 +00:00
// Output:
2014-09-21 17:33:12 +00:00
// [Bob: 31 John: 42 Michael: 17 Jenny: 26]
// [Michael: 17 Jenny: 26 Bob: 31 John: 42]
2018-12-28 15:30:48 +00:00
// [John: 42 Bob: 31 Jenny: 26 Michael: 17]
2012-03-27 23:13:14 +00:00
}