Contents

require "uri"
require "mongo"

module MiniMongo
  module Mapper
    def self.included(base)
      base.send :extend, ClassMethods
    end

    module ClassMethods
      def find(attrs={})
        attrs["_id"] = BSON::ObjectId(attrs["id"]) if attrs["id"]
        attrs.delete("id")
        docs = self.collection.find(attrs).to_a
        if docs.empty?
          raise DocumentNotFound, "Couldn't find #{self.collection_name.capitalize}, with #{attrs.to_a.collect {|p| p.join(' = ')}.join(', ')}"
        else
          result = []
          docs.each do |doc|
            result << self.class_eval("new(#{doc})") if doc
          end
          result
        end
      end

      def insert(attrs={})
        doc = {}
        doc["_id"] = self.collection.insert(attrs).to_s
        doc.merge!(attrs)
        self.new(doc)
      end

      def update(id, attrs={})
        self.collection.update({"_id" => BSON::ObjectId(id)}, attrs)
      end

      def remove(id)
        self.collection.remove("_id" => BSON::ObjectId(id))
      end

      def count
        self.collection.count
      end

      def remove_all
        self.collection.remove
      end

      def maps(name)
        self.collection_name = name.to_s
        self.collection = MiniMongo.db_connection.collection(name.to_s)
      end

      def collection
        self.class_variable_get(:@@collection)
      end

      def collection=(collection)
        self.class_variable_set(:@@collection, collection)
      end

      def collection_name
        self.class_variable_get(:@@collection_name)
      end

      def collection_name=(collection_name)
        self.class_variable_set(:@@collection_name, collection_name)
      end

    end
  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
mini_mongo-0.0.4	lib/mini_mongo/mapper.rb