Added support for specifying the RabbitMQ heartbeat

jplock · Nov 6, 2012 · 8192caa · 8192caa
1 parent ef7ca5b
commit 8192caa
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 1 deletion.
diff --git a/example-config.yml b/example-config.yml
@@ -8,7 +8,8 @@ Octobot:
         priority: 5,
         workers: 1,
         username: cilantro,
-        password: burrito
+        password: burrito,
+        heartbeat: 10
       }
 
   metrics_port: 1228

diff --git a/src/main/java/com/urbanairship/octobot/Queue.java b/src/main/java/com/urbanairship/octobot/Queue.java
@@ -12,6 +12,7 @@ public class Queue {
     public String username;
     public String password;
     public String vhost;
+    public Integer heartbeat;
 
     public Queue(String queueType, String queueName, String host, Integer port,
         String username, String password) {
@@ -41,6 +42,8 @@ public Queue(HashMap<String, Object> config) {
 
         if (config.get("port") != null)
             this.port = Integer.parseInt(((Long) config.get("port")).toString());
+        if (config.get("heartbeat") != null)
+            this.heartbeat = Integer.parseInt(((Long) config.get("heartbeat")).toString());
     }
 
 

diff --git a/src/main/java/com/urbanairship/octobot/RabbitMQ.java b/src/main/java/com/urbanairship/octobot/RabbitMQ.java
@@ -22,6 +22,9 @@ public RabbitMQ(Queue queue) {
         factory.setUsername(queue.username);
         factory.setPassword(queue.password);
         factory.setVirtualHost(queue.vhost);
+        if (queue.heartbeat != null) {
+            factory.setRequestedHeartbeat(queue.heartbeat);
+        }
     }
 
     // Returns a new connection to an AMQP queue.