Go语言递归建树时，子节点为空的原因及解决方案

Go语言递归构建树形结构时，子节点常常为空，本文针对Go语言递归查询地址库，children字段始终为nil的问题，深入分析其原因。问题源于`for...range`循环复制变量，导致`region.Children`修改的是副本而非原数组元素。文章提供了两种解决方案：一是将`data.RegionTree`结构体的`Children`字段修改为指针类型`*[]data.RegionTree`；二是使用索引`regionTree[i].Children`直接修改原数组元素。两种方法各有优劣，选择需根据实际情况权衡。

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

在使用Go语言进行数据库查询并构建树形结构时，经常会遇到子节点为空的情况，即使数据库中存在相关数据。本文分析Go语言递归查询地址库，最终树形结构的children字段为nil的问题，并提供解决方案。

问题描述：

代码旨在构建一个地址库的树形结构。数据库查询部分能够正确获取数据，但递归函数findChildren返回的结果赋值给region.children后，children始终为空。

核心代码（示例）：

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)
        fmt.Println("children count =", len(children)) // children count 始终为0
        region.Children = 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) // 这里的问题在于region是副本
        }
    }
    return regionTree
}

问题原因：

Go语言的for...range...循环会复制循环变量。在for _, region := range regionTree中，每次迭代都会创建一个region变量的副本。region.Children = children操作修改的是副本，而非regionTree中的原始元素。因此，getSysRegionTree函数返回的regionTree的children字段仍然为空。

解决方案：

有两种主要方法解决这个问题：

1. 使用指针: 将data.RegionTree结构体的Children字段声明为指针类型*[]data.RegionTree。这需要修改data.RegionTree的定义，并相应地修改代码，确保操作的是指针指向的内存地址。

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

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

选择哪种方法取决于具体需求和代码结构。使用指针方式效率更高，但需要修改数据结构；使用索引访问方式不需要修改数据结构，但效率可能略低。 建议根据实际情况选择最合适的方法。 记住修改data.RegionTree结构体定义，使其Children字段为*[]data.RegionTree，如果选择使用指针方法。

今天关于《Go语言递归建树时，子节点为空的原因及解决方案》的内容就介绍到这里了，是不是学起来一目了然！想要了解更多关于的内容请关注golang学习网公众号！