mongo 2.1.0.beta → 2.2.0

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

@@ -1,174 +0,0 @@
-# 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.
-
-require 'mongo/operation/write/bulk/bulk_mergable'
-require 'mongo/operation/write/bulk/legacy_bulk_mergable'
-
-module Mongo
-  module Operation
-    module Write
-      class BulkUpdate
-
-        # Defines custom behaviour of results when updating.
-        #
-        # @since 2.0.0
-        class Result < Operation::Result
-          include BulkMergable
-
-          # 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 += 1
-              else
-                n += 0
-              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 += 0
-              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.0.0
-          def n_modified
-            return 0 unless acknowledged?
-            @replies.reduce(0) do |n, reply|
-              n += reply.documents.first[MODIFIED] || 0
-            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 LegacyBulkMergable
-
-          # 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
-
-          private
-
-          def upsert?(reply)
-            !updated_existing?(reply) && reply.documents.first[N] == 1
-          end
-
-          def updated_existing?(reply)
-            reply.documents.first[UPDATED_EXISTING]
-          end
-        end
-      end
-    end
-  end
-end

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

@@ -1,154 +0,0 @@
-# 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.
-
-require 'mongo/operation/write/bulk/bulk_update/result'
-
-module Mongo
-  module Operation
-    module Write
-
-      # A MongoDB bulk update operation.
-      #
-      # @note If the server version is >= 2.5.5, a write command operation
-      #   will be created and sent instead.
-      #
-      # @example Create the update operation.
-      #   Write::BulkUpdate.new({
-      #     :updates => [
-      #       {
-      #         :q => { :foo => 1 },
-      #         :u => { :$set => { :bar => 1 }},
-      #         :multi  => true,
-      #         :upsert => false
-      #       }
-      #     ],
-      #     :db_name => 'test',
-      #     :coll_name => 'test_coll',
-      #     :write_concern => write_concern,
-      #     :ordered => false
-      #   })
-      #
-      # Initialization:
-      #   param [ Hash ] spec The specifications for the update.
-      #
-      #   option spec :updates [ Array ] The update documents.
-      #   option spec :db_name [ String ] The name of the database on which
-      #     the query should be run.
-      #   option spec :coll_name [ String ] The name of the collection on which
-      #     the query should be run.
-      #   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 BulkUpdate
-        include Executable
-        include Specifiable
-
-        # Execute the update operation.
-        #
-        # @example Execute the operation.
-        #   operation.execute(context)
-        #
-        # @param [ Mongo::Server::Context ] context The context for this operation.
-        #
-        # @return [ Result ] The operation result.
-        #
-        # @since 2.0.0
-        def execute(context)
-          if context.features.write_command_enabled?
-            execute_write_command(context)
-          else
-            execute_message(context)
-          end
-        end
-
-        # Set the write concern on this operation.
-        #
-        # @example Set a write concern.
-        #   new_op = operation.write_concern(:w => 2)
-        #
-        # @param [ Hash ] wc The write concern.
-        #
-        # @since 2.0.0
-        def write_concern(wc = nil)
-          if wc
-            self.class.new(spec.merge(write_concern: WriteConcern.get(wc)))
-          else
-            spec[WRITE_CONCERN]
-          end
-        end
-
-        private
-
-        def execute_write_command(context)
-          Result.new(Command::Update.new(spec).execute(context))
-        end
-
-        def execute_message(context)
-          replies = messages.map do |m|
-            context.with_connection do |connection|
-              result = LegacyResult.new(connection.dispatch([ m, gle ].compact, operation_id))
-              if stop_sending?(result)
-                return result
-              else
-                result.reply
-              end
-            end
-          end
-          LegacyResult.new(replies.compact.empty? ? nil : replies)
-        end
-
-        def stop_sending?(result)
-          ordered? && !result.successful?
-        end
-
-        # @todo put this somewhere else
-        def ordered?
-          @spec.fetch(:ordered, true)
-        end
-
-        def gle
-          gle_message = ( ordered? && write_concern.get_last_error.nil? ) ?
-                           Mongo::WriteConcern.get(:w => 1).get_last_error :
-                           write_concern.get_last_error
-          if gle_message
-            Protocol::Query.new(
-              db_name,
-              Database::COMMAND,
-              gle_message,
-              options.merge(limit: -1)
-            )
-          end
-        end
-
-        def initialize_copy(original)
-          @spec = original.spec.dup
-          @spec[UPDATES] = original.spec[UPDATES].dup
-        end
-
-        def messages
-          updates.collect do |u|
-            opts = { :flags => [] }
-            opts[:flags] << :multi_update if !!u[:multi]
-            opts[:flags] << :upsert if !!u[:upsert]
-            Protocol::Update.new(db_name, coll_name, u[:q], u[:u], opts)
-          end
-        end
-      end
-    end
-  end
-end

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

@@ -1,83 +0,0 @@
-# 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 LegacyBulkMergable
-
-        # Aggregate the write errors returned from this result.
-        #
-        # @example Aggregate the write errors.
-        #   result.aggregate_write_errors([0, 1, 2, 3])
-        #
-        # @param [ Array ] indexes The indexes of each operation as they
-        #   were listed in the Bulk API.
-        #
-        # @return [ Array ] The aggregate write errors.
-        #
-        # @since 2.0.0
-        def aggregate_write_errors(indexes)
-          @replies.each_with_index.reduce(nil) do |errors, (reply, i)|
-            if reply_write_errors?(reply)
-              errors ||= []
-              errors << { 'errmsg' => reply.documents.first[Error::ERROR],
-                          'index' => indexes[i],
-                          'code' => reply.documents.first[Error::CODE] }
-            end
-            errors
-          end
-        end
-
-        # Aggregate the write concern errors returned from this result.
-        #
-        # @example Aggregate the write concern errors.
-        #   result.aggregate_write_concern_errors([0, 1, 2, 3])
-        #
-        # @param [ Array ] indexes The indexes of each operation as they
-        #   were listed in the Bulk API.
-        #
-        # @return [ Array ] The aggregate write concern errors.
-        #
-        # @since 2.0.0
-        def aggregate_write_concern_errors(indexes)
-          @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' => indexes[i],
-                          'code' => code } if error_string
-            end
-            errors
-          end
-        end
-
-        private
-
-        def reply_write_errors?(reply)
-          reply.documents.first[Error::ERROR] ||
-            reply.documents.first[Error::ERRMSG]
-        end
-      end
-    end
-  end
-end

data/spec/mongo/grid/fs_spec.rb

@@ -1,160 +0,0 @@
-require 'spec_helper'
-
-describe Mongo::Grid::FS do
-
-  describe '#initialize' do
-
-    let(:fs) do
-      described_class.new(authorized_client.database)
-    end
-
-    let(:index) do
-      fs.chunks_collection.indexes.get(:files_id => 1, :n => 1)
-    end
-
-    it 'sets the files collection' do
-      expect(fs.files_collection.name).to eq('fs.files')
-    end
-
-    it 'sets the chunks collection' do
-      expect(fs.chunks_collection.name).to eq('fs.chunks')
-    end
-
-    it 'creates the index on the chunks collection' do
-      expect(index[:name]).to eq('files_id_1_n_1')
-    end
-  end
-
-  describe '#find_one' do
-
-    let(:fs) do
-      described_class.new(authorized_client.database)
-    end
-
-    let(:file) do
-      Mongo::Grid::File.new('hello world!', :filename => 'test.txt')
-    end
-
-    before do
-      fs.insert_one(file)
-    end
-
-    after do
-      fs.files_collection.find.delete_many
-      fs.chunks_collection.find.delete_many
-    end
-
-    let(:from_db) do
-      fs.find_one(:filename => 'test.txt')
-    end
-
-    it 'returns the assembled file from the db' do
-      expect(from_db.filename).to eq(file.metadata.filename)
-    end
-  end
-
-  describe '#insert_one' do
-
-    let(:fs) do
-      described_class.new(authorized_client.database)
-    end
-
-    let(:file) do
-      Mongo::Grid::File.new('Hello!', :filename => 'test.txt')
-    end
-
-    context 'when inserting the file once' do
-
-      let!(:result) do
-        fs.insert_one(file)
-      end
-
-      after do
-        fs.files_collection.find.delete_many
-        fs.chunks_collection.find.delete_many
-      end
-
-      let(:from_db) do
-        fs.find_one(:filename => 'test.txt')
-      end
-
-      it 'inserts the file into the database' do
-        expect(from_db.filename).to eq(file.metadata.filename)
-      end
-
-      it 'includes the chunks and data with the file' do
-        expect(from_db.data).to eq('Hello!')
-      end
-
-      it 'returns the file id' do
-        expect(result).to eq(file.id)
-      end
-    end
-
-    context 'when inserting the file more than once' do
-
-      after do
-        fs.files_collection.find.delete_many
-        fs.chunks_collection.find.delete_many
-      end
-
-      it 'raises an error' do
-        expect {
-          fs.insert_one(file)
-          fs.insert_one(file)
-        }.to raise_error(Mongo::Error::OperationFailure)
-      end
-    end
-
-    context 'when the file exceeds the max bson size' do
-
-      let(:fs) do
-        described_class.new(authorized_client.database)
-      end
-
-      let(:file) do
-        str = 'y' * 16777216
-        Mongo::Grid::File.new(str, :filename => 'large-file.txt')
-      end
-
-      before do
-        fs.insert_one(file)
-      end
-
-      after do
-        fs.files_collection.find.delete_many
-        fs.chunks_collection.find.delete_many
-      end
-
-      it 'successfully inserts the file' do
-        expect(
-          fs.find_one(:filename => 'large-file.txt').chunks
-        ).to eq(file.chunks)
-      end
-    end
-  end
-
-  describe '#delete_one' do
-
-    let(:fs) do
-      described_class.new(authorized_client.database)
-    end
-
-    let(:file) do
-      Mongo::Grid::File.new('Hello!', :filename => 'test.txt')
-    end
-
-    before do
-      fs.insert_one(file)
-      fs.delete_one(file)
-    end
-
-    let(:from_db) do
-      fs.find_one(:filename => 'test.txt')
-    end
-
-    it 'removes the file from the db' do
-      expect(from_db).to be_nil
-    end
-  end
-end

data/spec/mongo/loggable_spec.rb

@@ -1,63 +0,0 @@
-require 'spec_helper'
-
-describe Mongo::Loggable do
-
-  let(:operation) do
-    Class.new do
-      def log_message
-        'test'
-      end
-    end.new
-  end
-
-  describe '#log' do
-
-    let(:loggable) do
-      Class.new do
-        include Mongo::Loggable
-      end.new
-    end
-
-    let(:operation) do
-      double('operation')
-    end
-
-    before do
-      Mongo::Logger.level = Logger::DEBUG
-      expect(operation).to receive(:log_message).and_return('test')
-      expect(Mongo::Logger).to receive(:log).with(:debug, 'MONGO', 'test', anything())
-    end
-
-    context 'when a block is provided' do
-
-      context 'when an exception occurs' do
-
-        it 'logs the message' do
-          expect {
-            loggable.log(:debug, 'MONGO', [ operation ]) do
-              raise RuntimeError
-            end
-          }.to raise_error(RuntimeError)
-        end
-      end
-
-      context 'when no exception occurs' do
-
-        it 'executes the block and logs the message' do
-          expect(
-            loggable.log(:debug, 'MONGO', [ operation ]) do
-              'testing'
-            end
-          ).to eq('testing')
-        end
-      end
-    end
-
-    context 'when no block is provided' do
-
-      it 'logs the message' do
-        expect(loggable.log(:debug, 'MONGO', [ operation ])).to be_nil
-      end
-    end
-  end
-end