path: root/lib/action_cable/channel/redis.rb

module ActionCable
  module Channel

    module Redis
      extend ActiveSupport::Concern

      included do
        on_unsubscribe :unsubscribe_from_redis_channels
      end

      def subscribe_to(redis_channel, callback = nil)
        @_redis_channels ||= []
        @_redis_channels << redis_channel

        callback ||= -> (message) { broadcast ActiveSupport::JSON.decode(message) }
        redis.pubsub.subscribe(redis_channel, &callback)
      end

      protected
        def unsubscribe_from_redis_channels
          if @_redis_channels
            @_redis_channels.each { |channel| @connection.pubsub.unsubscribe(channel) }
          end
        end

        def redis
          @connection.redis
        end
    end

  end
end