Go语言中结构体的高效分组：利用Map实现多键聚合

type CatKey struct {
    Name string
    Age  int
}

type Cat struct {
    CatKey // 嵌入CatKey，方便访问分组键
    Kittens int
}

func GroupCatsByNameAndAge(cats []*Cat) map[CatKey][]*Cat {
    groupedCats := make(map[CatKey][]*Cat)
    for _, cat := range cats {
        if _, ok := groupedCats[cat.CatKey]; ok {
            // 如果键已存在，则追加到现有切片
            groupedCats[cat.CatKey] = append(groupedCats[cat.CatKey], cat)
        } else {
            // 如果键不存在，则创建一个新切片并添加猫咪
            groupedCats[cat.CatKey] = []*Cat{cat}
        }
    }
    return groupedCats
}

func GroupCatsByNameAndAge(cats []*Cat) map[CatKey][]*Cat {
    groupedCats := make(map[CatKey][]*Cat)
    for _, cat := range cats {
        // 直接追加：如果groupedCats[cat.CatKey]是nil，append会创建一个新切片
        // 否则，append会追加到现有切片
        groupedCats[cat.CatKey] = append(groupedCats[cat.CatKey], cat)
    }
    return groupedCats
}

package main

import (
    "errors"
    "fmt"
    "math/rand"
)

// CatKey 定义了用于分组的键结构体
type CatKey struct {
    Name string
    Age  int
}

// Cat 结构体，包含分组键和额外数据
type Cat struct {
    CatKey // 嵌入CatKey，作为分组键
    Kittens int
}

// NewCat 辅助函数，用于创建Cat实例
func NewCat(name string, age int) *Cat {
    return &Cat{CatKey: CatKey{Name: name, Age: age}, Kittens: rand.Intn(10)}
}

// GroupCatsByNameAndAge 以惯用方式根据CatKey对猫咪进行分组
func GroupCatsByNameAndAge(cats []*Cat) map[CatKey][]*Cat {
    groupedCats := make(map[CatKey][]*Cat)
    for _, cat := range cats {
        // 直接使用append，Go语言会处理nil切片的情况
        groupedCats[cat.CatKey] = append(groupedCats[cat.CatKey], cat)
    }
    return groupedCats
}

func main() {
    // 创建一组猫咪数据
    cats := []*Cat{
        NewCat("Leeroy", 12),
        NewCat("Doofus", 14),
        NewCat("Leeroy", 12),
        NewCat("Doofus", 14),
        NewCat("Leeroy", 12),
        NewCat("Doofus", 14),
        NewCat("Leeroy", 12),
        NewCat("Doofus", 14),
        NewCat("Leeroy", 12),
        NewCat("Doofus", 14),
    }

    // 调用分组函数
    groupedCats := GroupCatsByNameAndAge(cats)

    // 验证分组结果
    Assert(len(groupedCats) == 2, "Expected 2 groups") // 应该有"Leeroy, 12"和"Doofus, 14"两个组
    for key, value := range groupedCats {
        fmt.Printf("Group Key: %+v, Count: %d\n", key, len(value))
        Assert(len(value) == 5, "Expected 5 cats in each group") // 每个组应该有5只猫
    }

    fmt.Println("Success: Cats grouped idiomatically.")
}

// Assert 简单的断言函数，用于测试
func Assert(b bool, msg string) {
    if !b {
        panic(errors.New(msg))
    }
}