JavaScript递归遍历树形结构：如何收集节点后代值并返回

嘿，我来帮你搞定这个递归处理数据库树形节点的问题！你已经能打印值说明递归的遍历逻辑是通的，核心问题就是结果的收集和递归基准条件的明确，再加上回调的正确触发时机，咱们一步步来梳理：

首先明确递归的基准条件 #

递归的终止点很简单：当某个节点没有任何子节点的时候，就停止当前分支的递归，直接返回空数组——因为没有后代可以继续遍历了。

分情况实现：同步内存数据 vs 异步数据库查询 #

情况1：数据已在内存中（同步场景） #

如果你的树形数据已经加载到内存里（比如从数据库一次性查出来了），可以先写一个同步的递归函数来收集后代，再包装成带回调的版本：

// 模拟内存中的树形节点数据
const treeNodes = [
  { id: 1, parentId: null },
  { id: 2, parentId: 1 },
  { id: 3, parentId: 1 },
  { id: 4, parentId: 2 },
  { id: 5, parentId: 4 },
];

// 同步递归函数：返回指定节点的所有后代数组
function getDescendantsSync(targetId, nodes) {
  // 第一步：找到当前节点的直接子节点
  const children = nodes.filter(node => node.parentId === targetId);
  
  // 基准条件：没有子节点，返回空数组终止递归
  if (children.length === 0) return [];
  
  // 合并子节点本身 + 每个子节点的后代（用flatMap自动展开数组）
  return children.flatMap(child => [child, ...getDescendantsSync(child.id, nodes)]);
}

// 包装成带回调的函数
function getAllDescendants(targetId, nodes, callback) {
  try {
    const descendants = getDescendantsSync(targetId, nodes);
    // 成功时把结果传给回调
    callback(null, descendants);
  } catch (error) {
    // 出错时把错误传给回调
    callback(error);
  }
}

// 调用示例
getAllDescendants(1, treeNodes, (err, result) => {
  if (err) console.error('出错了：', err);
  else console.log('所有后代：', result);
});

情况2：直接从数据库异步查询（更贴近你的场景） #

如果每次都要查数据库获取子节点（不能一次性加载全量数据），就得处理异步递归的回调时机——必须等所有子节点的递归都完成，才能把完整的后代数组传给回调：

// 假设你有一个数据库查询函数，参数：SQL语句、查询参数、回调
// db.query(sql, params, (err, rows) => { ... })

function getAllDescendants(targetId, callback) {
  // 第一步：查询当前节点的直接子节点
  db.query('SELECT * FROM nodes WHERE parent_id = ?', [targetId], (err, children) => {
    if (err) return callback(err); // 数据库查询出错，直接传给回调

    // 基准条件：没有子节点，返回空数组
    if (children.length === 0) {
      return callback(null, []);
    }

    let allDescendants = [...children]; // 先把直接子节点加进去
    let completedRecursions = 0; // 计数器：跟踪子节点的递归是否全部完成

    // 遍历每个子节点，递归查询它们的后代
    children.forEach(child => {
      getAllDescendants(child.id, (err, childDescendants) => {
        if (err) return callback(err);

        // 把当前子节点的后代合并到总数组
        allDescendants = [...allDescendants, ...childDescendants];
        completedRecursions++;

        // 当所有子节点的递归都完成时，才触发回调返回完整结果
        if (completedRecursions === children.length) {
          callback(null, allDescendants);
        }
      });
    });
  });
}

// 调用示例
getAllDescendants(1, (err, result) => {
  if (err) console.error('查询失败：', err);
  else console.log('指定节点的所有后代：', result);
});

你之前遇到的问题分析 #

你说“能打印所有值但无法捕获”，大概率是这两个原因：

1. 没有正确合并递归结果：只是在递归里打印，但没把每个分支的结果收集到同一个数组里；
2. 异步回调时机不对：如果是数据库异步查询，没等所有子节点的递归完成就触发了回调，导致返回的数组不完整。

现在用上面的代码，就能把所有后代节点正确收集，并且通过回调返回完整结果啦。

内容的提问来源于stack exchange，提问作者brycejl