Go语言递归构建树形结构时，子节点为空的常见问题及解决方法

在使用Go语言进行数据库查询并构建树形结构时，经常会遇到子节点为空的问题。本文分析此问题，并提供解决方法。

问题描述:

代码尝试构建一个树形结构，但数据库中明明存在子节点数据，构建的树形结构却显示子节点为空（children 为 nil）。

立即学习“go语言免费学习笔记（深入）”；

示例代码:

以下代码片段演示了这个问题：getSysRegionTree 函数递归调用 findChildren 函数，构建树形结构。尽管数据库中有子节点，findChildren 返回的 children 切片在 getSysRegionTree 中却为空。

Go语言（Golang）1.26.0

Go语言（Golang）1.26.0版本官方下载，版本号 1.26.0，适合旧项目维护、兼容性测试和指定版本开发环境搭建。

下载

func getSysRegionTree(pid string, deep int) (err error, result []data.RegionTree) {
    var regionTree []data.RegionTree
    err = tx.Model(&model.SysRegion{}).Where("pid = ? and enabled = true", pid).Order("id asc").Find(&regionTree).Error
    if err != nil {
        panic(err)
    }
    for _, region := range regionTree {
        children := findChildren(tx, region.ID, deep-1) // ID假设为string类型
        fmt.Println("children count =", len(children))
        region.Children = children // 问题在此，修改的是region的副本
    }
    return err, regionTree
}

func findChildren(tx *gorm.DB, pid string, left int) []data.RegionTree {
    var regionTree []data.RegionTree
    err := tx.Model(&model.SysRegion{}).Where("pid = ? and enabled = true", pid).Order("id desc").Find(&regionTree).Error
    if err != nil {
        panic(err)
    }
    if left > 1 {
        for _, region := range regionTree {
            region.Children = findChildren(tx, region.ID, left-1) // 问题在此，修改的是region的副本
        }
    }
    return regionTree
}

问题分析:

根本原因在于 for...range... 循环的特性。Go语言的 for...range... 循环会复制循环变量的值。在 getSysRegionTree 函数中，for _, region := range regionTree 复制了 regionTree 中每个元素的值到局部变量 region。因此，region.Children = children 修改的是局部变量 region 的副本，而不是 regionTree 中的原始元素。

解决方案:

有两种方法可以解决这个问题：

1. 使用指针: 将 data.RegionTree 结构体的 Children 字段定义为指针类型 *[]data.RegionTree。这样，修改 region.Children 就直接修改了 regionTree 中的原始元素。 这需要修改 data.RegionTree 结构体定义以及相关代码。

2. 使用索引循环: 使用索引循环遍历 regionTree，直接修改 regionTree 中元素的值。

   for i, region := range regionTree {
       children := findChildren(tx, region.ID, deep-1)
       regionTree[i].Children = children
   }

通过以上两种方法，可以正确构建树形结构，避免子节点为空的问题。 选择哪种方法取决于代码的整体设计和可维护性。 使用指针通常更符合 Go 的习惯，但需要仔细考虑指针的使用，避免内存管理问题。 使用索引循环则更直接，但代码可能略显冗长。