Gremlin实现广度优先搜索并按层级输出顶点列表

嘿，我懂你想要的效果——把这棵树的BFS结果按层级拆成单独的列表，而不是挤在一个大列表里对吧？你的当前查询已经能保证顶点顺序正确，咱们只需要加一些分组和层级记录的逻辑就行。

这里给你两个可行的Gremlin查询方案：

方案一：直接记录层级分组（包含根节点层级） #

这个方案会把根节点单独归为层级0，后续每一层子节点按遍历次数分组：

g.V().hasLabel("root")
  // 先把根节点分到层级0
  .group().by(constant(0))
  // 合并根节点分组和子节点的层级分组
  .union(
    identity(),
    repeat(out("tree_edge").barrier().dedup())
      .emit()
      // 用loops()记录遍历次数，对应层级1、2...
      .group().by(__.loops())
  )
  // 展开分组键值对
  .unfold()
  // 按层级序号排序
  .order().by(keys)
  // 提取每个层级的顶点列表
  .select(values)

方案二：通过路径长度计算层级 #

这种方法利用遍历路径的长度来推导层级，适合需要更灵活层级定义的场景：

g.V().hasLabel("root")
  .repeat(out("tree_edge").barrier().dedup())
    .emit()
    // 记录从根到当前顶点的完整路径
    .path()
  // 按路径长度减1分组（路径长度=层级+1）
  .group().by(path().count().minus(1))
  .unfold()
  .order().by(keys)
  .select(values)
  // 可选：如果需要输出顶点ID而非顶点对象，加上这行
  .map(unfold().id())

小说明 #

• 方案一的层级0是根节点，层级1是根的直接子节点，以此类推；
• 方案二里，路径长度为2对应层级1（根→子节点），如果需要包含根节点的层级0，可以在开头加上对根节点的path处理，和后面的结果union起来。

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