lib/mq.rb in amqp-0.6.6 vs lib/mq.rb in amqp-0.6.7

@@ -738,10 +738,11 @@
       @error_callback.call(msg) if @error_callback and msg
     end
   end
 
   def prefetch(size)
+    @prefetch_size = size
     send Protocol::Basic::Qos.new(:prefetch_size => 0, :prefetch_count => size, :global => false)
     self
   end
 
   # Asks the broker to redeliver all unacknowledged messages on this
@@ -805,9 +806,11 @@
     exs.each{ |_,e| e.reset } if exs
 
     qus = @queues
     @queues = {}
     qus.each{ |_,q| q.reset } if qus
+
+    prefetch(@prefetch_size) if @prefetch_size
   end
 
   private
 
   def log *args
\ No newline at end of file