ConfirmDontLoseMessages: make msgCount a command-line parameter

Author: Alexandru Scvortov

test/src/com/rabbitmq/examples/ConfirmDontLoseMessages.java

@@ -46,18 +46,24 @@
 import java.io.IOException;
 
 public class ConfirmDontLoseMessages {
-    final static int MSG_COUNT = 10000;
+    static int msgCount = 10000;
     final static String QUEUE_NAME = "confirm-test";
     static ConnectionFactory connectionFactory;
 
     public static void main(String[] args)
         throws IOException, InterruptedException
     {
+        if (args.length > 0) {
+                msgCount = Integer.parseInt(args[0]);
+        }
+
+        System.out.printf("msgCount = %d\n", msgCount);
+
         connectionFactory = new ConnectionFactory();
 
-        // Publish MSG_COUNT messages and wait for confirms.
+        // Publish msgCount messages and wait for confirms.
         (new Thread(new Consumer())).start();
-        // Consume MSG_COUNT messages.
+        // Consume msgCount messages.
         (new Thread(new Publisher())).start();
     }
 
@@ -87,7 +93,7 @@ public void handleAck(long seqNo,
                 ch.confirmSelect();
 
                 // Publish
-                for (long i = 0; i < MSG_COUNT; ++i) {
+                for (long i = 0; i < msgCount; ++i) {
                     ackSet.add(i);
                     ch.basicPublish("", QUEUE_NAME,
                                     MessageProperties.PERSISTENT_BASIC,
@@ -124,7 +130,7 @@ public void run() {
                 // Consume
                 QueueingConsumer qc = new QueueingConsumer(ch);
                 ch.basicConsume(QUEUE_NAME, true, qc);
-                for (int i = 0; i < MSG_COUNT; ++i) {
+                for (int i = 0; i < msgCount; ++i) {
                     qc.nextDelivery();
                 }