View Javadoc

1   /**
2    * Copyright (c) 2002-2011 "Neo Technology,"
3    * Network Engine for Objects in Lund AB [http://neotechnology.com]
4    *
5    * This file is part of Neo4j.
6    *
7    * Neo4j is free software: you can redistribute it and/or modify
8    * it under the terms of the GNU General Public License as published by
9    * the Free Software Foundation, either version 3 of the License, or
10   * (at your option) any later version.
11   *
12   * This program is distributed in the hope that it will be useful,
13   * but WITHOUT ANY WARRANTY; without even the implied warranty of
14   * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
15   * GNU General Public License for more details.
16   *
17   * You should have received a copy of the GNU General Public License
18   * along with this program.  If not, see <http://www.gnu.org/licenses/>.
19   */
20  package org.neo4j.server.webadmin.console;
21  
22  import groovy.lang.Closure;
23  
24  import java.util.Arrays;
25  import java.util.Collections;
26  import java.util.Iterator;
27  import java.util.Map;
28  
29  import org.codehaus.groovy.tools.shell.IO;
30  
31  /**
32   * Version of ResultHookClosure that does not add ==> to the output. This is
33   * instead moved to the console or print stream to handle, so that all content
34   * printed to the output stream gets the same prompt.
35   * 
36   * @author Marko A. Rodriguez (http://markorodriguez.com), Jacob Hansson
37   *         <jacob@voltvoodoo.com>
38   */
39  public class GremlinResultHook extends Closure
40  {
41      /**
42       * 
43       */
44      private static final long serialVersionUID = -8258126015903649440L;
45      private final IO io;
46  
47      public GremlinResultHook( final Object owner, final IO io )
48      {
49          super( owner );
50          this.io = io;
51      }
52  
53      @SuppressWarnings( "rawtypes" )
54      public Object call( final Object[] args )
55      {
56          Object result = args[0];
57          Iterator itty;
58          if ( result instanceof Iterator )
59          {
60              itty = (Iterator) result;
61          }
62          else if ( result instanceof Iterable )
63          {
64              itty = ( (Iterable) result ).iterator();
65          }
66          else if ( result instanceof Object[] )
67          {
68              itty = Arrays.asList( (Object[])result ).iterator();
69          }
70          else if ( result instanceof Map )
71          {
72              itty = ( (Map) result ).entrySet().iterator();
73          }
74          else if ( result == null)
75          {
76              itty = Collections.singleton( "" ).iterator();
77          }
78          else
79          {
80              itty = Collections.singleton( result ).iterator();
81          }
82  
83          while ( itty.hasNext() )
84          {
85              this.io.out.println( itty.next() );
86          }
87  
88          return null;
89      }
90  }