langrove-plugin-mongodb 0.1

data/README.md

@@ -0,0 +1,61 @@
+`Plugin::MongoDB`
+=================
+
+Intergates MongoDB support into LanGrove
+
+`gem install langrove-plugin-mongodb`
+
+Implements 
+----------
+
+* `LanGrove::Plugin::Persistor`
+
+Supports
+---------
+
+* `LanGrove::Behaviour::Persistable`
+* `LanGrove::Behaviour::Preloadable`
+
+Example
+-------
+
+### Configuring the plugin.
+
+<pre>
+
+	...
+
+    :plugins:
+        :a_mongodb_collection:
+        	:class: MongoDB
+            :database: database_name
+            :table: collection_name
+            :key:
+            	_id: unique_field_name
+            	#
+            	# assuming @capsule['unique_field_name'] exists
+            	#
+
+    ...
+
+</pre>
+
+<pre>
+
+	...
+
+    :server:
+        :behavior:
+            :persistable:
+                :plugin: :a_mongodb_collection
+                :at:
+                    :handler_after_receive: 
+                    #
+                    # will default to storing the caspsule 
+                    # after the Handler.receive( data )
+                    # method returns.
+                    #
+
+    ...
+
+</pre>

data/lib/langrove-plugin-mongodb.rb

@@ -0,0 +1,2 @@
+require 'langrove' unless defined?( LanGrove )
+require 'plugin/mongo_db'

data/lib/plugin/mongo_db.rb

@@ -0,0 +1,125 @@
+require 'mongo'
+
+module Plugin
+  
+  class MongoDB < LanGrove::Plugin::Persistor
+    
+    def validate_config
+      
+      super
+      
+      connect
+      
+      #
+      # Polulate this from :db:username: and :db:password
+      #
+      # auth = db.authenticate(username, password)
+      
+    end
+    
+    def connect
+      
+      @logger.info "Connecting to mongo #{@database}.#{@table}"
+      @connection = Mongo::Connection.new
+      @db = @connection.db( @database )
+      @collection = @db.collection( @table )
+      
+    end
+    
+    
+    def all_capsules( &block )
+      
+      #
+      # raise Exception on failed (should bring down the daemon)
+      # 
+      @collection.find().to_a.each do |capsule|
+      
+        yield capsule['_id'], capsule
+      
+      end
+      
+    end
+    
+    
+    def store( handler )
+      
+      gen_key( handler ) if handler.key.nil?
+    
+      begin
+        
+        id = handler.key['key']['_id']
+        
+        @collection.update(
+        
+          {'_id' => id }, 
+          handler.capsule, 
+          :upsert  => true
+          
+        )
+        
+        @logger.debug "Wrote #{handler.class} to mongo:#{@database}.#{@table} at _id: #{id}"
+      
+      rescue Mongo::ConnectionFailure => e
+      
+        @logger.error "ERROR: on fetch() #{e}"
+      
+        connect
+        
+        retry
+        
+      end
+      
+      true
+      
+    end
+    
+    
+    def fetch( handler )
+      
+      gen_key( handler ) if handler.key.nil?
+      
+      id = handler.key['key']['_id']
+      
+      begin 
+      
+        capsule = @collection.find_one('_id' => id )
+        
+        if capsule.nil?
+          
+          unless @allow_learn.nil?
+
+            handler.capsule = {}
+            
+            @logger.warn( "WARN: Assumed NEW capsule for #{@table}._id: #{id}" )
+
+            return true
+            
+          else
+            
+            @logger.error "ERROR: Ignored #{@table}._id: #{id}. Enable :allow_learn:"
+            
+            return false
+            
+          end
+
+        end
+      
+      rescue Mongo::ConnectionFailure => e
+      
+        @logger.error "ERROR: on store() #{e}"
+      
+        connect
+        
+        retry
+        
+      end
+      
+      handler.capsule = capsule
+      
+      return true
+      
+    end
+    
+  end
+  
+end

data/spec/plugin/mongo_db_spec.rb

@@ -0,0 +1,94 @@
+require 'langrove'
+require 'plugin/mongo_db'
+
+ROOT = LanGrove::FakeRoot.new
+
+MONGO_PERSISTOR_CONFIG = { 
+  
+  :database => 'test',
+  
+  :table => 'collection',
+  
+  :class => 'MongoDB', 
+  
+  :key => { 
+
+    '_id' => 'hostname'
+    
+  } 
+  
+}
+
+describe Plugin::MongoDB do
+    
+    before :each do 
+      @hostname = 'woot@server.awesome'
+      class FakeHandler
+         
+       attr_accessor :key, :capsule
+       @key = {}
+       @capsule = {} 
+        
+      end
+    end
+    
+  subject {
+    
+    #
+    # Persist this handler
+    #
+    
+    @handler = LanGrove::Handler::Base.new( nil )
+    
+    @handler.instance_variable_set( 
+      
+      :@capsule, {
+        'thing_a' => 'DATA',
+        'thing_b' => 'DATATA',
+        'hostname' => @hostname,
+        'more_data' => {
+          'date' => Time.now
+        }
+      }
+    )
+
+    LanGrove::ClassLoader.create( { 
+    
+      :module => 'Plugin',
+      :class  => 'MongoDB'
+      
+    } ).new( ROOT, MONGO_PERSISTOR_CONFIG, nil )
+    
+  }   
+
+  it 'can store()' do
+    
+    subject.store( @handler )
+    
+  end
+  
+  it 'can fetch()' do
+    
+    handler = FakeHandler.new
+    handler.key = {'key' => {'_id' => @hostname}}
+    subject.fetch( handler )
+    
+    handler.capsule['thing_a'].should == 'DATA'
+   
+  end
+  
+  it 'can fetch_all()' do
+    
+    result = nil
+    
+    subject.all_capsules() do |key,capsule|
+   
+      result = capsule
+   
+    end
+
+    result['hostname'].should == @hostname
+    
+  end
+  
+end

metadata

@@ -0,0 +1,99 @@
+--- !ruby/object:Gem::Specification
+name: langrove-plugin-mongodb
+version: !ruby/object:Gem::Version
+  version: '0.1'
+  prerelease: 
+platform: ruby
+authors:
+- Richard
+- ''
+- ''
+- ''
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2012-07-08 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: langrove
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: mongo
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: bson_ext
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+description: ''
+email: ops@clue.co.za
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/langrove-plugin-mongodb.rb
+- lib/plugin/mongo_db.rb
+- spec/plugin/mongo_db_spec.rb
+- README.md
+homepage: ''
+licenses: []
+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.24
+signing_key: 
+specification_version: 3
+summary: Intergates MongoDB support into LanGrove persistance layer
+test_files: []