Go语言递归构建树形结构时,子节点为空的常见问题及解决方案
在使用Go语言从数据库查询数据并构建树形结构时,经常会遇到子节点为空的情况,即使数据库中存在子节点数据。本文分析此问题,并提供解决方案。
问题描述:
假设要构建一个地区树形结构。数据库查询能正确获取数据,但递归函数返回的子节点始终无法正确赋值到父节点的children字段,导致最终树形结构所有节点的children字段都为空。
立即学习“go语言免费学习笔记(深入)”;
示例代码:
func getSysRegionTree(pid string, deep int) (error, []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 // 问题在此行,修改的是副本 } 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) // 同样修改的是副本 } } return regionTree}
登录后复制
问题根源:
Go语言的for…range…循环会创建循环变量的副本。在getSysRegionTree函数中,for _, region := range regionTree 创建了region变量的副本,region.Children = children 操作修改的是副本,而非regionTree切片中的原始元素。
解决方案:
主要有两种解决方案:
使用指针: 将data.RegionTree 结构体的Children字段定义为*[]data.RegionTree,即指向切片的指针。修改指针指向的切片内容会直接影响regionTree中的元素。 这需要修改数据结构定义和相关代码。
使用索引遍历: 放弃for…range…,使用基于索引的循环:
for i := range regionTree { children := findChildren(tx, regionTree[i].ID, deep-1) regionTree[i].Children = children}
登录后复制
这种方法直接修改regionTree切片中的元素,避免了值拷贝问题。
选择哪种方法取决于代码结构和个人偏好。使用指针更符合Go语言的习惯,但需要修改数据结构;使用索引遍历则无需修改数据结构,但代码略显冗长。 建议优先考虑使用指针,以提高代码的可读性和效率。
以上就是Go语言递归构建树状结构时,子节点为空的原因是什么及如何解决?的详细内容,更多请关注【创想鸟】其它相关文章!
版权声明:本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌抄袭侵权/违法违规的内容, 请发送邮件至253000106@qq.com举报,一经查实,本站将立刻删除。
发布者:PHP中文网,转转请注明出处:https://www.chuangxiangniao.com/p/2537766.html