Как в Go отсортировать срез (slice) структур?

Ответ

В Go для сортировки пользовательских типов данных, таких как срезы структур, используется стандартный пакет sort.

Существует два основных подхода:

1. Классический метод (с помощью sort.Interface)

Этот метод требует реализации трех методов (Len, Less, Swap) для вашего типа. Это полезно, если вы часто сортируете один и тот же тип данных по одному и тому же критерию.

package main

import (
    "fmt"
    "sort"
)

type Person struct {
    Name string
    Age  int
}

// ByAge реализует sort.Interface для []Person по полю Age.
type ByAge []Person

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 }

func main() {
    people := []Person{
        {"Bob", 31},
        {"John", 42},
        {"Alice", 25},
        {"Michael", 17},
    }
    sort.Sort(ByAge(people))
    fmt.Println(people) // [{Michael 17} {Alice 25} {Bob 31} {John 42}]
}

2. Современный и более простой метод (с sort.Slice)

Начиная с Go 1.8, можно использовать функцию sort.Slice. Она не требует создания нового типа и реализации интерфейса, что делает код короче и удобнее для одноразовых сортировок.

// ... в функции main
people := []Person{
    {"Bob", 31},
    {"John", 42},
    {"Alice", 25},
    {"Michael", 17},
}

// Сортировка по полю Age
sort.Slice(people, func(i, j int) bool {
    return people[i].Age < people[j].Age
})

fmt.Println(people) // [{Michael 17} {Alice 25} {Bob 31} {John 42}]

// Для сортировки в обратном порядке достаточно изменить условие
sort.Slice(people, func(i, j int) bool {
    return people[i].Age > people[j].Age // '>' вместо '<'
})

Ответ 18+ 🔞

А, Го, сортировка структур, да? Ну, тут, блядь, как в жизни — можно по старинке, с церемониями, а можно просто и быстро, по-современному. Слушай, как есть.

Вот представь, у тебя есть срез каких-нибудь Person — людей, то есть. Имя, возраст, всякое такое. И надо их упорядочить, например, по годам. В Go для этого, сука, есть два пути, как в той сказке — налево пойдёшь, направо пойдёшь...

Первый путь — классика, для ценителей и консерваторов.

Тут надо, блядь, реализовать целый интерфейс sort.Interface. Три метода: Len, Swap, Less. Как будто ты в театре, сука, ставишь спектакль. Зато потом сортируешь одной строчкой. Подходит, если ты этим одним типом данных будешь сортироваться до посинения.

package main

import (
    "fmt"
    "sort"
)

type Person struct {
    Name string
    Age  int
}

// ByAge — это наш специальный тип-обёртка, чтобы прикрутить методы.
type ByAge []Person

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 }

func main() {
    people := []Person{
        {"Bob", 31},
        {"John", 42},
        {"Alice", 25},
        {"Michael", 17},
    }
    sort.Sort(ByAge(people)) // Всё, отсортировали. Красота.
    fmt.Println(people) // [{Michael 17} {Alice 25} {Bob 31} {John 42}]
}

Второй путь — для тех, кому влом городить огород.

С Go 1.8 пришла функция sort.Slice. Это, блядь, просто песня! Никаких новых типов, никаких трёх методов. Просто берёшь срез и пишешь функцию сравнения двух элементов. Всё, пиздец.

// ... допустим, в функции main
people := []Person{
    {"Bob", 31},
    {"John", 42},
    {"Alice", 25},
    {"Michael", 17},
}

// Хочешь по возрасту? Пожалуйста!
sort.Slice(people, func(i, j int) bool {
    return people[i].Age < people[j].Age // Младше — значит раньше в списке.
})

fmt.Println(people) // [{Michael 17} {Alice 25} {Bob 31} {John 42}]

// А хочешь наоборот, от старых к молодым? Да не вопрос!
sort.Slice(people, func(i, j int) bool {
    return people[i].Age > people[j].Age // Просто знак меняешь, и всё, ёпта!
})

Вот и вся магия. Первый способ — это как настройка старого радиоприёмника, крутишь ручки, знаешь каждую деталь. Второй — как нажать кнопку на пульте. Выбирай, что душе угодно. Главное — не запутайся, а то отсортируешь так, что сам потом не разберёшь, где кто.