hiera-zookeeper 0.1.0

data/lib/hiera/backend/zookeeper_backend.rb

@@ -0,0 +1,64 @@
+class Hiera
+    module Backend
+        class Zookeeper_backend
+            # ZooKeeper backend for Hiera. Recognizes :server and :datadir config options. If no :server is specified, uses localhost:2181.
+            # Will look for child nodes with names matching the lookup key (optionally followed by digits to support ZooKeeper's sequential node creation).
+            # If none are found, or if resolution type is array or hash, the node will be parsed as YAML and checked for a key equal to the lookup key.
+
+            def initialize
+                require 'rubygems'
+                require 'yaml'
+                require 'zookeeper'
+
+                Hiera.debug("Hiera ZooKeeper backend starting")
+            end
+
+            def lookup(key, scope, order_override, resolution_type)
+                answer = nil
+                Hiera.debug("Looking up #{key} in ZooKeeper backend")
+                server = Backend.parse_string(Config[:zookeeper][:server], scope) || 'localhost:2181'
+                zk = Zookeeper.new(server)
+
+                Backend.datasources(scope, order_override) do |source|
+                    Hiera.debug("Looking for data source #{source}")
+
+                    datadir = "#{Backend.datadir("zookeeper", scope)}/#{source}"
+                    children = zk.get_children(:path => datadir)[:children] || []
+                    children = children.select { |child| child =~ /\A#{Regexp.escape(key)}\d*\z/ }.sort
+                    if children.length > 0
+                        children.each do |child|
+                            new_answer = YAML.load(zk.get(:path => "#{datadir}/#{child}")[:data])
+                            next if new_answer.nil?
+                            answer = Backend.empty_answer(resolution_type) if answer.nil?
+                            case resolution_type
+                            when :array
+                                answer << new_answer
+                            when :hash
+                                answer = new_answer.merge answer
+                            else
+                                return new_answer
+                            end
+                        end
+                    end
+                    data = zk.get(:path => datadir)[:data]
+                    data = YAML.load(data) if data
+                    next if (data.nil? || !data.is_a?(Hash) || !data.include?(key))
+                    new_answer = data[key]
+                    answer = Backend.empty_answer(resolution_type) if answer.nil?
+                    case resolution_type
+                    when :array
+                        answer << new_answer
+                    when :hash
+                        answer = new_answer.merge answer
+                    else
+                        return new_answer
+                    end
+                end
+                return answer
+
+                rescue ZookeeperExceptions::ZookeeperException
+                    return Backend.empty_answer(resolution_type)
+            end
+        end
+    end
+end

metadata

@@ -0,0 +1,72 @@
+--- !ruby/object:Gem::Specification 
+name: hiera-zookeeper
+version: !ruby/object:Gem::Version 
+  version: 0.1.0
+platform: ruby
+authors: 
+- Some Dude
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2012-04-11 00:00:00 -05:00
+default_executable: 
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: hiera
+  type: :runtime
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: "0"
+    version: 
+- !ruby/object:Gem::Dependency 
+  name: zookeeper
+  type: :runtime
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: "0"
+    version: 
+description: 
+email: 
+executables: []
+
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- lib/hiera/backend/zookeeper_backend.rb
+has_rdoc: false
+homepage: 
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.3.1
+signing_key: 
+specification_version: 2
+summary: A ZooKeeper backend for Hiera.
+test_files: []
+