Search code examples
groovyrecursionjvmclosuresjvm-languages

How does Groovy handle closure scope and recursion?


I have a recursive Python function that builds a tree, and I'm trying to translate it into Groovy.

Here's the Python version...

def get_tree(vertices):
    results = []

    if type(vertices) != list:
        vertices = [vertices]

    for vertex in vertices:
        results.append(vertex)
        children = get_children(vertex)
        if children:
            child_tree = get_tree(children)
            results.append(child_tree)

    return results

Here's the output of get_tree(1)...

  [1, [2, 3, 4, [5, 3]]]

And here's my attempt to translate this into a Groovy closure...

_tree = { vertices ->

  results = []

  vertices.each() {
    results << it
    children = it."$direction"().toList()
    if (children) {
      child_tree = _tree(children)
      results << child_tree
    }
  }
  results
}

But this doesn't work -- this is what it returns...

gremlin> g.v(1).outTree()    
==>[v[5], v[3], (this Collection), (this Collection)]

What are those "this Collection"s about?

I have only a cursory understanding of Groovy, and I suspect it's something to do with how Groovy handles recursion and closure scope.

Please enlighten me :)


Solution

  • The solution was to add def to results = []:

    _tree = { vertices ->
    
      def results = []
    
      vertices.each() {
        results << it
        children = it."$direction"().toList()
        if (children) {
          child_tree = _tree(children)
          results << child_tree
        }
      }
      results
    }
    

    See https://groups.google.com/d/msg/gremlin-users/iCPUifiU_wk/mrUhsjOM2h0J