Golang树形数据处理技巧

Go用嵌入而非继承实现组合模式，因无传统继承机制，需靠接口抽象+值聚合；节点统一实现TreeNode接口，Composite用[]TreeNode聚合子节点，Leaf返回空切片，避免nil导致遍历错误。

为什么用嵌入而非继承来实现组合模式

Go 没有传统面向对象的继承机制，struct 嵌入（embedding）是模拟“is-a”关系的唯一自然方式。树形结构中，父节点和子节点本质是同一类数据（比如都是 Node），但需要表达层级关系——这时用嵌入 Node 字段反而会破坏一致性；正确做法是让子节点作为独立值存在，由父节点通过切片持有，再用接口统一行为。

关键点在于：组合模式在 Go 中不靠“父类字段复用”，而靠“接口抽象 + 值聚合”。常见错误是强行给 Node 加一个 Parent *Node 字段试图模拟双向链表，这会让遍历逻辑耦合、内存难管理、且违背组合模式“透明性”原则。

• 所有节点类型实现同一接口（如 TreeNode），调用方无需关心具体是叶子还是分支
• 分支节点内用 []TreeNode 聚合子节点，而不是 []*Node 或嵌入其他 struct
• 避免在节点 struct 中保存 Parent、Depth 等派生状态，这些应在遍历时按需计算

定义统一接口与两种实现：Leaf 和 Composite

核心是分离「能被遍历」和「能持有子节点」两个职责。Leaf 不含子节点，Composite 必须提供 Children() 方法返回子节点列表。接口不能暴露内部结构（如直接暴露 children []TreeNode 字段），否则调用方会绕过抽象直接操作。

type TreeNode interface {
    Name() string
    Size() int64
    Children() []TreeNode // 叶子节点返回空切片，不是 nil
}

type Leaf struct {
    name string
    size int64
}

func (l Leaf) Name() string  { return l.name }
func (l Leaf) Size() int64   { return l.size }
func (l Leaf) Children() []TreeNode { return nil }

type Composite struct {
    name     string
    children []TreeNode
}

func (c Composite) Name() string { return c.name }
func (c Composite) Size() int64 {
    total := int64(0)
    for _, child := range c.children {
        total += child.Size()
    }
    return total
}
func (c Composite) Children() []TreeNode { return c.children }

递归遍历必须显式处理 nil 切片与空切片

Children() 返回 nil 还是 []TreeNode{} 看似无关紧要，但在递归中会导致 panic 或逻辑跳过。Go 的 for range nil 是安全的（不迭代），但若后续代码假设切片非 nil 并取 len(children) 或索引访问，就会出错。统一约定：所有 Children() 方法返回空切片而非 nil，是最小意外原则。

• Leaf 的 Children() 应返回 []TreeNode{}，不是 nil
• Composite 初始化时用 children: make([]TreeNode, 0)，避免零值为 nil
• 遍历函数开头不加 if children == nil 判断，直接 for range children

否则你可能在某次重构后突然发现某个叶子节点没被计入总大小——因为它的 Children() 返回了 nil，而你的统计函数里写了 if len(node.Children()) > 0 { ... }。

小心深拷贝与指针接收器带来的副作用

如果节点 struct 包含 map、slice 或指针字段，并使用指针接收器实现方法（如 func (n *Composite) Add(child TreeNode)），那么对 Children() 返回的切片做修改（如 append）会意外影响原始数据。组合模式要求节点行为“只读”或“明确可变”，否则遍历时可能污染状态。

• 对外暴露的 Children() 方法应返回副本切片：copy(dst, c.children); return dst，或文档注明“返回引用，请勿修改”
• 添加子节点应通过显式方法（如 AddChild）并返回新实例，而非就地修改 —— 尤其在并发或函数式风格下
• 若用指针接收器，确保所有方法不隐式修改 children 内容（例如 c.children = append(c.children, child) 是危险的）

最稳妥的做法是把 Composite 设计为不可变：每次添加子节点都返回新 Composite，底层 slice 重新分配。虽然略增开销，但彻底规避共享状态引发的竞态和逻辑混乱。