Handle expired zookeeper sessions gracefully

liquidm · Apr 5, 2013 · 350972b · 350972b
1 parent f597ae2
commit 350972b
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/lib/druid/zoo_handler.rb b/lib/druid/zoo_handler.rb
@@ -11,6 +11,14 @@ def initialize(uri, opts = {})
       @discovery_path = opts[:discovery_path] || '/discoveryPath'
       @watched_services = Hash.new
 
+      init_zookeeper
+    end
+
+    def init_zookeeper
+      @zk.on_expired_session do
+        init_zookeeper
+      end
+
       @zk.register(@discovery_path, :only => :child) do |event|
         check_services
       end

diff --git a/ruby-druid.gemspec b/ruby-druid.gemspec
@@ -2,7 +2,7 @@ require 'rake'
 
 Gem::Specification.new do |gem|
   gem.name          = 'ruby-druid'
-  gem.version       = '0.0.5'
+  gem.version       = '0.0.6'
   gem.date          = '2013-04-04'
   gem.summary       = 'Ruby client for druid'
   gem.description   = 'Ruby client for metamx druid'