Go语言递归构建树形结构时，子节点为空的常见问题及解决方法
在使用Go语言进行数据库查询并构建树形结构时，经常会遇到子节点为空的问题。本文分析此问题，并提供解决方法。
问题描述:
代码尝试构建一个树形结构，但数据库中明明存在子节点数据，构建的树形结构却显示子节点为空（children 为 nil）。
立即学习“go语言免费学习笔记（深入）”；
示例代码:
以下代码片段演示了这个问题：getSysRegionTree 函数递归调用 findChildren 函数，构建树形结构。尽管数据库中有子节点，findChildren 返回的 children 切片在 getSysRegionTree 中却为空。
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(®ionTree).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(®ionTree).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 中的原始元素。
解决方案:
有两种方法可以解决这个问题：
-
使用指针: 将
data.RegionTree结构体的Children字段定义为指针类型*[]data.RegionTree。这样，修改region.Children就直接修改了regionTree中的原始元素。 这需要修改data.RegionTree结构体定义以及相关代码。 -
使用索引循环: 使用索引循环遍历
regionTree，直接修改regionTree中元素的值。for i, region := range regionTree { children := findChildren(tx, region.ID, deep-1) regionTree[i].Children = children }
通过以上两种方法，可以正确构建树形结构，避免子节点为空的问题。 选择哪种方法取决于代码的整体设计和可维护性。 使用指针通常更符合 Go 的习惯，但需要仔细考虑指针的使用，避免内存管理问题。 使用索引循环则更直接，但代码可能略显冗长。
