Allow idiomatic :drop_dups in addition to :dropDups
on index creation. Don't raise exception if :dropDups results in duplicate key error.
This commit is contained in:
parent
004ac48230
commit
5aa8721b25
|
@ -358,10 +358,10 @@ module Mongo
|
||||||
# @option opts [Boolean] :unique (false) if true, this index will enforce a uniqueness constraint.
|
# @option opts [Boolean] :unique (false) if true, this index will enforce a uniqueness constraint.
|
||||||
# @option opts [Boolean] :background (false) indicate that the index should be built in the background. This
|
# @option opts [Boolean] :background (false) indicate that the index should be built in the background. This
|
||||||
# feature is only available in MongoDB >= 1.3.2.
|
# feature is only available in MongoDB >= 1.3.2.
|
||||||
# @option opts [Boolean] :dropDups If creating a unique index on a collection with pre-existing records,
|
# @option opts [Boolean] :drop_dups (nil) If creating a unique index on a collection with pre-existing records,
|
||||||
# this option will keep the first document the database indexes and drop all subsequent with duplicate values.
|
# this option will keep the first document the database indexes and drop all subsequent with duplicate values.
|
||||||
# @option opts [Integer] :min specify the minimum longitude and latitude for a geo index.
|
# @option opts [Integer] :min (nil) specify the minimum longitude and latitude for a geo index.
|
||||||
# @option opts [Integer] :max specify the maximum longitude and latitude for a geo index.
|
# @option opts [Integer] :max (nil) specify the maximum longitude and latitude for a geo index.
|
||||||
#
|
#
|
||||||
# @example Creating a compound index:
|
# @example Creating a compound index:
|
||||||
# @posts.create_index([['subject', Mongo::ASCENDING], ['created_at', Mongo::DESCENDING]])
|
# @posts.create_index([['subject', Mongo::ASCENDING], ['created_at', Mongo::DESCENDING]])
|
||||||
|
@ -381,7 +381,7 @@ module Mongo
|
||||||
#
|
#
|
||||||
# @core indexes create_index-instance_method
|
# @core indexes create_index-instance_method
|
||||||
def create_index(spec, opts={})
|
def create_index(spec, opts={})
|
||||||
opts.assert_valid_keys(:min, :max, :name, :background, :unique, :dropDups)
|
opts[:dropDups] = opts.delete(:drop_dups) if opts[:drop_dups]
|
||||||
field_spec = BSON::OrderedHash.new
|
field_spec = BSON::OrderedHash.new
|
||||||
if spec.is_a?(String) || spec.is_a?(Symbol)
|
if spec.is_a?(String) || spec.is_a?(Symbol)
|
||||||
field_spec[spec.to_s] = 1
|
field_spec[spec.to_s] = 1
|
||||||
|
@ -407,10 +407,17 @@ module Mongo
|
||||||
:key => field_spec
|
:key => field_spec
|
||||||
}
|
}
|
||||||
selector.merge!(opts)
|
selector.merge!(opts)
|
||||||
begin
|
|
||||||
response = insert_documents([selector], Mongo::DB::SYSTEM_INDEX_COLLECTION, false, true)
|
insert_documents([selector], Mongo::DB::SYSTEM_INDEX_COLLECTION, false, false)
|
||||||
rescue Mongo::OperationFailure
|
response = @db.get_last_error
|
||||||
raise Mongo::OperationFailure, "Failed to create index #{selector.inspect} with the following errors: #{response}"
|
|
||||||
|
if response['err']
|
||||||
|
if response['code'] == 11000 && selector[:dropDups]
|
||||||
|
# NOP. If the user is intentionally dropping dups, we can ignore duplicate key errors.
|
||||||
|
else
|
||||||
|
raise Mongo::OperationFailure, "Failed to create index #{selector.inspect} with the following error: " +
|
||||||
|
"#{response['err']}"
|
||||||
|
end
|
||||||
end
|
end
|
||||||
name
|
name
|
||||||
end
|
end
|
||||||
|
|
|
@ -628,6 +628,22 @@ class TestCollection < Test::Unit::TestCase
|
||||||
assert @collection.index_information['a_1']['unique'] == true
|
assert @collection.index_information['a_1']['unique'] == true
|
||||||
end
|
end
|
||||||
|
|
||||||
|
should "drop duplicates" do
|
||||||
|
@collection.insert({:a => 1})
|
||||||
|
@collection.insert({:a => 1})
|
||||||
|
assert_equal 2, @collection.find({:a => 1}).count
|
||||||
|
@collection.create_index([['a', Mongo::ASCENDING]], :unique => true, :dropDups => true)
|
||||||
|
assert_equal 1, @collection.find({:a => 1}).count
|
||||||
|
end
|
||||||
|
|
||||||
|
should "drop duplicates with ruby-like drop_dups key" do
|
||||||
|
@collection.insert({:a => 1})
|
||||||
|
@collection.insert({:a => 1})
|
||||||
|
assert_equal 2, @collection.find({:a => 1}).count
|
||||||
|
@collection.create_index([['a', Mongo::ASCENDING]], :unique => true, :drop_dups => true)
|
||||||
|
assert_equal 1, @collection.find({:a => 1}).count
|
||||||
|
end
|
||||||
|
|
||||||
should "create an index in the background" do
|
should "create an index in the background" do
|
||||||
if @@version > '1.3.1'
|
if @@version > '1.3.1'
|
||||||
@collection.create_index([['b', Mongo::ASCENDING]], :background => true)
|
@collection.create_index([['b', Mongo::ASCENDING]], :background => true)
|
||||||
|
|
Loading…
Reference in New Issue