Add Subscription#each for convenient message handling.

Author: ioquatix

lib/async/redis/context/subscribe.rb

@@ -31,6 +31,14 @@ def listen
 					end
 				end
 
+				def each
+					return to_enum unless block_given?
+					
+					while response = self.listen
+						yield response
+					end
+				end
+				
 				def subscribe(channels)
 					@connection.write_request ['SUBSCRIBE', *channels]
 					@connection.flush

test/async/redis/context/subscribe.rb

@@ -68,4 +68,26 @@
 	ensure
 		subscription.close
 	end
+	
+	with "#each" do
+		it "should iterate over messages" do
+			subscription = client.subscribe(news_channel)
+			
+			listener = reactor.async do
+				subscription.each do |type, name, message|
+					expect(type).to be == 'message'
+					expect(name).to be == news_channel
+					expect(message).to be == 'Hello'
+					
+					break
+				end
+			end
+			
+			client.publish(news_channel, 'Hello')
+			
+			listener.wait
+		ensure
+			subscription.close
+		end
+	end
 end