mongo 2.11.6 → 2.12.0.rc0

data/lib/mongo/index/view.rb

@@ -158,7 +158,7 @@ module Mongo
                   session: session
                  }
           spec[:write_concern] = write_concern if server.features.collation_enabled?
-          Operation::CreateIndex.new(spec).execute(server)
+          Operation::CreateIndex.new(spec).execute(server, client: client)
         end
       end
 
@@ -236,7 +236,7 @@ module Mongo
                  }
           server = next_primary(nil, session)
           spec[:write_concern] = write_concern if server.features.collation_enabled?
-          Operation::DropIndex.new(spec).execute(server)
+          Operation::DropIndex.new(spec).execute(server, client: client)
         end
       end
 
@@ -272,7 +272,7 @@ module Mongo
       end
 
       def send_initial_query(server, session)
-        initial_query_op(session).execute(server)
+        initial_query_op(session).execute(server, client: client)
       end
 
       def with_generated_names(models, server)

data/lib/mongo/monitoring/event/command_started.rb

@@ -99,7 +99,12 @@ module Mongo
             address,
             payload[:request_id],
             operation_id,
-            payload[:command],
+            # All op_msg  payloads have a $db field. Legacy payloads do not
+            # have a $db field. To emulate op_msg when publishing command
+            # monitoring events for legacy servers, add $db to the payload,
+            # copying the database name. Note that the database name is also
+            # available as a top-level attribute on the command started event.
+            payload[:command].merge('$db' => payload[:database_name]),
             socket_object_id: socket_object_id,
             connection_id: connection_id,
           )

data/lib/mongo/operation/collections_info.rb

@@ -31,17 +31,20 @@ module Mongo
       # Execute the operation.
       #
       # @example
-      #   operation.execute(server)
+      #   operation.execute(server, client: nil)
       #
       # @param [ Mongo::Server ] server The server to send the operation to.
+      # @param [ Mongo::Client ] client The client that will be used to
+      #   perform auto-encryption if it is necessary to encrypt the command
+      #   being executed (optional).
       #
       # @return [ Mongo::Operation::CollectionsInfo::Result,
       #           Mongo::Operation::ListCollections::Result ] The operation result.
       #
       # @since 2.0.0
-      def execute(server)
+      def execute(server, client:)
         if server.features.list_collections_enabled?
-          return Operation::ListCollections.new(spec).execute(server)
+          return Operation::ListCollections.new(spec).execute(server, client: client)
         end
 
         super

data/lib/mongo/operation/delete/op_msg.rb

@@ -36,7 +36,7 @@ module Mongo
         end
 
         def message(server)
-          section = {  type: 1, payload: { identifier: IDENTIFIER, sequence: send(IDENTIFIER) } }
+          section = Protocol::Msg::Section1.new(IDENTIFIER, send(IDENTIFIER))
           Protocol::Msg.new(flags, {}, command(server), section)
         end
       end

data/lib/mongo/operation/find/op_msg.rb

@@ -19,9 +19,12 @@ module Mongo
       # Execute the operation.
       #
       # @example
-      #   operation.execute(server)
+      #   operation.execute(server, client: nil)
       #
       # @param [ Mongo::Server ] server The server to send the operation to.
+      # @param [ Mongo::Client ] client The client that will be used to
+      #   perform auto-encryption if it is necessary to encrypt the command
+      #   being executed (optional).
       #
       # @return [ Mongo::Operation::Find::Result ] The operation result.
       #

data/lib/mongo/operation/get_more/op_msg.rb

@@ -19,9 +19,12 @@ module Mongo
       # Execute the operation.
       #
       # @example
-      #   operation.execute(server)
+      #   operation.execute(server, client: nil)
       #
       # @param [ Mongo::Server ] server The server to send the operation to.
+      # @param [ Mongo::Client ] client The client that will be used to
+      #   perform auto-encryption if it is necessary to encrypt the command
+      #   being executed (optional).
       #
       # @return [ Mongo::Operation::GetMore::Result ] The operation result.
       #

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

@@ -32,8 +32,8 @@ module Mongo
 
         private
 
-        def get_result(server)
-          Result.new(dispatch_message(server), @ids)
+        def get_result(server, client)
+          Result.new(dispatch_message(server, client), @ids)
         end
 
         def selector(server)

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

@@ -30,8 +30,8 @@ module Mongo
 
         private
 
-        def get_result(server)
-          Result.new(dispatch_message(server), @ids)
+        def get_result(server, client)
+          Result.new(dispatch_message(server, client), @ids)
         end
 
         def selector

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

@@ -30,9 +30,9 @@ module Mongo
 
         private
 
-        def get_result(server)
+        def get_result(server, client)
           # This is a Mongo::Operation::Insert::Result
-          Result.new(dispatch_message(server), @ids)
+          Result.new(dispatch_message(server, client), @ids)
         end
 
         def selector(server)
@@ -42,7 +42,7 @@ module Mongo
         end
 
         def message(server)
-          section = { type: 1, payload: { identifier: IDENTIFIER, sequence: send(IDENTIFIER) } }
+          section = Protocol::Msg::Section1.new(IDENTIFIER, send(IDENTIFIER))
           Protocol::Msg.new(flags, { validating_keys: true }, command(server), section)
         end
       end

data/lib/mongo/operation/result.rb

@@ -18,7 +18,13 @@ require 'mongo/operation/shared/result/use_legacy_error_parser'
 module Mongo
   module Operation
 
-    # Result wrapper for operations.
+    # Result wrapper for wire protocol replies.
+    #
+    # An operation has zero or one replies. The only operations producing zero
+    # replies are unacknowledged writes; all other operations produce one reply.
+    # This class provides an object that can be operated on (for example, to
+    # check whether an operation succeeded) even when the operation did not
+    # produce a reply (in which case it is assumed to have succeeded).
     #
     # @since 2.0.0
     class Result
@@ -67,17 +73,33 @@ module Mongo
 
       # Initialize a new result.
       #
-      # @example Instantiate the result.
-      #   Result.new(replies)
+      # For an unkacknowledged write, pass nil in replies.
       #
-      # @param [ Protocol::Reply ] replies The wire protocol replies.
+      # For all other operations, replies must be a Protocol::Message instance
+      # or an array containing a single Protocol::Message instance.
       #
-      # @since 2.0.0
+      # @param [ Protocol::Message | Array<Protocol::Message> | nil ] replies
+      #  The wire protocol replies.
+      #
+      # @api private
       def initialize(replies)
-        @replies = [ *replies ] if replies
+        if replies
+          if replies.is_a?(Array)
+            if replies.length != 1
+              raise ArgumentError, "Only one (or zero) reply is supported, given #{replies.length}"
+            end
+            reply = replies.first
+          else
+            reply = replies
+          end
+          unless reply.is_a?(Protocol::Message)
+            raise ArgumentError, "Argument must be a Message instance, but is a #{reply.class}: #{reply.inspect}"
+          end
+          @replies = [ reply ]
+        end
       end
 
-      # @return [ Array<Protocol::Reply> ] replies The wrapped wire protocol replies.
+      # @return [ Array<Protocol::Message> ] replies The wrapped wire protocol replies.
       attr_reader :replies
 
       # @api private
@@ -99,19 +121,6 @@ module Mongo
         !!@replies
       end
 
-      # Determine if this result is a collection of multiple replies from the
-      # server.
-      #
-      # @example Is the result for multiple replies?
-      #   result.multiple?
-      #
-      # @return [ true, false ] If the result is for multiple replies.
-      #
-      # @since 2.0.0
-      def multiple?
-        replies.size > 1
-      end
-
       # Get the cursor id if the response is acknowledged.
       #
       # @note Cursor ids of 0 indicate there is no cursor on the server.
@@ -146,7 +155,7 @@ module Mongo
       # @since 2.0.0
       def documents
         if acknowledged?
-          replies.flat_map{ |reply| reply.documents }
+          replies.flat_map(&:documents)
         else
           []
         end
@@ -180,12 +189,12 @@ module Mongo
         "#<#{self.class.name}:0x#{object_id} documents=#{documents}>"
       end
 
-      # Get the first reply from the result.
+      # Get the reply from the result.
       #
-      # @example Get the first reply.
-      #   result.reply
+      # Returns nil if there is no reply (i.e. the operation was an
+      # unacknowledged write).
       #
-      # @return [ Protocol::Reply ] The first reply.
+      # @return [ Protocol::Message ] The first reply.
       #
       # @since 2.0.0
       def reply
@@ -206,7 +215,7 @@ module Mongo
       # @since 2.0.0
       def returned_count
         if acknowledged?
-          multiple? ? aggregate_returned_count : reply.number_returned
+          reply.number_returned
         else
           0
         end
@@ -308,7 +317,7 @@ module Mongo
       # @since 2.0.0
       def written_count
         if acknowledged?
-          multiple? ? aggregate_written_count : (first_document[N] || 0)
+          first_document[N] || 0
         else
           0
         end

data/lib/mongo/operation/shared/executable.rb

@@ -22,11 +22,11 @@ module Mongo
 
       include ResponseHandling
 
-      def do_execute(server)
+      def do_execute(server, client)
         unpin_maybe(session) do
           add_error_labels do
             add_server_diagnostics(server) do
-              get_result(server).tap do |result|
+              get_result(server, client).tap do |result|
                 process_result(result, server)
               end
             end
@@ -34,8 +34,8 @@ module Mongo
         end
       end
 
-      def execute(server)
-        do_execute(server).tap do |result|
+      def execute(server, client:)
+        do_execute(server, client).tap do |result|
           validate_result(result, server)
         end
       end
@@ -46,14 +46,16 @@ module Mongo
         Result
       end
 
-      def get_result(server)
-        result_class.new(dispatch_message(server))
+      def get_result(server, client)
+        result_class.new(dispatch_message(server, client))
       end
 
       # Returns a Protocol::Message or nil
-      def dispatch_message(server)
+      def dispatch_message(server, client)
         server.with_connection do |connection|
-          connection.dispatch([ build_message(server) ], operation_id)
+          message = build_message(server)
+          message = message.maybe_encrypt(server, client)
+          connection.dispatch([ message ], operation_id, client)
         end
       end
 

data/lib/mongo/operation/shared/executable_no_validate.rb

@@ -21,8 +21,8 @@ module Mongo
     # @api private
     module ExecutableNoValidate
 
-      def execute(server)
-        do_execute(server)
+      def execute(server, client:)
+        do_execute(server, client)
       end
     end
   end

data/lib/mongo/operation/shared/op_msg_or_command.rb

@@ -22,9 +22,9 @@ module Mongo
     module OpMsgOrCommand
       include PolymorphicLookup
 
-      def execute(server)
+      def execute(server, client:)
         operation = final_operation(server)
-        operation.execute(server)
+        operation.execute(server, client: client)
       end
 
       private

data/lib/mongo/operation/shared/op_msg_or_find_command.rb

@@ -23,9 +23,9 @@ module Mongo
     module OpMsgOrFindCommand
       include PolymorphicLookup
 
-      def execute(server)
+      def execute(server, client:)
         operation = final_operation(server)
-        operation.execute(server)
+        operation.execute(server, client: client)
       end
 
       private

data/lib/mongo/operation/shared/op_msg_or_list_indexes_command.rb

@@ -23,9 +23,9 @@ module Mongo
     module OpMsgOrListIndexesCommand
       include PolymorphicLookup
 
-      def execute(server)
+      def execute(server, client:)
         operation = final_operation(server)
-        operation.execute(server)
+        operation.execute(server, client: client)
       end
 
       private

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

@@ -25,42 +25,49 @@ module Mongo
       # Execute the operation.
       #
       # @example
-      #   operation.execute(server)
+      #   operation.execute(server, client: nil)
       #
       # @param [ Mongo::Server ] server The server to send the operation to.
+      # @param [ Mongo::Client ] client The client that will be used to
+      #   perform auto-encryption if it is necessary to encrypt the command
+      #   being executed (optional).
       #
       # @return [ Mongo::Operation::Result ] The operation result.
       #
       # @since 2.5.2
-      def execute(server)
+      def execute(server, client:)
         validate!
-        result = if server.features.op_msg_enabled?
-            self.class::OpMsg.new(spec).execute(server)
+        op = if server.features.op_msg_enabled?
+            self.class::OpMsg.new(spec)
           elsif !acknowledged_write?
-            self.class::Legacy.new(spec).execute(server)
+            self.class::Legacy.new(spec)
           else
-            self.class::Command.new(spec).execute(server)
+            self.class::Command.new(spec)
           end
+        result = op.execute(server, client: client)
         validate_result(result, server)
       end
 
       # Execute the bulk write operation.
       #
       # @example
-      #   operation.bulk_execute(server)
+      #   operation.bulk_execute(server, client: nil)
       #
       # @param [ Mongo::Server ] server The server to send the operation to.
+      # @param [ Mongo::Client ] client The client that will be used to
+      #   perform auto-encryption if it is necessary to encrypt the command
+      #   being executed (optional).
       #
       # @return [ Mongo::Operation::Delete::BulkResult,
       #           Mongo::Operation::Insert::BulkResult,
       #           Mongo::Operation::Update::BulkResult ] The bulk result.
       #
       # @since 2.5.2
-      def bulk_execute(server)
+      def bulk_execute(server, client:)
         if server.features.op_msg_enabled?
-          self.class::OpMsg.new(spec).execute(server).bulk_result
+          self.class::OpMsg.new(spec).execute(server, client: client).bulk_result
         else
-          self.class::Command.new(spec).execute(server).bulk_result
+          self.class::Command.new(spec).execute(server, client: client).bulk_result
         end
       end
 

data/lib/mongo/operation/update/op_msg.rb

@@ -36,7 +36,7 @@ module Mongo
         end
 
         def message(server)
-          section = { type: 1, payload: { identifier: IDENTIFIER, sequence: send(IDENTIFIER) } }
+          section = Protocol::Msg::Section1.new(IDENTIFIER, send(IDENTIFIER))
           Protocol::Msg.new(flags, {}, command(server), section)
         end
       end

data/lib/mongo/protocol/compressed.rb

@@ -60,15 +60,13 @@ module Mongo
         @request_id = message.request_id
       end
 
-      # Inflate an OP_COMRESSED message and return the original message.
-      #
-      # @example Inflate a compressed message.
-      #  message.inflate!
+      # Inflates an OP_COMRESSED message and returns the original message.
       #
       # @return [ Protocol::Message ] The inflated message.
       #
       # @since 2.5.0
-      def inflate!
+      # @api private
+      def maybe_inflate
         message = Registry.get(@original_op_code).allocate
         uncompressed_message = Zlib::Inflate.inflate(@compressed_message)
 
@@ -81,6 +79,9 @@ module Mongo
             Message.deserialize_field(message, buf, field)
           end
         end
+        if message.is_a?(Msg)
+          message.fix_after_deserialization
+        end
         message
       end