如何编写MongoDB查询语句检测集合中的节点循环依赖

核心思路 #

• 首先将每个节点correlatedNodes字段下所有嵌套的nodes数组合并为一个打平的关联节点列表
• 对集合做自连接查询，匹配满足「A关联B且B关联A」的节点对
• 去重后输出所有存在循环依赖的节点名称

具体聚合查询 #

假设你的集合名为nodeRelations，直接执行以下聚合语句即可：

db.nodeRelations.aggregate([
  // 步骤1：打平每个节点的所有关联节点，去重
  {
    $addFields: {
      allRelatedNodes: {
        $reduce: {
          input: "$correlatedNodes",
          initialValue: [],
          in: { $setUnion: ["$$value", "$$this.nodes"] }
        }
      }
    }
  },
  // 步骤2：自连接查询匹配循环依赖对
  {
    $lookup: {
      from: "nodeRelations",
      let: { currentNode: "$nodeName", currentRelated: "$allRelatedNodes" },
      pipeline: [
        {
          $addFields: {
            targetRelatedNodes: {
              $reduce: {
                input: "$correlatedNodes",
                initialValue: [],
                in: { $setUnion: ["$$value", "$$this.nodes"] }
              }
            }
          }
        },
        {
          $match: {
            $expr: {
              $and: [
                // B在A的关联列表中
                { $in: ["$nodeName", "$$currentRelated"] },
                // A在B的关联列表中
                { $in: ["$$currentNode", "$targetRelatedNodes"] },
                // 排除自身循环，同时避免(A,B)和(B,A)重复匹配
                { $lt: ["$$currentNode", "$nodeName"] }
              ]
            }
          }
        },
        { $project: { nodeName: 1, _id: 0 } }
      ],
      as: "cyclePairs"
    }
  },
  // 步骤3：过滤出存在循环依赖的记录
  { $match: { cyclePairs: { $ne: [] } } },
  // 步骤4：提取所有循环节点，去重后输出
  {
    $project: {
      cycleNodes: { $setUnion: [["$nodeName"], "$cyclePairs.nodeName"] },
      _id: 0
    }
  },
  { $unwind: "$cycleNodes" },
  { $group: { _id: null, allCycleNodes: { $addToSet: "$cycleNodes" } } },
  { $project: { _id: 0, allCycleNodes: 1 } }
])

如果你需要检测节点自身关联自己的自循环场景，将上述查询中$lt: ["$$currentNode", "$nodeName"]替换为$ne: ["$$currentNode", "$nodeName"]即可。

结果说明 #

• 你提供的示例数据中没有符合条件的循环依赖对，查询会返回空数组
• 如果存在循环依赖，最终会在allCycleNodes字段返回所有涉事节点的nodeName列表

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