hiera_zookeeper 0.2.0

data/lib/hiera/backend/zookeeper_backend.rb

@@ -0,0 +1,61 @@
+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.
+      # The node data is expected to be in YAML.
+
+      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")
+        unless Config[:zookeeper].nil? or Config[:zookeeper][:server].nil? then
+          server = Backend.parse_string(Config[:zookeeper][:server], scope)
+        else
+          server = 'localhost:2181'
+        end 
+        zk = Zookeeper.new(server)
+
+        Backend.datasources(scope, order_override) do |source|
+          Hiera.debug("Looking for data source #{source}")
+          # Determine the zookeeper path(s) to get data from
+          unless Config[:zookeeper].nil? or Config[:zookeeper][:datadir].nil? then
+            datadir = "#{Backend.datadir(Config[:zookeeper][:datadir], scope)}/#{source}"
+          else
+            datadir = "/hiera/#{source}"
+          end
+          data = zk.get(:path => "#{datadir}/#{key}")
+          if data[:stat].exists then
+            Hiera.debug("Found #{key} in #{source}")
+            new_answer = Backend.resolve_answer(YAML.load(data[:data]), resolution_type)
+          end
+          case resolution_type
+          when :array
+            raise Exception, "Hiera type mismatch: expected Array and got #{new_answer.class}" unless new_answer.kind_of?(Array) or new_answer.kind_of?(String)
+            answer ||= []
+            answer << new_answer
+          when :hash
+            raise Exception, "Hiera type mismatch: expected Hash and got #{new_answer.class}" unless new_answer.kind_of?(Hash)
+            answer ||= {}
+            answer = new_answer.merge(answer)
+          else
+            answer = new_answer
+            break
+          end
+        end
+        zk.close if !zk.nil?
+        return answer
+
+        #rescue ZookeeperExceptions::ZookeeperException, RuntimeError
+        #  zk.close if !zk.nil?
+        #  return nil
+      end
+    end
+  end
+end

metadata

@@ -0,0 +1,79 @@
+--- !ruby/object:Gem::Specification
+name: hiera_zookeeper
+version: !ruby/object:Gem::Version
+  version: 0.2.0
+  prerelease: 
+platform: ruby
+authors:
+- zahna
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2012-09-24 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: hiera
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: 1.0.0
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: 1.0.0
+- !ruby/object:Gem::Dependency
+  name: zookeeper
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: 1.3.0
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: 1.3.0
+description: Hiera_zookeeper is a plugin for the Hiera hierarchical database to get
+  data from Zookeeper
+email: scott@zahna.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/hiera/backend/zookeeper_backend.rb
+homepage: http://rubygems.org/gems/hiera_zookeeper
+licenses: []
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: 1.8.7
+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: Hiera_zookeeper is a plugin for the Hiera hierarchical database to get data
+  from Zookeeper
+test_files: []