Skip to content

Commit

Permalink
Update sorting-by-functions example to use slices.SortFunc
Browse files Browse the repository at this point in the history
  • Loading branch information
eliben committed Aug 21, 2023
1 parent 655a3b6 commit 2166e61
Show file tree
Hide file tree
Showing 4 changed files with 106 additions and 89 deletions.
66 changes: 37 additions & 29 deletions examples/sorting-by-functions/sorting-by-functions.go
Original file line number Diff line number Diff line change
Expand Up @@ -7,39 +7,47 @@
package main

import (
"cmp"
"fmt"
"sort"
"slices"
)

// In order to sort by a custom function in Go, we need a
// corresponding type. Here we've created a `byLength`
// type that is just an alias for the builtin `[]string`
// type.
type byLength []string

// We implement `sort.Interface` - `Len`, `Less`, and
// `Swap` - on our type so we can use the `sort` package's
// generic `Sort` function. `Len` and `Swap`
// will usually be similar across types and `Less` will
// hold the actual custom sorting logic. In our case we
// want to sort in order of increasing string length, so
// we use `len(s[i])` and `len(s[j])` here.
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])
}

// With all of this in place, we can now implement our
// custom sort by converting the original `fruits` slice
// to `byLength`, and then use `sort.Sort` on that typed
// slice.
func main() {
fruits := []string{"peach", "banana", "kiwi"}
sort.Sort(byLength(fruits))

// We implement a comparison function for string
// lengths. `cmp.Compare` is helpful for this.
lenCmp := func(a, b string) int {
return cmp.Compare(len(a), len(b))
}

// Now we can call `slices.SortFunc` with this custom
// comparison function to sort `fruits` by name length.
slices.SortFunc(fruits, lenCmp)
fmt.Println(fruits)

// We can use the same technique to sort a slice of
// values that aren't built-in types.
type Person struct {
name string
age int
}

people := []Person{
Person{name: "Jax", age: 37},
Person{name: "TJ", age: 25},
Person{name: "Alex", age: 72},
}

// Sort `people` by age using `slices.SortFunc`.
//
// Note: if the `Person` struct is large,
// you may want the slice to contain `*Person` instead
// and adjust the sorting function accordingly. If in
// doubt, [benchmark](testing-and-benchmarking)!
slices.SortFunc(people,
func(a, b Person) int {
return cmp.Compare(a.age, b.age)
})
fmt.Println(people)
}
4 changes: 2 additions & 2 deletions examples/sorting-by-functions/sorting-by-functions.hash
Original file line number Diff line number Diff line change
@@ -1,2 +1,2 @@
e5a6006366e05ee7785eebac8ba588e4b937a197
h4g4vaLBtkw
9902d1c5654b64d8b381ea7888c0793ac8ab4a97
3EaTknAZHMu
9 changes: 1 addition & 8 deletions examples/sorting-by-functions/sorting-by-functions.sh
Original file line number Diff line number Diff line change
@@ -1,10 +1,3 @@
# Running our program shows a list sorted by string
# length, as desired.
$ go run sorting-by-functions.go
[kiwi peach banana]

# By following this same pattern of creating a custom
# type, implementing the three `Interface` methods on that
# type, and then calling sort.Sort on a collection of that
# custom type, we can sort Go slices by arbitrary
# functions.
[{TJ 25} {Jax 37} {Alex 72}]
116 changes: 66 additions & 50 deletions public/sorting-by-functions

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

0 comments on commit 2166e61

Please sign in to comment.