Sha256: 6c5822a35a0e916924263264a29b5fa6555901978065eacb4669eb37974ac0ad

Contents?: true

Size: 1.96 KB

Versions: 17

Compression:

Stored size: 1.96 KB

Contents

# frozen_string_literal: true

class Redis
  class SubscribedClient
    def initialize(client)
      @client = client
    end

    def call(command)
      @client.process([command])
    end

    def subscribe(*channels, &block)
      subscription("subscribe", "unsubscribe", channels, block)
    end

    def subscribe_with_timeout(timeout, *channels, &block)
      subscription("subscribe", "unsubscribe", channels, block, timeout)
    end

    def psubscribe(*channels, &block)
      subscription("psubscribe", "punsubscribe", channels, block)
    end

    def psubscribe_with_timeout(timeout, *channels, &block)
      subscription("psubscribe", "punsubscribe", channels, block, timeout)
    end

    def unsubscribe(*channels)
      call([:unsubscribe, *channels])
    end

    def punsubscribe(*channels)
      call([:punsubscribe, *channels])
    end

    protected

    def subscription(start, stop, channels, block, timeout = 0)
      sub = Subscription.new(&block)

      unsubscribed = false

      @client.call_loop([start, *channels], timeout) do |line|
        type, *rest = line
        sub.callbacks[type].call(*rest)
        unsubscribed = type == stop && rest.last == 0
        break if unsubscribed
      end
      # No need to unsubscribe here. The real client closes the connection
      # whenever an exception is raised (see #ensure_connected).
    end
  end

  class Subscription
    attr :callbacks

    def initialize
      @callbacks = Hash.new do |hash, key|
        hash[key] = ->(*_) {}
      end

      yield(self)
    end

    def subscribe(&block)
      @callbacks["subscribe"] = block
    end

    def unsubscribe(&block)
      @callbacks["unsubscribe"] = block
    end

    def message(&block)
      @callbacks["message"] = block
    end

    def psubscribe(&block)
      @callbacks["psubscribe"] = block
    end

    def punsubscribe(&block)
      @callbacks["punsubscribe"] = block
    end

    def pmessage(&block)
      @callbacks["pmessage"] = block
    end
  end
end

Version data entries

17 entries across 17 versions & 2 rubygems

Version Path
blacklight-spotlight-3.6.0.beta8 vendor/bundle/ruby/3.2.0/gems/redis-4.8.1/lib/redis/subscribe.rb
redis-4.8.1 lib/redis/subscribe.rb
redis-4.8.0 lib/redis/subscribe.rb
redis-4.7.1 lib/redis/subscribe.rb
redis-4.7.0 lib/redis/subscribe.rb
redis-4.6.0 lib/redis/subscribe.rb
redis-4.5.1 lib/redis/subscribe.rb
redis-4.5.0 lib/redis/subscribe.rb
redis-4.4.0 lib/redis/subscribe.rb
redis-4.3.1 lib/redis/subscribe.rb
redis-4.3.0 lib/redis/subscribe.rb
redis-4.2.5 lib/redis/subscribe.rb
redis-4.2.4 lib/redis/subscribe.rb
redis-4.2.3 lib/redis/subscribe.rb
redis-4.2.2 lib/redis/subscribe.rb
redis-4.2.1 lib/redis/subscribe.rb
redis-4.2.0 lib/redis/subscribe.rb