Contents

require 'zk'

module Cellect
  class NodeSet
    include Celluloid
    ConnectionError = Class.new(StandardError)

    attr_accessor :zk, :state

    # Sets up the node set and starts connecting to ZooKeeper
    def initialize(zk_url = nil)
      @zk_url = zk_url
      self.state = :initializing
      after(0.001){ async.initialize_zk } # don't block waiting for ZK to connect
    end

    # Connect to ZooKeeper, setup this node, and change state
    def initialize_zk
      Timeout::timeout(timeout_duration) do
        self.zk = ZK.new zk_url, chroot: '/cellect'
      end
      setup
      self.state = :ready
    end

    def ready?
      state == :ready
    end

    protected

    def timeout_duration
      ENV.fetch('ZK_TIMEOUT', 5).to_i
    end

    def zk_url
      @zk_url || ENV.fetch('ZK_URL', 'localhost:2181')
    end

    def setup
      # Specialized in subclasses
    end
  end
end

Version data entries

8 entries across 8 versions & 2 rubygems

Version	Path
cellect-client-1.3.3	lib/cellect/node_set.rb
cellect-server-1.3.3	lib/cellect/node_set.rb
cellect-client-1.3.2	lib/cellect/node_set.rb
cellect-server-1.3.2	lib/cellect/node_set.rb
cellect-client-1.3.1	lib/cellect/node_set.rb
cellect-server-1.3.1	lib/cellect/node_set.rb
cellect-client-1.3.0	lib/cellect/node_set.rb
cellect-server-1.3.0	lib/cellect/node_set.rb