Gremlin遍历图数据库：如何根据isTemporary属性优先选择临时边？

嗨，作为Gremlin新手能想到用choose步骤已经很棒了！你说得对，文档里的choose示例大多是返回值，但它完全可以用来控制遍历分支，帮你实现优先走临时边的需求。下面我一步步给你讲清楚怎么实现：

1. 先构建你的示例图（方便测试） #

首先我们把你给出的示例图用Gremlin语句创建出来，这样你可以直接在控制台测试：

g.addV('vertex').property(id, 'A').as('a')
  .addV('vertex').property(id, 'B').as('b')
  .addV('vertex').property(id, 'C').as('c')
  .addV('vertex').property(id, 'D').as('d')
  .addE('connect').from('a').to('b').property('isTemporary', true).property('value', 1)
  .addE('connect').from('a').to('c').property('isTemporary', false).property('value', 2)
  .addE('connect').from('b').to('d').property('isTemporary', false).property('value', 3)

2. 核心遍历查询 #

要实现优先走isTemporary=true的边，没有则走isTemporary=false的边的遍历逻辑，我们可以结合repeat、choose和path()来完成：

g.V('A')
  .repeat(
    // 判断当前顶点是否有临时边，有则走临时边，否则走非临时边
    choose(
      outE().has('isTemporary', true).count().is(gt(0)),
      outE().has('isTemporary', true).inV(),
      outE().has('isTemporary', false).inV()
    )
  )
  .until(outE().count().is(0)) // 直到当前顶点没有出边时停止遍历
  .path() // 返回完整的遍历路径

为什么这样写？ #

• repeat(...)：用来重复执行遍历逻辑，直到满足until的终止条件
• choose(condition, trueBranch, falseBranch)：
  • 第一个参数outE().has('isTemporary', true).count().is(gt(0))：判断当前顶点是否存在临时边
  • 第二个参数：如果存在临时边，就遍历这些边并跳到目标顶点
  • 第三个参数：如果没有临时边，就遍历非临时边跳到目标顶点
• until(outE().count().is(0))：当遍历到没有出边的顶点（比如示例中的D）时停止
• path()：把整个遍历的顶点路径返回出来

执行这个查询后，你会得到预期的路径：[v[A], v[B], v[D]]

3. 额外优化：处理多临时边的情况 #

如果某个顶点有多条临时边，上面的查询会遍历所有临时边对应的顶点。如果你需要只选其中一条（比如按value排序选最大/最小），可以在临时边的分支里加过滤或排序，比如：

g.V('A')
  .repeat(
    choose(
      outE().has('isTemporary', true).count().is(gt(0)),
      // 选value最大的临时边
      outE().has('isTemporary', true).order().by('value', desc).limit(1).inV(),
      outE().has('isTemporary', false).inV()
    )
  )
  .until(outE().count().is(0))
  .path()

4. 替代方案：用coalesce简化逻辑 #

其实还有更简洁的写法——用coalesce步骤，它会返回第一个非空的遍历分支，刚好符合我们“优先选临时边”的需求：

g.V('A')
  .repeat(
    coalesce(
      outE().has('isTemporary', true).inV(),
      outE().has('isTemporary', false).inV()
    )
  )
  .until(outE().count().is(0))
  .path()

这个查询和之前的choose版本效果完全一样，但代码更短，推荐你试试！

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