Added CsvOutput

which serializes any `Input` to CSV files.

community/import-tool/src/test/java/org/neo4j/tooling/CsvOutput.java

@@ -0,0 +1,144 @@
+/*
+ * Copyright (c) 2002-2016 "Neo Technology,"
+ * Network Engine for Objects in Lund AB [http://neotechnology.com]
+ *
+ * This file is part of Neo4j.
+ *
+ * Neo4j is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program.  If not, see <http://www.gnu.org/licenses/>.
+ */
+package org.neo4j.tooling;
+
+import java.io.File;
+import java.io.IOException;
+import java.io.PrintStream;
+import java.util.function.Function;
+
+import org.neo4j.unsafe.impl.batchimport.BatchImporter;
+import org.neo4j.unsafe.impl.batchimport.InputIterable;
+import org.neo4j.unsafe.impl.batchimport.InputIterator;
+import org.neo4j.unsafe.impl.batchimport.input.Input;
+import org.neo4j.unsafe.impl.batchimport.input.InputEntity;
+import org.neo4j.unsafe.impl.batchimport.input.csv.Configuration;
+import org.neo4j.unsafe.impl.batchimport.input.csv.Deserialization;
+import org.neo4j.unsafe.impl.batchimport.input.csv.Header;
+
+public class CsvOutput implements BatchImporter
+{
+    private final File targetDirectory;
+    private final Header nodeHeader;
+    private final Header relationshipHeader;
+    private final Deserialization<String> deserialization;
+
+    public CsvOutput( File targetDirectory, Header nodeHeader, Header relationshipHeader, Configuration config )
+    {
+        this.targetDirectory = targetDirectory;
+        assert targetDirectory.isDirectory();
+        this.nodeHeader = nodeHeader;
+        this.relationshipHeader = relationshipHeader;
+        this.deserialization = new StringDeserialization( config );
+        targetDirectory.mkdirs();
+    }
+
+    @Override
+    public void doImport( Input input ) throws IOException
+    {
+        consume( "nodes.csv", input.nodes(), nodeHeader, (node) ->
+        {
+            deserialization.clear();
+            for ( Header.Entry entry : nodeHeader.entries() )
+            {
+                switch ( entry.type() )
+                {
+                case ID:
+                    deserialization.handle( entry, node.id() );
+                    break;
+                case PROPERTY:
+                    deserialization.handle( entry, property( node, entry.name() ) );
+                    break;
+                case LABEL:
+                    deserialization.handle( entry, node.labels() );
+                    break;
+                }
+            }
+            return deserialization.materialize();
+        } );
+        consume( "relationships.csv", input.relationships(), relationshipHeader, (relationship) ->
+        {
+            deserialization.clear();
+            for ( Header.Entry entry : relationshipHeader.entries() )
+            {
+                switch ( entry.type() )
+                {
+                case PROPERTY:
+                    deserialization.handle( entry, property( relationship, entry.name() ) );
+                    break;
+                case TYPE:
+                    deserialization.handle( entry, relationship.type() );
+                    break;
+                case START_ID:
+                    deserialization.handle( entry, relationship.startNode() );
+                    break;
+                case END_ID:
+                    deserialization.handle( entry, relationship.endNode() );
+                    break;
+                }
+            }
+            return deserialization.materialize();
+        } );
+    }
+
+    private Object property( InputEntity entity, String key )
+    {
+        Object[] properties = entity.properties();
+        for ( int i = 0; i < properties.length; i += 2 )
+        {
+            if ( properties[i].equals( key ) )
+            {
+                return properties[i+1];
+            }
+        }
+        return null;
+    }
+
+    private <ENTITY extends InputEntity> void consume( String name, InputIterable<ENTITY> entities, Header header,
+            Function<ENTITY,String> deserializer ) throws IOException
+    {
+        try ( PrintStream out = file( name ) )
+        {
+            serialize( out, header );
+            try ( InputIterator<ENTITY> iterator = entities.iterator() )
+            {
+                while ( iterator.hasNext() )
+                {
+                    out.println( deserializer.apply( iterator.next() ) );
+                }
+            }
+        }
+    }
+
+    private void serialize( PrintStream out, Header header )
+    {
+        deserialization.clear();
+        for ( Header.Entry entry : header.entries() )
+        {
+            deserialization.handle( entry, entry.toString() );
+        }
+        out.println( deserialization.materialize() );
+    }
+
+    private PrintStream file( String name ) throws IOException
+    {
+        return new PrintStream( new File( targetDirectory, name ) );
+    }
+}

community/import-tool/src/test/java/org/neo4j/tooling/QuickImport.java

@@ -76,8 +76,9 @@ public static void main( String[] arguments ) throws IOException
         int relationshipTypeCount = args.getNumber( "relationship-types", 4 ).intValue();
         File dir = new File( args.get( ImportTool.Options.STORE_DIR.key() ) );
         long randomSeed = args.getNumber( "random-seed", currentTimeMillis() ).longValue();
+        Configuration config = COMMAS;
 
-        Extractors extractors = new Extractors( COMMAS.arrayDelimiter() );
+        Extractors extractors = new Extractors( config.arrayDelimiter() );
         IdType idType = IdType.valueOf( args.get( "id-type", IdType.ACTUAL.name() ) );
 
         Header nodeHeader = parseNodeHeader( args, idType, extractors );
@@ -105,11 +106,20 @@ public int denseNodeThreshold()
                 nodeCount, relationshipCount,
                 generator.nodes(), generator.relationships(),
                 idType, silentBadCollector( 0 ) );
-        BatchImporter importer = new ParallelBatchImporter( dir, importConfig,
-                new SimpleLogService( sysoutLogProvider, sysoutLogProvider ),
-                defaultVisible(),
-                Config.defaults() );
-        importer.doImport( input );
+
+        BatchImporter consumer;
+        if ( args.getBoolean( "to-csv" ) )
+        {
+            consumer = new CsvOutput( dir, nodeHeader, relationshipHeader, config );
+        }
+        else
+        {
+            consumer = new ParallelBatchImporter( dir, importConfig,
+                    new SimpleLogService( sysoutLogProvider, sysoutLogProvider ),
+                    defaultVisible(),
+                    Config.defaults() );
+        }
+        consumer.doImport( input );
     }
 
     private static Header parseNodeHeader( Args args, IdType idType, Extractors extractors )