lib/waffle/transports/rabbitmq.rb in waffle-0.5.0 vs lib/waffle/transports/rabbitmq.rb in waffle-0.5.1

@@ -9,18 +9,18 @@
       protected
       def publish_impl(flow = 'events', message = '')
         exchange.publish(encoder.encode(message), :key => flow)
       end
 
-      def subscribe_impl(flow = 'events')
+      def subscribe_impl(flow = 'events', &block)
         if flow.is_a?(Array)
           flow.each{|f| queue.bind(exchange, :key => f)}
         else
           queue.bind(exchange, :key => flow)
         end
 
         queue.subscribe do |message|
-          yield(message[:delivery_details][:routing_key], encoder.decode(message[:payload]))
+          block.call(message[:delivery_details][:routing_key], encoder.decode(message[:payload]))
         end
       end
 
       def connection_exceptions
         [Bunny::ServerDownError, Bunny::ConnectionError, Errno::ECONNRESET]