Merge pull request #22 from cloudamqp/add-z-option

Author: lukas8219

src/amqpcat.cr

@@ -13,6 +13,10 @@ class AMQPCat
     @client = AMQP::Client.new(u)
   end
 
+  def connect
+    @client.connect
+  end
+
   def produce(exchange : String, routing_key : String, exchange_type : String, publish_confirm : Bool, props : AMQP::Client::Properties)
     STDIN.blocking = false
     loop do

src/cli.cr

@@ -61,6 +61,7 @@ p = OptionParser.parse do |parser|
   parser.on("--content-encoding=ENC", "Content encoding header") { |v| props.content_encoding = v }
   parser.on("--priority=LEVEL", "Priority header") { |v| props.priority = v.to_u8? || abort "Priority must be between 0 and 255" }
   parser.on("--expiration=TIME", "Expiration header (ms before msg is dead lettered)") { |v| props.expiration = v }
+  parser.on("-z", "Exit on connection") { mode = :connect }
   parser.on("-v", "--version", "Display version") { puts AMQPCat::VERSION; exit 0 }
   parser.on("-h", "--help", "Show this help message") { puts parser; exit 0 }
   parser.invalid_option do |flag|
@@ -70,6 +71,7 @@ p = OptionParser.parse do |parser|
 end
 
 cat = AMQPCat.new(uri)
+
 case mode
 when :producer
   unless exchange || queue
@@ -93,4 +95,7 @@ when :rpc
     abort p
   end
   cat.rpc(exchange, routing_key.not_nil!, exchange_type, format)
+when :connect
+  cat.connect
+  exit 0
 end