Use list to handle multi-URIs parameter

Fixes #16

Author: acogoluegnes

src/main/java/com/rabbitmq/perf/MulticastSet.java

@@ -22,6 +22,7 @@
 import java.net.URISyntaxException;
 import java.security.KeyManagementException;
 import java.security.NoSuchAlgorithmException;
+import java.util.List;
 import java.util.Random;
 import java.util.UUID;
 import java.util.concurrent.TimeoutException;
@@ -32,12 +33,12 @@ public class MulticastSet {
     private final ConnectionFactory factory;
     private final MulticastParams params;
     private final String testID;
-    private final String [] uris;
+    private final List<String> uris;
 
     private final Random random = new Random();
 
     public MulticastSet(Stats stats, ConnectionFactory factory,
-        MulticastParams params, String [] uris) {
+        MulticastParams params, List<String> uris) {
         if (params.getRoutingKey() == null) {
             this.id = UUID.randomUUID().toString();
         } else {
@@ -51,7 +52,7 @@ public MulticastSet(Stats stats, ConnectionFactory factory,
     }
 
     public MulticastSet(Stats stats, ConnectionFactory factory,
-        MulticastParams params, String testID, String [] uris) {
+        MulticastParams params, String testID, List<String> uris) {
         if (params.getRoutingKey() == null) {
             this.id = UUID.randomUUID().toString();
         } else {
@@ -139,7 +140,7 @@ private void setUri() throws URISyntaxException, NoSuchAlgorithmException, KeyMa
     }
 
     private String uri() {
-        String uri = uris[random.nextInt(uris.length)];
+        String uri = uris.get(random.nextInt(uris.size()));
         return uri;
     }
 }

src/main/java/com/rabbitmq/perf/PerfTest.java

@@ -18,6 +18,7 @@
 import java.text.SimpleDateFormat;
 import java.util.Arrays;
 import java.util.Calendar;
+import java.util.Collections;
 import java.util.List;
 
 import org.apache.commons.cli.CommandLine;
@@ -76,14 +77,15 @@ public static void main(String[] args) {
 
             String uri               = strArg(cmd, 'h', "amqp://localhost");
             String urisParameter     = strArg(cmd, 'H', null);
-            String [] uris = null;
+            List<String> uris = null;
             if(urisParameter != null) {
-                uris = urisParameter.split(",");
-                for(int i = 0; i< uris.length; i++) {
-                    uris[i] = uris[i].trim();
+                String [] urisArray = urisParameter.split(",");
+                for(int i = 0; i< urisArray.length; i++) {
+                    urisArray[i] = urisArray[i].trim();
                 }
+                uris = Arrays.asList(urisArray);
             } else {
-                uris = new String [] {uri};
+                uris = Collections.singletonList(uri);
             }
 
             //setup
@@ -97,7 +99,7 @@ public static void main(String[] args) {
 
             ConnectionFactory factory = new ConnectionFactory();
             factory.setShutdownTimeout(0); // So we still shut down even with slow consumers
-            factory.setUri(uris[0]);
+            factory.setUri(uris.get(0));
             factory.setRequestedFrameMax(frameMax);
             factory.setRequestedHeartbeat(heartbeat);