ruby-redis 0.0.1

data/lib/redis/pubsub.rb

@@ -0,0 +1,153 @@
+require File.expand_path '../redis', File.dirname(__FILE__)
+require 'eventmachine'
+
+class Redis
+  
+  module PubSub
+    
+    class Subscription
+      include EventMachine::Deferrable
+      
+      @@channels ||= {}
+      @@pchannels ||= {}
+      
+      def self.publish channel, message
+        subs = @@channels[channel] || []
+        subs.each do |sub|
+          sub.call channel, message
+        end
+        sent = subs.size
+        (@@pchannels || []).each do |pchannel, subs|
+          next unless File.fnmatch(pchannel, channel)
+          subs.each do |sub|
+            sub.pcall pchannel, channel, message
+          end
+          sent += subs.size
+        end
+        sent
+      end
+      
+      def initialize connection
+        @connection = connection
+        @subs = []
+        @psubs = []
+      end
+      
+      def bound
+        size > 0
+      end
+      
+      def size
+        @subs.size + @psubs.size
+      end
+      
+      def unbind
+        unbind_channels
+        unbind_patterns
+      end
+      
+      def unbind_channels
+        @subs.each do |channel|
+          (@@channels[channel] || []).delete self
+        end
+        @subs.clear
+      end
+      
+      def unbind_patterns
+        @psubs.each do |channel|
+          (@@pchannels[channel] || []).delete self
+        end
+        @psubs.clear
+      end
+      
+      def subscribe channel
+        c = (@@channels[channel] ||= [])
+        unless c.include? self
+          c << self 
+          @subs << channel
+        end
+        @connection.send_redis ['subscribe', channel, size]
+      end
+      
+      def psubscribe channel
+        c = (@@pchannels[channel] ||= [])
+        unless c.include? self
+          c << self 
+          @psubs << channel
+        end
+        @connection.send_redis ['psubscribe', channel, size]
+      end
+
+      def unsubscribe channel
+        c = (@@channels[channel] || [])
+        @subs.delete channel if c.delete self
+        @connection.send_redis ['unsubscribe', channel, size]
+      end
+      
+      def punsubscribe channel
+        c = (@@pchannels[channel] || [])
+        @psubs.delete channel if c.delete self
+        @connection.send_redis ['punsubscribe', channel, size]
+      end
+      
+      def call channel, message
+        @connection.send_redis ['message', channel, message]
+      end
+
+      def pcall pchannel, channel, message
+        @connection.send_redis ['pmessage', pchannel, channel, message]
+      end
+      
+    end
+    
+    def redis_SUBSCRIBE *channels
+      sub = @deferred
+      sub = Subscription.new self unless Subscription === sub
+      channels.each do |channel|
+        sub.subscribe channel
+      end
+      sub
+    end
+    
+    def redis_UNSUBSCRIBE *channels
+      sub = @deferred
+      sub = Subscription.new self unless Subscription === sub
+      if channels.empty?
+        sub.unbind_channels
+      else
+        channels.each do |channel|
+          sub.unsubscribe channel
+        end
+      end
+      sub
+    end
+    
+    def redis_PSUBSCRIBE *channels
+      sub = @deferred
+      sub = Subscription.new self unless Subscription === sub
+      channels.each do |channel|
+        sub.psubscribe channel
+      end
+      sub
+    end
+    
+    def redis_PUNSUBSCRIBE *channels
+      sub = @deferred
+      sub = Subscription.new self unless Subscription === sub
+      if channels.empty?
+        sub.unbind_patterns
+      else
+        channels.each do |channel|
+          sub.punsubscribe channel
+        end
+      end
+      sub
+    end
+    
+    def redis_PUBLISH channel, message
+      Subscription.publish channel, message
+    end
+    
+      
+  end
+end