Use heartbeats in demo_relay.

majek · Oct 22, 2009 · ebb0bdb · ebb0bdb
1 parent 208e490
commit ebb0bdb
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/examples/demo_relay.py b/examples/demo_relay.py
@@ -171,6 +171,8 @@ def handle_delivery(self, method, header, body):
                       help="set password number for subsequent server definitions")
     parser.add_option("-v", "--virtual-host",
                       help="set virtual-host for subsequent server definitions")
+    parser.add_option("-H", "--heartbeat", type="int", default=3,
+                      help="set heartbeat for subsequent server definitions")
 
     def spec_for(id_and_host, parser):
         try:
@@ -189,7 +191,8 @@ def spec_for(id_and_host, parser):
         return (id, pika.ConnectionParameters(hostname,
                                               parser.values.port or None,
                                               parser.values.virtual_host or "/",
-                                              creds))
+                                              creds,
+                                              heartbeat = parser.values.heartbeat or 0))
 
     def set_source(option, opt_str, value, parser):
         setattr(parser.values, "source", spec_for(value, parser))