MulticastMain routing key

Author: Emile Joubert

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

@@ -48,6 +48,7 @@ public static void main(String[] args) {
             String exchangeType  = strArg(cmd, 't', "direct");
             String exchangeName  = strArg(cmd, 'e', exchangeType);
             String queueName     = strArg(cmd, 'u', "");
+            String routingKey    = strArg(cmd, 'k', null);
             int samplingInterval = intArg(cmd, 'i', 1);
             float rateLimit      = floatArg(cmd, 'r', 0.0f);
             int producerCount    = intArg(cmd, 'x', 1);
@@ -104,6 +105,7 @@ public static void main(String[] args) {
             p.setProducerMsgCount( producerMsgCount);
             p.setProducerTxSize(   producerTxSize);
             p.setQueueName(        queueName);
+            p.setRoutingKey(       routingKey);
             p.setRateLimit(        rateLimit);
             p.setTimeLimit(        timeLimit);
 
@@ -134,6 +136,7 @@ private static Options getOptions() {
         options.addOption(new Option("t", "type",          true, "exchange type"));
         options.addOption(new Option("e", "exchange",      true, "exchange name"));
         options.addOption(new Option("u", "queue",         true, "queue name"));
+        options.addOption(new Option("k", "routingKey",    true, "routing key"));
         options.addOption(new Option("i", "interval",      true, "sampling interval"));
         options.addOption(new Option("r", "rate",          true, "rate limit"));
         options.addOption(new Option("x", "producers",     true, "producer count"));

test/src/com/rabbitmq/examples/perf/MulticastParams.java

@@ -43,6 +43,7 @@ public class MulticastParams {
     private String exchangeName = "direct";
     private String exchangeType = "direct";
     private String queueName = "";
+    private String routingKey = null;
 
     private List<?> flags = new ArrayList<Object>();
 
@@ -65,6 +66,10 @@ public void setQueueName(String queueName) {
         this.queueName = queueName;
     }
 
+    public void setRoutingKey(String routingKey) {
+        this.routingKey = routingKey;
+    }
+
     public void setRateLimit(float rateLimit) {
         this.rateLimit = rateLimit;
     }
@@ -150,6 +155,10 @@ public int getMinMsgSize() {
         return minMsgSize;
     }
 
+    public String getRoutingKey() {
+        return routingKey;
+    }
+
     public Producer createProducer(Connection connection, Stats stats, String id) throws IOException {
         Channel channel = connection.createChannel();
         if (producerTxSize > 0) channel.txSelect();

test/src/com/rabbitmq/examples/perf/MulticastSet.java

@@ -31,7 +31,11 @@ public class MulticastSet {
 
     public MulticastSet(Stats stats, ConnectionFactory factory,
                         MulticastParams params) {
-        this.id = UUID.randomUUID().toString();
+        if (params.getRoutingKey() == null) {
+            this.id = UUID.randomUUID().toString();
+        } else {
+            this.id = params.getRoutingKey();
+        }
         this.stats = stats;
         this.factory = factory;
         this.params = params;