mongo 2.0.0 → 2.0.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: f1b1c17b52cc160be41aa083e33789a71d99b6b8
-  data.tar.gz: ba505d3b59d858a210a570670a98c6baded60489
+  metadata.gz: d95f4cbddad1184a95d53cb257befaceb5b9d1b0
+  data.tar.gz: 2639f1045f2be8059283b80bcd4e7a867b1ee35c
 SHA512:
-  metadata.gz: 9e770b0aee96c3226681ae96fb0de826fbb4c10ddb8851bd4d9ce763b0cd6d8c1067ec50f42748d84ac31e1ca009e641d2f7f66e780166a8cd4adc33e9db6183
-  data.tar.gz: 5a34d970f35bcd33d06dbfc61ae533cd05a5f585bb242d9e591805d3545902d2cfd74071587bbd47f997318ad9012012c3ad4d951ca0f79ce9c5488324d890cf
+  metadata.gz: 806e70e8f37683ba17c3416a9eb4c6d662305c4b8138f62b9af82c234b3cd98c10d1cd3119040b2b5751208e2058cd57532b4dfd63acee6628da692e8af55bad
+  data.tar.gz: e5a1c81c7930634054d5ace16c452ef4ba0c313a1fc779c2cc3deca75910b7014cae202b73cdd659d50f47d2e66f549fbd1e6017651815fec257b09db6ff4bd6

data/lib/mongo/operation/result.rb

@@ -119,7 +119,7 @@ module Mongo
         documents.each(&block)
       end
 
-      # Initialize a new result result.
+      # Initialize a new result.
       #
       # @example Instantiate the result.
       #   Result.new(replies)

data/lib/mongo/operation/write.rb

@@ -12,6 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+require 'mongo/operation/write/idable'
 require 'mongo/operation/write/bulk'
 require 'mongo/operation/write/delete'
 require 'mongo/operation/write/insert'

data/lib/mongo/operation/write/bulk/bulk_insert.rb

@@ -47,6 +47,7 @@ module Mongo
       # @since 2.0.0
       class BulkInsert
         include Specifiable
+        include Idable
 
         # Execute the bulk insert operation.
         #
@@ -69,21 +70,22 @@ module Mongo
         private
 
         def execute_write_command(context)
-          Result.new(Command::Insert.new(spec).execute(context))
+          command_spec = spec.merge(:documents => ensure_ids(documents))
+          Result.new(Command::Insert.new(command_spec).execute(context), @ids)
         end
 
         def execute_message(context)
           replies = []
           messages.map do |m|
             context.with_connection do |connection|
-              result = LegacyResult.new(connection.dispatch([ m, gle ].compact))
+              result = LegacyResult.new(connection.dispatch([ m, gle ].compact), @ids)
               replies << result.reply
               if stop_sending?(result)
-                return LegacyResult.new(replies)
+                return LegacyResult.new(replies, @ids)
               end
             end
           end
-          LegacyResult.new(replies.compact.empty? ? nil : replies)
+          LegacyResult.new(replies.compact.empty? ? nil : replies, @ids)
         end
 
         def stop_sending?(result)
@@ -117,10 +119,10 @@ module Mongo
         def messages
           if ordered? || gle
             documents.collect do |doc|
-              Protocol::Insert.new(db_name, coll_name, [ doc ], options)
+              Protocol::Insert.new(db_name, coll_name, ensure_ids([ doc ]), options)
             end
           else
-            [ Protocol::Insert.new(db_name, coll_name, documents, { :flags => [:continue_on_error] }) ]
+            [ Protocol::Insert.new(db_name, coll_name, ensure_ids(documents), { :flags => [:continue_on_error] }) ]
           end
         end
       end

data/lib/mongo/operation/write/bulk/bulk_insert/result.rb

@@ -26,6 +26,25 @@ module Mongo
         class Result < Operation::Result
           include BulkMergable
 
+          # Get the ids of the inserted documents.
+          #
+          # @since 2.0.0
+          attr_reader :inserted_ids
+
+          # Initialize a new result.
+          #
+          # @example Instantiate the result.
+          #   Result.new(replies, inserted_ids)
+          #
+          # @param [ Protocol::Reply ] replies The wire protocol replies.
+          # @params [ Array<Object> ] ids The ids of the inserted documents.
+          #
+          # @since 2.0.0
+          def initialize(replies, ids)
+            @replies = replies.is_a?(Protocol::Reply) ? [ replies ] : replies
+            @inserted_ids = ids
+          end
+
           # Gets the number of documents inserted.
           #
           # @example Get the number of documents inserted.
@@ -37,6 +56,18 @@ module Mongo
           def n_inserted
             written_count
           end
+
+          # Gets the id of the document inserted.
+          #
+          # @example Get id of the document inserted.
+          #   result.inserted_id
+          #
+          # @return [ Object ] The id of the document inserted.
+          #
+          # @since 2.0.0
+          def inserted_id
+            inserted_ids.first
+          end
         end
 
         # Defines custom behaviour of results when inserting.
@@ -46,6 +77,25 @@ module Mongo
         class LegacyResult < Operation::Result
           include LegacyBulkMergable
 
+          # Get the ids of the inserted documents.
+          #
+          # @since 2.0.0
+          attr_reader :inserted_ids
+
+          # Initialize a new result.
+          #
+          # @example Instantiate the result.
+          #   Result.new(replies, inserted_ids)
+          #
+          # @param [ Protocol::Reply ] replies The wire protocol replies.
+          # @params [ Array<Object> ] ids The ids of the inserted documents.
+          #
+          # @since 2.0.0
+          def initialize(replies, ids)
+            @replies = replies.is_a?(Protocol::Reply) ? [ replies ] : replies
+            @inserted_ids = ids
+          end
+
           # Gets the number of documents inserted.
           #
           # @example Get the number of documents inserted.
@@ -61,6 +111,18 @@ module Mongo
               n
             end
           end
+
+          # Gets the id of the document inserted.
+          #
+          # @example Get id of the document inserted.
+          #   result.inserted_id
+          #
+          # @return [ Object ] The id of the document inserted.
+          #
+          # @since 2.0.0
+          def inserted_id
+            inserted_ids.first
+          end
         end
       end
     end

data/lib/mongo/operation/write/idable.rb

@@ -0,0 +1,41 @@
+# Copyright (C) 2014-2015 MongoDB, Inc.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+module Mongo
+  module Operation
+    module Write
+      module Idable
+
+        private
+
+        def id(doc)
+          doc.respond_to?(:id) ? doc.id : (doc['_id'] || doc[:_id])
+        end
+
+        def has_id?(doc)
+          id(doc)
+        end
+
+        def ensure_ids(documents)
+          @ids ||= []
+          documents.collect do |doc|
+            doc_with_id = has_id?(doc) ? doc : doc.merge(_id: BSON::ObjectId.new)
+            @ids << id(doc_with_id)
+            doc_with_id
+          end
+        end
+      end
+    end
+  end
+end

data/lib/mongo/operation/write/insert.rb

@@ -44,6 +44,7 @@ module Mongo
       class Insert
         include Executable
         include Specifiable
+        include Idable
 
         # Execute the insert operation.
         #
@@ -66,12 +67,13 @@ module Mongo
         private
 
         def execute_write_command(context)
-          Result.new(Command::Insert.new(spec).execute(context)).validate!
+          command_spec = spec.merge(:documents => ensure_ids(documents))
+          Result.new(Command::Insert.new(command_spec).execute(context), @ids).validate!
         end
 
         def execute_message(context)
           context.with_connection do |connection|
-            Result.new(connection.dispatch([ message, gle ].compact)).validate!
+            Result.new(connection.dispatch([ message, gle ].compact), @ids).validate!
           end
         end
 
@@ -82,7 +84,7 @@ module Mongo
 
         def message
           opts = !!options[:continue_on_error] ? { :flags => [:continue_on_error] } : {}
-          Protocol::Insert.new(db_name, coll_name, documents, opts)
+          Protocol::Insert.new(db_name, coll_name, ensure_ids(documents), opts)
         end
       end
     end

data/lib/mongo/operation/write/insert/result.rb

@@ -25,6 +25,36 @@ module Mongo
         # @since 2.0.0
         class Result < Operation::Result
 
+          # Get the ids of the inserted documents.
+          #
+          # @since 2.0.0
+          attr_reader :inserted_ids
+
+          # Initialize a new result.
+          #
+          # @example Instantiate the result.
+          #   Result.new(replies, inserted_ids)
+          #
+          # @param [ Protocol::Reply ] replies The wire protocol replies.
+          # @params [ Array<Object> ] ids The ids of the inserted documents.
+          #
+          # @since 2.0.0
+          def initialize(replies, ids)
+            @replies = replies.is_a?(Protocol::Reply) ? [ replies ] : replies
+            @inserted_ids = ids
+          end
+
+          # Gets the id of the document inserted.
+          #
+          # @example Get id of the document inserted.
+          #   result.inserted_id
+          #
+          # @return [ Object ] The id of the document inserted.
+          #
+          # @since 2.0.0
+          def inserted_id
+            inserted_ids.first
+          end
         end
       end
     end

data/lib/mongo/version.rb

@@ -17,5 +17,5 @@ module Mongo
   # The current version of the driver.
   #
   # @since 2.0.0
-  VERSION = '2.0.0'.freeze
+  VERSION = '2.0.1'.freeze
 end

data/spec/mongo/collection_spec.rb

@@ -294,6 +294,10 @@ describe Mongo::Collection do
     it 'inserts the documents into the collection', unless: write_command_enabled? do
       expect(result.written_count).to eq(0)
     end
+
+    it 'contains the ids in the result' do
+      expect(result.inserted_ids.size).to eq(2)
+    end
   end
 
   describe '#insert_one' do
@@ -313,6 +317,10 @@ describe Mongo::Collection do
     it 'inserts the document into the collection', unless: write_command_enabled? do
       expect(result.written_count).to eq(0)
     end
+
+    it 'contains the id in the result' do
+      expect(result.inserted_id).to_not be_nil
+    end
   end
 
   describe '#inspect' do

data/spec/mongo/operation/write/{bulk_insert_spec.rb → bulk/bulk_insert_spec.rb}

@@ -19,6 +19,10 @@ describe Mongo::Operation::Write::BulkInsert do
     described_class.new(spec)
   end
 
+  after do
+    authorized_collection.find.delete_many
+  end
+
   describe '#initialize' do
 
     context 'spec' do
@@ -81,6 +85,29 @@ describe Mongo::Operation::Write::BulkInsert do
     end
   end
 
+  describe 'document ids' do
+
+    context 'when documents do not contain an id' do
+
+      let(:documents) do
+        [{ 'field' => 'test' },
+         { 'field' => 'test' }]
+      end
+
+      let(:inserted_ids) do
+        op.execute(authorized_primary.context).inserted_ids
+      end
+
+      let(:collection_ids) do
+        authorized_collection.find(field: 'test').collect { |d| d['_id'] }
+      end
+
+      it 'adds an id to the documents' do
+        expect(inserted_ids).to eq(collection_ids)
+      end
+    end
+  end
+
   describe '#execute' do
 
     before do

data/spec/mongo/operation/write/command/insert_spec.rb

@@ -3,7 +3,7 @@ require 'spec_helper'
 describe Mongo::Operation::Write::Command::Insert do
   include_context 'operation'
 
-  let(:documents) { [{ :foo => 1 }] }
+  let(:documents) { [{ :_id => 1, :foo => 1 }] }
   let(:spec) do
     { :documents     => documents,
       :db_name       => db_name,

data/spec/mongo/operation/write/insert_spec.rb

@@ -15,6 +15,10 @@ describe Mongo::Operation::Write::Insert do
     }
   end
 
+  after do
+    authorized_collection.find.delete_many
+  end
+
   let(:insert) do
     described_class.new(spec)
   end
@@ -80,6 +84,29 @@ describe Mongo::Operation::Write::Insert do
     end
   end
 
+  describe 'document ids' do
+
+    context 'when documents do not contain an id' do
+
+      let(:documents) do
+        [{ 'field' => 'test' },
+         { 'field' => 'test' }]
+      end
+
+      let(:inserted_ids) do
+        insert.execute(authorized_primary.context).inserted_ids
+      end
+
+      let(:collection_ids) do
+        authorized_collection.find(field: 'test').collect { |d| d['_id'] }
+      end
+
+      it 'adds an id to the documents' do
+        expect(inserted_ids).to eq(collection_ids)
+      end
+    end
+  end
+
   describe '#execute' do
 
     before do

data/spec/support/crud/write.rb

@@ -121,15 +121,13 @@ module Mongo
         end
 
         def insert_many(collection)
-          collection.insert_many(documents)
-          # returning inserted_ids is optional
-          { 'insertedIds' => documents.collect { |d| d['_id'] } }
+          result = collection.insert_many(documents)
+          { 'insertedIds' => result.inserted_ids }
         end
 
         def insert_one(collection)
           result = collection.insert_one(document)
-          # returning inserted_id is optional
-          { 'insertedId' => document['_id'] }
+          { 'insertedId' => result.inserted_id }
         end
 
         def update_return_doc(result)

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: mongo
 version: !ruby/object:Gem::Version
-  version: 2.0.0
+  version: 2.0.1
 platform: ruby
 authors:
 - Tyler Brock
@@ -32,7 +32,7 @@ cert_chain:
   o2UXDbWtz5PqoFd8EgNJAn3+BG1pwC9S9pVFG3WPucfAx/bE8iq/vvchHei5Y/Vo
   aAz5f/hY4zFeYWvGDBHYEXE1rTN2hhMSyJscPcFbmz0=
   -----END CERTIFICATE-----
-date: 2015-03-25 00:00:00.000000000 Z
+date: 2015-03-26 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bson
@@ -186,6 +186,7 @@ files:
 - lib/mongo/operation/write/delete.rb
 - lib/mongo/operation/write/delete/result.rb
 - lib/mongo/operation/write/drop_index.rb
+- lib/mongo/operation/write/idable.rb
 - lib/mongo/operation/write/insert.rb
 - lib/mongo/operation/write/insert/result.rb
 - lib/mongo/operation/write/remove_user.rb
@@ -293,9 +294,9 @@ files:
 - spec/mongo/operation/read/query_spec.rb
 - spec/mongo/operation/result_spec.rb
 - spec/mongo/operation/specifiable_spec.rb
-- spec/mongo/operation/write/bulk_delete_spec.rb
-- spec/mongo/operation/write/bulk_insert_spec.rb
-- spec/mongo/operation/write/bulk_update_spec.rb
+- spec/mongo/operation/write/bulk/bulk_delete_spec.rb
+- spec/mongo/operation/write/bulk/bulk_insert_spec.rb
+- spec/mongo/operation/write/bulk/bulk_update_spec.rb
 - spec/mongo/operation/write/command/delete_spec.rb
 - spec/mongo/operation/write/command/insert_spec.rb
 - spec/mongo/operation/write/command/update_spec.rb
@@ -514,9 +515,9 @@ test_files:
 - spec/mongo/operation/read/query_spec.rb
 - spec/mongo/operation/result_spec.rb
 - spec/mongo/operation/specifiable_spec.rb
-- spec/mongo/operation/write/bulk_delete_spec.rb
-- spec/mongo/operation/write/bulk_insert_spec.rb
-- spec/mongo/operation/write/bulk_update_spec.rb
+- spec/mongo/operation/write/bulk/bulk_delete_spec.rb
+- spec/mongo/operation/write/bulk/bulk_insert_spec.rb
+- spec/mongo/operation/write/bulk/bulk_update_spec.rb
 - spec/mongo/operation/write/command/delete_spec.rb
 - spec/mongo/operation/write/command/insert_spec.rb
 - spec/mongo/operation/write/command/update_spec.rb