Tree

tree 包提供泛型树构建器，将扁平切片转换为层级树结构。

快速开始

import "github.com/coldsmirk/vef-framework-go/tree"

type Department struct {
    ID       string        `json:"id"`
    ParentID *string       `json:"parentId"`
    Name     string        `json:"name"`
    Children []Department  `json:"children"`
}

// 定义适配器
adapter := tree.Adapter[Department]{
    GetID:       func(d Department) string { return d.ID },
    GetParentID: func(d Department) *string { return d.ParentID },
    GetChildren: func(d Department) []Department { return d.Children },
    SetChildren: func(d *Department, children []Department) { d.Children = children },
}

// 从扁平切片构建树
roots := tree.Build(flatDepartments, adapter)

适配器

Adapter[T] 结构体定义了树构建器如何访问节点属性：

type Adapter[T any] struct {
    GetID       func(T) string      // 提取节点 ID
    GetParentID func(T) *string     // 提取父级 ID（nil = 根节点）
    GetChildren func(T) []T         // 获取子节点切片
    SetChildren func(*T, []T)       // 设置子节点切片
}

关键规则：

• ParentID 为 nil 的节点被视为根节点
• ParentID 引用不存在的节点时也被视为根节点
• 通过已访问标记防止循环引用

查找节点

FindNode

在树中按 ID 搜索节点：

node, found := tree.FindNode(roots, "dept-123", adapter)
if found {
    fmt.Println(node.Name)
}

FindNodePath

获取从根节点到目标节点的完整路径：

path, found := tree.FindNodePath(roots, "dept-456", adapter)
if found {
    for _, node := range path {
        fmt.Println(node.Name) // 输出：根节点 → 父节点 → dept-456
    }
}

框架集成

tree 包被 CRUD 的 FindTree 构建器使用。创建树端点时：

crud.NewFindTree[Department, DepartmentSearch](tree.Build)

框架会将 tree.Build 作为树构建函数传入，使用模型的树适配器配置。