2008-12-17 16:49:06 +00:00
|
|
|
# --
|
2009-01-06 15:51:01 +00:00
|
|
|
# Copyright (C) 2008-2009 10gen Inc.
|
2008-11-22 01:00:51 +00:00
|
|
|
#
|
|
|
|
# This program is free software: you can redistribute it and/or modify it
|
|
|
|
# under the terms of the GNU Affero General Public License, version 3, as
|
|
|
|
# published by the Free Software Foundation.
|
|
|
|
#
|
|
|
|
# This program is distributed in the hope that it will be useful, but WITHOUT
|
|
|
|
# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
|
|
|
|
# FITNESS FOR A PARTICULAR PURPOSE. See the GNU Affero General Public License
|
|
|
|
# for more details.
|
|
|
|
#
|
|
|
|
# You should have received a copy of the GNU Affero General Public License
|
|
|
|
# along with this program. If not, see <http://www.gnu.org/licenses/>.
|
2008-12-17 16:49:06 +00:00
|
|
|
# ++
|
2008-11-22 01:00:51 +00:00
|
|
|
|
|
|
|
require 'mongo/query'
|
|
|
|
|
|
|
|
module XGen
|
|
|
|
module Mongo
|
|
|
|
module Driver
|
2008-12-17 16:43:08 +00:00
|
|
|
|
|
|
|
# A named collection of records in a database.
|
2008-11-22 01:00:51 +00:00
|
|
|
class Collection
|
|
|
|
|
2009-01-13 20:51:41 +00:00
|
|
|
attr_reader :db, :name, :hint_fields
|
2009-01-13 20:38:46 +00:00
|
|
|
|
2008-11-22 01:00:51 +00:00
|
|
|
def initialize(db, name)
|
|
|
|
@db = db
|
|
|
|
@name = name
|
|
|
|
end
|
|
|
|
|
2009-01-13 20:51:41 +00:00
|
|
|
# Set hint fields to use and return +self+. hint_fields may be a
|
2009-01-14 15:26:12 +00:00
|
|
|
# single field name, array of field names, or a hash whose keys will
|
|
|
|
# become the hint field names. May be +nil+.
|
2009-01-13 20:51:41 +00:00
|
|
|
def hint(hint_fields)
|
2009-01-14 15:26:12 +00:00
|
|
|
@hint_fields = case hint_fields
|
|
|
|
when String
|
|
|
|
[hint_fields]
|
|
|
|
when Hash
|
|
|
|
hint_fields.keys
|
|
|
|
when nil
|
|
|
|
nil
|
|
|
|
else
|
|
|
|
hint_fields.to_a
|
|
|
|
end
|
2009-01-13 20:51:41 +00:00
|
|
|
self
|
|
|
|
end
|
|
|
|
|
2008-12-17 16:43:08 +00:00
|
|
|
# Return records that match a +selector+ hash. See Mongo docs for
|
|
|
|
# details.
|
|
|
|
#
|
2008-12-08 21:47:52 +00:00
|
|
|
# Options:
|
2008-12-17 16:43:08 +00:00
|
|
|
# :fields :: Array of collection field names; only those will be returned (plus _id if defined)
|
|
|
|
# :offset :: Start at this record when returning records
|
|
|
|
# :limit :: Maximum number of records to return
|
|
|
|
# :sort :: Either hash of field names as keys and 1/-1 as values; 1 ==
|
|
|
|
# ascending, -1 == descending, or array of field names (all
|
|
|
|
# assumed to be sorted in ascending order).
|
2008-12-08 21:47:52 +00:00
|
|
|
def find(selector={}, options={})
|
|
|
|
fields = options.delete(:fields)
|
2008-11-22 01:00:51 +00:00
|
|
|
fields = nil if fields && fields.empty?
|
2009-01-07 00:14:45 +00:00
|
|
|
offset = options.delete(:offset) || 0
|
|
|
|
limit = options.delete(:limit) || 0
|
|
|
|
sort = options.delete(:sort)
|
|
|
|
raise RuntimeError, "Unknown options [#{options.inspect}]" unless options.empty?
|
2009-01-13 20:38:46 +00:00
|
|
|
@db.query(self, Query.new(selector, fields, offset, limit, sort))
|
2008-11-22 01:00:51 +00:00
|
|
|
end
|
|
|
|
|
2009-01-07 14:45:56 +00:00
|
|
|
# Insert +objects+, which are hashes. "<<" is aliased to this method.
|
2008-11-22 01:00:51 +00:00
|
|
|
def insert(*objects)
|
2008-12-05 00:10:00 +00:00
|
|
|
objects = objects.first if objects.size == 1 && objects.first.is_a?(Array)
|
2008-12-08 21:41:52 +00:00
|
|
|
res = @db.insert_into_db(@name, objects)
|
|
|
|
res.size > 1 ? res : res.first
|
2008-11-22 01:00:51 +00:00
|
|
|
end
|
2008-12-08 20:04:07 +00:00
|
|
|
alias_method :<<, :insert
|
2008-11-22 01:00:51 +00:00
|
|
|
|
2008-12-17 16:43:08 +00:00
|
|
|
# Remove the records that match +selector+.
|
2008-11-22 01:00:51 +00:00
|
|
|
def remove(selector={})
|
|
|
|
@db.remove_from_db(@name, selector)
|
|
|
|
end
|
|
|
|
|
2008-12-17 16:43:08 +00:00
|
|
|
# Remove all records.
|
2008-11-22 01:00:51 +00:00
|
|
|
def clear
|
|
|
|
remove({})
|
|
|
|
end
|
|
|
|
|
2008-12-17 16:43:08 +00:00
|
|
|
# Update records that match +selector+ by applying +obj+ as an update.
|
|
|
|
# If no match, inserts (???).
|
2008-11-22 01:00:51 +00:00
|
|
|
def repsert(selector, obj)
|
|
|
|
@db.repsert_in_db(@name, selector, obj)
|
|
|
|
end
|
|
|
|
|
2008-12-17 16:43:08 +00:00
|
|
|
# Update records that match +selector+ by applying +obj+ as an update.
|
2008-11-22 01:00:51 +00:00
|
|
|
def replace(selector, obj)
|
|
|
|
@db.replace_in_db(@name, selector, obj)
|
|
|
|
end
|
|
|
|
|
2008-12-17 16:43:08 +00:00
|
|
|
# Update records that match +selector+ by applying +obj+ as an update.
|
|
|
|
# Both +selector+ and +modifier_obj+ are required.
|
|
|
|
def modify(selector, modifier_obj)
|
|
|
|
raise "no object" unless modifier_obj
|
2008-11-22 01:00:51 +00:00
|
|
|
raise "no selector" unless selector
|
2008-12-17 16:43:08 +00:00
|
|
|
@db.modify_in_db(@name, selector, modifier_obj)
|
2008-11-22 01:00:51 +00:00
|
|
|
end
|
|
|
|
|
2008-12-17 16:43:08 +00:00
|
|
|
# Create a new index named +index_name+. +fields+ should be an array
|
|
|
|
# of field names.
|
2008-11-22 01:00:51 +00:00
|
|
|
def create_index(name, *fields)
|
|
|
|
@db.create_index(@name, name, fields)
|
|
|
|
end
|
|
|
|
|
2008-12-17 16:43:08 +00:00
|
|
|
# Drop index +name+.
|
2008-11-22 01:00:51 +00:00
|
|
|
def drop_index(name)
|
|
|
|
@db.drop_index(@name, name)
|
|
|
|
end
|
|
|
|
|
2008-12-17 16:43:08 +00:00
|
|
|
# Drop all indexes.
|
2008-11-22 01:00:51 +00:00
|
|
|
def drop_indexes
|
2008-12-08 16:38:42 +00:00
|
|
|
# just need to call drop indexes with no args; will drop them all
|
|
|
|
@db.drop_index(@name, '*')
|
2008-11-22 01:00:51 +00:00
|
|
|
end
|
|
|
|
|
2009-01-15 16:25:23 +00:00
|
|
|
# Drop the entire collection. USE WITH CAUTION.
|
|
|
|
def drop
|
|
|
|
@db.drop_collection(@name)
|
|
|
|
end
|
|
|
|
|
2008-12-17 16:43:08 +00:00
|
|
|
# Return an array of hashes, one for each index. Each hash contains:
|
|
|
|
#
|
|
|
|
# :name :: Index name
|
|
|
|
#
|
|
|
|
# :keys :: Hash whose keys are the names of the fields that make up
|
|
|
|
# the key and values are integers.
|
|
|
|
#
|
|
|
|
# :ns :: Namespace; same as this collection's name.
|
2008-11-22 01:00:51 +00:00
|
|
|
def index_information
|
|
|
|
@db.index_information(@name)
|
|
|
|
end
|
|
|
|
|
2008-12-17 18:52:10 +00:00
|
|
|
# Return a hash containing options that apply to this collection.
|
|
|
|
# 'create' will be the collection name. For the other possible keys
|
|
|
|
# and values, see DB#create_collection.
|
|
|
|
def options
|
|
|
|
@db.collections_info(@name).next_object()['options']
|
|
|
|
end
|
|
|
|
|
2008-12-17 16:43:08 +00:00
|
|
|
# Return the number of records that match +selector+. If +selector+ is
|
|
|
|
# +nil+ or an empty hash, returns the count of all records.
|
2008-11-22 01:00:51 +00:00
|
|
|
def count(selector={})
|
|
|
|
@db.count(@name, selector || {})
|
|
|
|
end
|
|
|
|
|
|
|
|
end
|
|
|
|
end
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|