jones 0.1.0

data/lib/client.rb

@@ -0,0 +1,124 @@
+=begin
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+=end
+
+
+require 'forwardable'
+require 'json'
+require 'socket'
+require 'zk'
+
+
+class JonesClient
+  extend Forwardable
+  attr_accessor :data
+  def_delegators :@data, :[]
+
+  def initialize(options = {})
+    @data = nil
+    @callbacks = []
+    @conf_sub = nil
+    @conf_path = nil
+    @logger = Logger.new(STDOUT)
+
+    parse_options(options)
+    setup_zk
+    read_nodemap
+  end
+
+  def register_callback(&block)
+    @callbacks += [block]
+  end
+
+  private
+
+  def nodemap_changed(event)
+    if event.node_changed?
+      read_nodemap
+    else
+      @logger.error("Unknown ZK node event: #{event.inspect}")
+    end
+  end
+
+  def read_nodemap
+    data = @zk.get(nodemap_path, :watch => true).first
+    if data.nil?
+      conf_path = conf_root
+    else
+      mapping = JSON.load(data)
+      conf_path = mapping.fetch(@host, conf_root)
+    end
+
+    setup_conf(conf_path)
+  end
+
+  def setup_conf(conf_path)
+    if conf_path != @conf_path
+      # path changed so we need to register a new handler
+      @conf_path = conf_path
+      @conf_sub.unsubscribe unless @conf_sub.nil?
+      @conf_sub = @zk.register(@conf_path) { |event| conf_changed(event) }
+      read_conf
+    end
+  end
+
+  def conf_changed(event)
+    if event.node_changed?
+      read_conf
+    else
+      logger.error("Unknown ZK node event: #{event.inspect}")
+    end
+  end
+
+  def read_conf()
+    @data = JSON.load(@zk.get(@conf_path, :watch => true).first)
+    @callbacks.each { |cb| cb.call(@data) }
+  end
+
+  def setup_zk
+    @zk = ZK.new(@zkservers) if @zkservers
+    @zk.on_expired_session { setup_zk }
+    @zk.register(nodemap_path, :only => :changed) { |event| nodemap_changed(event) }
+    # reset watch in case of disconnect
+    @zk.on_connected { read_nodemap }
+  end
+
+  def parse_options(options)
+    @zk, @zkservers = options.values_at(:zk, :zkservers)
+    if [@zk, @zkservers].all? || [@zk, @zkservers].none?
+      raise ArgumentError, 'must specify :zk or :zkservers'
+    end
+
+    @service = options[:service]
+    @host = options.fetch(:host, Socket.gethostname)
+    @zkroot = "/services/#{options[:service]}"
+  end
+
+  def nodemap_path
+    "#{@zkroot}/nodemaps"
+  end
+
+  def conf_root
+    "#{@zkroot}/conf"
+  end
+end
+
+
+#zk = ZK.new('localhost:2181')
+
+jc = JonesClient.new(:zkservers => 'localhost:2181', :service => 'test')
+jc.register_callback { |data| puts data['field'] }
+puts jc['field']
+
+sleep(20)
+puts jc['field']

metadata

@@ -0,0 +1,62 @@
+--- !ruby/object:Gem::Specification
+name: jones
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+  prerelease: 
+platform: ruby
+authors:
+- Matthew Hooker
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2013-08-05 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: zk
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - '='
+      - !ruby/object:Gem::Version
+        version: 1.8.0
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - '='
+      - !ruby/object:Gem::Version
+        version: 1.8.0
+description: 
+email: mwhooker@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/client.rb
+homepage: https://github.com/mwhooker/jones
+licenses:
+- Apache 2.0
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.23
+signing_key: 
+specification_version: 3
+summary: Client for Jones configuration management server
+test_files: []