如何计算 Gremlin 中子树的聚合？

Question

我有一棵有很多层的树，其中叶节点可能具有属性“计数”。 我想计算每个子树的总数，并将这些值缓存在每个子树的根节点中。 Gremlin 有可能吗？

Answer 1

您可以使用 sideEffect 来实现 - 这非常简单。我们设置了一个简单的树：

gremlin> g = new TinkerGraph()                                                                 
==>tinkergraph[vertices:0 edges:0]
gremlin> v1 = g.addVertex()                                                                    
==>v[0]
gremlin> v2 = g.addVertex()                                                                    
==>v[1]
gremlin> v3 = g.addVertex([count:2])                                                           
==>v[2]
gremlin> v4 = g.addVertex([count:3])                                                           
==>v[3]
gremlin> v1.addEdge('child',v2)                                                                
==>e[4][0-child->1]
gremlin> v1.addEdge('child',v3)                                                                
==>e[5][0-child->2]
gremli                                                                                         
gremlin> v2.addEdge('child',v4)
==>e[6][1-child->3]

然后是整个树中每个子树的计算：

gremlin> g.V().filter{it.outE().hasNext()}.sideEffect{                                           
gremlin>   c=0;                                                                                  
gremlin>   it.as('a').out().sideEffect{leaf -> c+=(leaf.getProperty('count')?:0)}.loop('a'){true}.iterate()
gremlin>   it.setProperty('total',c)                                                                       
gremlin> }                                                                                                 
==>v[0]
==>v[1]
gremlin> g.v(0).total
==>5
gremlin> g.v(1).total                                                                                      
==>3

该查询像这样分解。首先，这件作品：

g.V().filter{it.outE().hasNext()}

获取树的任何不是叶子节点的部分（即，应该至少有一个出边不是叶子）。其次，我们使用sideEffect处理子树的每个根：

it.as('a').out().sideEffect{leaf -> c+=(leaf.getProperty('count')?:0)}.loop('a'){true}.iterate()

将每个子树的“count”属性的总和存储在名为c 的变量中。使用 elvis 运算符 (?:) 可以检查没有“计数”属性的顶点并在这些情况下返回零，这有一点非常棒的优点。在遍历树计算 c 后，您可以通过以下方式将 c 的值存储在子树的根节点中：

it.setProperty('total',c)