convenience.rb

module Papertrail
  module Zookeeper
    module Convenience
      def assign(key, value, options = {})
        set(key, value, options)
      rescue ZK::Exceptions::NoNode
        begin
          create(key, value, options)
        rescue ZK::Exceptions::NodeExists
          set(key, value, options)
        rescue ZK::Exceptions::NoNode
          parent = File.dirname(key)
          if parent && parent != '.' && parent != '' && parent != '/'
            mkdir_p(parent)
            retry
          else
            raise
          end
        end
      end

      def get_value(key)
        result, stat = get(key)
        result
      rescue ZK::Exceptions::NoNode
      end
    end
  end
end