Contents

# -*- encoding : utf-8 -*-

module Kabutops

  module Adapters

    class Mongo < DatabaseAdapter
      include Extensions::Parameterable

      params :host, :port, :db, :collection, :user, :password

      def store result
        existing = collection.find('id' => result[:id])

        if existing.count > 0
          existing.each do |document|
            collection.update({'_id' => document['_id']}, result.to_hash)
          end
        else
          collection.insert(result.to_hash)
        end
      end

      def nested?
        true
      end

      protected

      def client
        @@client ||= ::Mongo::MongoClient.new(
          params[:host] || 'localhost',
          params[:port] || 27017,
        )
      end

      def client_db
        @@client_db ||= client.db(params[:db].to_s || 'kabutops')
        if params[:user] && params[:password]
          ok = @@client.authenticate(params[:user], params[:password])
          raise 'mongo authentication failed' unless ok
        end
        @@client_db
      end

      def collection
        @@collection ||= client_db.collection(params[:collection] || 'kabutops')
      end
    end

  end

end

Version data entries

15 entries across 15 versions & 1 rubygems

Version	Path
kabutops-0.1.6	lib/kabutops/adapters/mongo.rb
kabutops-0.1.5	lib/kabutops/adapters/mongo.rb
kabutops-0.1.4	lib/kabutops/adapters/mongo.rb
kabutops-0.1.3	lib/kabutops/adapters/mongo.rb
kabutops-0.1.2	lib/kabutops/adapters/mongo.rb
kabutops-0.1.1	lib/kabutops/adapters/mongo.rb
kabutops-0.1.0	lib/kabutops/adapters/mongo.rb
kabutops-0.0.15	lib/kabutops/adapters/mongo.rb
kabutops-0.0.14	lib/kabutops/adapters/mongo.rb
kabutops-0.0.13	lib/kabutops/adapters/mongo.rb
kabutops-0.0.12	lib/kabutops/adapters/mongo.rb
kabutops-0.0.11	lib/kabutops/adapters/mongo.rb
kabutops-0.0.10	lib/kabutops/adapters/mongo.rb
kabutops-0.0.9	lib/kabutops/adapters/mongo.rb
kabutops-0.0.8	lib/kabutops/adapters/mongo.rb