Add the possibility to load multiple schemas

William-Hill-Online · Jul 11, 2017 · ca2e545 · ca2e545
1 parent 8214579
commit ca2e545
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/cassandra-unit/src/main/java/org/cassandraunit/cli/CassandraUnitCommandLineStarter.java b/cassandra-unit/src/main/java/org/cassandraunit/cli/CassandraUnitCommandLineStarter.java
@@ -56,7 +56,7 @@ private static boolean parseCommandLine(String[] args) {
     protected static void load() {
         System.out.println("Starting Cassandra...");
         String port = commandLine.getOptionValue("p");
-        String schema = commandLine.getOptionValue("s");
+        String schemaOption = commandLine.getOptionValue("s");
         String installationFolder = commandLine.getOptionValue("d");
         String timeout = commandLine.getOptionValue("t");
 
@@ -77,8 +77,11 @@ protected static void load() {
 
         try {
             EmbeddedCassandraServerHelper.startEmbeddedCassandra(new File(installationFolder, CASSANDRA_YAML), "temp", Long.parseLong(timeout));
-            if (hasValidValue(schema)) {
-                dataSetLoad(LOCALHOST, port, schema);
+            if (hasValidValue(schemaOption)) {
+                String[] schemas = schemaOption.split(",");
+                for (String schema : schemas) {
+                    dataSetLoad(LOCALHOST, port, schema);
+                }
             }
         } catch (TTransportException | IOException e) {
             e.printStackTrace();