Sha256: 9fca2e41d05b3b8dd04dc9fe8da33549d94c57f69937102ba7128041ea095d85

Contents?: true

Size: 888 Bytes

Versions: 8

Compression:

Stored size: 888 Bytes

Contents

require 'cellect/node_set'

module Cellect
  module Client
    class NodeSet < Cellect::NodeSet
      attr_accessor :nodes

      # Sets up an empty node set
      def initialize(zk_url = nil)
        self.nodes = { }
        super
      end

      protected

      # Respond to a node coming online or timing out
      def nodes_changed(nodes)
        self.nodes = { }
        nodes.each do |node|
          self.nodes[node] = zk.get("/nodes/#{ node }").first
        end
      end

      # Register with ZooKeeper and get the list of nodes
      def setup
        watch_nodes
        zk.mkdir_p '/nodes'
        nodes_changed zk.children('/nodes', watch: true)
      end

      # Watch ZooKeeper for changes to the node set
      def watch_nodes
        zk.register('/nodes') do |event|
          nodes_changed zk.children('/nodes', watch: true)
        end
      end
    end
  end
end

Version data entries

8 entries across 8 versions & 1 rubygems

Version Path
cellect-client-1.3.3 lib/cellect/client/node_set.rb
cellect-client-1.3.2 lib/cellect/client/node_set.rb
cellect-client-1.3.1 lib/cellect/client/node_set.rb
cellect-client-1.3.0 lib/cellect/client/node_set.rb
cellect-client-1.2.0 lib/cellect/client/node_set.rb
cellect-client-1.1.0 lib/cellect/client/node_set.rb
cellect-client-1.0.1 lib/cellect/client/node_set.rb
cellect-client-1.0.0 lib/cellect/client/node_set.rb