mongo 2.4.1 → 2.14.0

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

@@ -1,71 +0,0 @@
-# Copyright (C) 2014-2016 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.
-
-require 'mongo/operation/write/bulk/delete/result'
-
-module Mongo
-  module Operation
-    module Write
-      module Bulk
-
-        # A MongoDB bulk delete operation.
-        #
-        # @note If a server with version >= 2.5.5 is selected, a write command
-        #   operation will be created and sent instead.
-        #
-        # @example Create the delete operation.
-        #   Write::Bulk::Delete.new({
-        #     :deletes => [{ :q => { :foo => 1 }, :limit => 1 }],
-        #     :db_name => 'test',
-        #     :coll_name => 'test_coll',
-        #     :write_concern => write_concern
-        #   })
-        #
-        # Initialization:
-        #   param [ Hash ] spec The specifications for the delete.
-        #
-        #   option spec :deletes [ Array ] The delete documents.
-        #   option spec :db_name [ String ] The name of the database on which
-        #     the delete should be executed.
-        #   option spec :coll_name [ String ] The name of the collection on which
-        #     the delete should be executed.
-        #   option spec :write_concern [ Mongo::WriteConcern ] The write concern
-        #     for this operation.
-        #   option spec :ordered [ true, false ] Whether the operations should be
-        #     executed in order.
-        #   option spec :options [Hash] Options for the command, if it ends up being a
-        #     write command.
-        #
-        # @since 2.0.0
-        class Delete
-          include Bulkable
-          include Specifiable
-
-          private
-
-          def execute_write_command(server)
-            Result.new(Command::Delete.new(spec).execute(server))
-          end
-
-          def messages
-            deletes.collect do |del|
-              opts = ( del[Operation::LIMIT] || 0 ) <= 0 ? {} : { :flags => [ :single_remove ] }
-              Protocol::Delete.new(db_name, coll_name, del[Operation::Q], opts)
-            end
-          end
-        end
-      end
-    end
-  end
-end

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

@@ -1,129 +0,0 @@
-# Copyright (C) 2014-2016 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 Bulk
-        class Insert
-
-          # Defines custom behaviour of results when inserting.
-          #
-          # @since 2.0.0
-          class Result < Operation::Result
-            include Mergable
-
-            # 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.
-            # @param [ 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.
-            #   result.n_inserted
-            #
-            # @return [ Integer ] The number of documents inserted.
-            #
-            # @since 2.0.0
-            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.
-          # For server versions < 2.5.5 (that don't use write commands).
-          #
-          # @since 2.0.0
-          class LegacyResult < Operation::Result
-            include LegacyMergable
-
-            # 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.
-            # @param [ 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.
-            #   result.n_inserted
-            #
-            # @return [ Integer ] The number of documents inserted.
-            #
-            # @since 2.0.0
-            def n_inserted
-              return 0 unless acknowledged?
-              @replies.reduce(0) do |n, reply|
-                n += 1 unless reply_write_errors?(reply)
-                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
-    end
-  end
-end

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

@@ -1,96 +0,0 @@
-# Copyright (C) 2014-2016 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.
-
-require 'mongo/operation/write/bulk/insert/result'
-
-module Mongo
-  module Operation
-    module Write
-      module Bulk
-
-        # A MongoDB bulk insert operation.
-        # This class should only be used by the Bulk API.
-        #
-        # @note If a server with version >= 2.5.5 is being used, a write command
-        #   operation will be created and sent instead.
-        #
-        # @example Create the new insert operation.
-        #   Write::BulkInsert.new({
-        #     :documents => [{ :foo => 1 }],
-        #     :db_name => 'test',
-        #     :coll_name => 'test_coll',
-        #     :write_concern => write_concern,
-        #     :ordered => false
-        #   })
-        #
-        # Initialization:
-        #   param [ Hash ] spec The specifications for the insert.
-        #
-        #   option spec :documents [ Array ] The documents to insert.
-        #   option spec :db_name [ String ] The name of the database.
-        #   option spec :coll_name [ String ] The name of the collection.
-        #   option spec :write_concern [ Mongo::WriteConcern ] The write concern.
-        #   option spec :ordered [ true, false ] Whether the operations should be
-        #     executed in order.
-        #   option spec :options [ Hash ] Options for the command, if it ends up being a
-        #     write command.
-        #
-        # @since 2.0.0
-        class Insert
-          include Bulkable
-          include Specifiable
-          include Idable
-
-          private
-
-          def execute_write_command(server)
-            command_spec = spec.merge(:documents => ensure_ids(documents))
-            Result.new(Command::Insert.new(command_spec).execute(server), @ids)
-          end
-
-          def execute_message(server)
-            replies = []
-            messages.map do |m|
-              server.with_connection do |connection|
-                result = LegacyResult.new(connection.dispatch([ m, gle ].compact, operation_id), @ids)
-                replies << result.reply
-                if stop_sending?(result)
-                  return LegacyResult.new(replies, @ids)
-                end
-              end
-            end
-            LegacyResult.new(replies.compact.empty? ? nil : replies, @ids)
-          end
-
-          def messages
-            if ordered? || gle
-              documents.collect do |doc|
-                Protocol::Insert.new(db_name, coll_name, ensure_ids([ doc ]), spec)
-              end
-            else
-              [
-                Protocol::Insert.new(
-                  db_name,
-                  coll_name,
-                  ensure_ids(documents),
-                  spec.merge({ :flags => [:continue_on_error] })
-                )
-              ]
-            end
-          end
-        end
-      end
-    end
-  end
-end

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

@@ -1,87 +0,0 @@
-# Copyright (C) 2014-2016 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 Bulk
-
-        # This module contains common functionality for merging results from
-        # writes during a bulk operation. Used for server versions < 2.6.
-        #
-        # @since 2.0.0
-        module LegacyMergable
-
-          # Aggregate the write errors returned from this result.
-          #
-          # @example Aggregate the write errors.
-          #   result.aggregate_write_errors(0)
-          #
-          # @param [ Integer ] count The number of documents already executed.
-          #
-          # @return [ Array ] The aggregate write errors.
-          #
-          # @since 2.0.0
-          def aggregate_write_errors(count)
-            @replies.each_with_index.reduce(nil) do |errors, (reply, i)|
-              if reply_write_errors?(reply)
-                errors ||= []
-                errors << { 'errmsg' => reply.documents.first[Error::ERROR],
-                            'index' => count + i,
-                            'code' => reply.documents.first[Error::CODE] }
-              end
-              errors
-            end if @replies
-          end
-
-          # Aggregate the write concern errors returned from this result.
-          #
-          # @example Aggregate the write concern errors.
-          #   result.aggregate_write_concern_errors(4)
-          #
-          # @param [ Integer ] count The number of documents already executed.
-          #
-          # @return [ Array ] The aggregate write concern errors.
-          #
-          # @since 2.0.0
-          def aggregate_write_concern_errors(count)
-            @replies.each_with_index.reduce(nil) do |errors, (reply, i)|
-              if error = reply_write_errors?(reply)
-                errors ||= []
-                if note = reply.documents.first['wnote'] || reply.documents.first['jnote']
-                  code = reply.documents.first['code'] || Error::BAD_VALUE
-                  error_string = "#{code}: #{note}"
-                elsif error == 'timeout'
-                  code = reply.documents.first['code'] || Error::UNKNOWN_ERROR
-                  error_string = "#{code}: #{error}"
-                end
-                errors << { 'errmsg' => error_string,
-                            'index' => count + i,
-                            'code' => code } if error_string
-              end
-              errors
-            end if @replies
-          end
-
-          private
-
-          def reply_write_errors?(reply)
-            reply.documents.first[Error::ERROR] ||
-              reply.documents.first[Error::ERRMSG]
-          end
-        end
-      end
-    end
-  end
-end

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

@@ -1,71 +0,0 @@
-# Copyright (C) 2014-2016 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 Bulk
-
-        # This module contains common functionality for merging results from
-        # write commands during a bulk operation. Used for server versions >= 2.6.
-        #
-        # @since 2.0.0
-        module Mergable
-
-          # Aggregate the write errors returned from this result.
-          #
-          # @example Aggregate the write errors.
-          #   result.aggregate_write_errors(0)
-          #
-          # @param [ Integer ] count The number of documents already executed.
-          #
-          # @return [ Array ] The aggregate write errors.
-          #
-          # @since 2.0.0
-          def aggregate_write_errors(count)
-            @replies.reduce(nil) do |errors, reply|
-              if write_errors = reply.documents.first[Error::WRITE_ERRORS]
-                wes = write_errors.collect do |we|
-                  we.merge!('index' => count + we['index'])
-                end
-                (errors || []) << wes if wes
-              end
-            end
-          end
-
-          # Aggregate the write concern errors returned from this result.
-          #
-          # @example Aggregate the write concern errors.
-          #   result.aggregate_write_concern_errors(100)
-          #
-          # @param [ Integer ] count The number of documents already executed.
-          #
-          # @return [ Array ] The aggregate write concern errors.
-          #
-          # @since 2.0.0
-          def aggregate_write_concern_errors(count)
-            @replies.each_with_index.reduce(nil) do |errors, (reply, _)|
-              if write_concern_errors = reply.documents.first[Error::WRITE_CONCERN_ERRORS]
-                (errors || []) << write_concern_errors.reduce(nil) do |errs, wce|
-                    wce.merge!('index' => count + wce['index'])
-                    (errs || []) << write_concern_error
-                end
-              end
-            end
-          end
-        end
-      end
-    end
-  end
-end

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

@@ -1,205 +0,0 @@
-# Copyright (C) 2014-2016 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 Bulk
-        class Update
-
-          # Defines custom behaviour of results when updating.
-          #
-          # @since 2.0.0
-          class Result < Operation::Result
-            include Mergable
-
-            # The number of modified docs field in the result.
-            #
-            # @since 2.0.0
-            MODIFIED = 'nModified'.freeze
-
-            # The upserted docs field in the result.
-            #
-            # @since 2.0.0
-            UPSERTED = 'upserted'.freeze
-
-            # Gets the number of documents upserted.
-            #
-            # @example Get the upserted count.
-            #   result.n_upserted
-            #
-            # @return [ Integer ] The number of documents upserted.
-            #
-            # @since 2.0.0
-            def n_upserted
-              return 0 unless acknowledged?
-              @replies.reduce(0) do |n, reply|
-                if upsert?(reply)
-                  n += reply.documents.first[UPSERTED].size
-                else
-                  n
-                end
-              end
-            end
-
-            # Gets the number of documents matched.
-            #
-            # @example Get the matched count.
-            #   result.n_matched
-            #
-            # @return [ Integer ] The number of documents matched.
-            #
-            # @since 2.0.0
-            def n_matched
-              return 0 unless acknowledged?
-              @replies.reduce(0) do |n, reply|
-                if upsert?(reply)
-                  reply.documents.first[N] - n_upserted
-                else
-                  if reply.documents.first[N]
-                    n += reply.documents.first[N]
-                  else
-                    n
-                  end
-                end
-              end
-            end
-
-            # Gets the number of documents modified.
-            # Not that in a mixed sharded cluster a call to
-            # update could return nModified (>= 2.6) or not (<= 2.4).
-            # If any call does not return nModified we can't report
-            # a valid final count so set the field to nil.
-            #
-            # @example Get the modified count.
-            #   result.n_modified
-            #
-            # @return [ Integer ] The number of documents modified.
-            #
-            # @since 2.0.0
-            def n_modified
-              return 0 unless acknowledged?
-              @replies.reduce(0) do |n, reply|
-                if n && reply.documents.first[MODIFIED]
-                  n += reply.documents.first[MODIFIED]
-                else
-                  nil
-                end
-              end
-            end
-
-            # Get the upserted documents.
-            #
-            # @example Get upserted documents.
-            #   result.upserted
-            #
-            # @return [ Array<BSON::Document> ] The upserted document info
-            #
-            # @since 2.1.0
-            def upserted
-              reply.documents.first[UPSERTED] || []
-            end
-
-            private
-
-            def upsert?(reply)
-              upserted.any?
-            end
-          end
-
-          # Defines custom behaviour of results when updating.
-          # For server versions < 2.5.5 (that don't use write commands).
-          #
-          # @since 2.0.0
-          class LegacyResult < Operation::Result
-            include LegacyMergable
-
-            # The updated existing field in the result.
-            #
-            # @since 2.0.0
-            UPDATED_EXISTING = 'updatedExisting'.freeze
-
-            # Gets the number of documents upserted.
-            #
-            # @example Get the upserted count.
-            #   result.n_upserted
-            #
-            # @return [ Integer ] The number of documents upserted.
-            #
-            # @since 2.0.0
-            def n_upserted
-              return 0 unless acknowledged?
-              @replies.reduce(0) do |n, reply|
-                if upsert?(reply)
-                  n += reply.documents.first[N]
-                else
-                  n
-                end
-              end
-            end
-
-            # Gets the number of documents matched.
-            #
-            # @example Get the matched count.
-            #   result.n_matched
-            #
-            # @return [ Integer ] The number of documents matched.
-            #
-            # @since 2.0.0
-            def n_matched
-              return 0 unless acknowledged?
-              @replies.reduce(0) do |n, reply|
-                if upsert?(reply)
-                  n
-                else
-                  n += reply.documents.first[N]
-                end
-              end
-            end
-
-            # Gets the number of documents modified.
-            #
-            # @example Get the modified count.
-            #   result.n_modified
-            #
-            # @return [ Integer ] The number of documents modified.
-            #
-            # @since 2.2.3
-            def n_modified
-              return 0 unless acknowledged?
-              @replies.reduce(0) do |n, reply|
-                if upsert?(reply)
-                  n
-                else
-                  updated_existing?(reply) ? n += reply.documents.first[N] : n
-                end
-              end
-            end
-
-            private
-
-            def upsert?(reply)
-              reply.documents.first[BulkWrite::Result::UPSERTED] ||
-                (!updated_existing?(reply) && reply.documents.first[N] == 1)
-            end
-
-            def updated_existing?(reply)
-              reply.documents.first[UPDATED_EXISTING]
-            end
-          end
-        end
-      end
-    end
-  end
-end