mongo 2.12.4 → 2.13.0.beta1

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

@@ -1,4 +1,4 @@
-# Copyright (C) 2018-2019 MongoDB, Inc.
+# Copyright (C) 2018-2020 MongoDB Inc.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -26,7 +26,7 @@ module Mongo
 
         private
 
-        def selector(server)
+        def selector(connection)
           { :dropUser => user_name }
         end
       end

data/lib/mongo/operation/result.rb

@@ -1,4 +1,4 @@
-# Copyright (C) 2014-2019 MongoDB, Inc.
+# Copyright (C) 2014-2020 MongoDB Inc.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -80,9 +80,14 @@ module Mongo
       #
       # @param [ Protocol::Message | Array<Protocol::Message> | nil ] replies
       #  The wire protocol replies.
+      # @param [ Server::Description | nil ] connection_description
+      #   Server description of the server that performed the operation that
+      #   this result is for. This parameter is allowed to be nil for
+      #   compatibility with existing mongo_kerberos library, but should
+      #   always be not nil in the driver proper.
       #
       # @api private
-      def initialize(replies)
+      def initialize(replies, connection_description = nil)
         if replies
           if replies.is_a?(Array)
             if replies.length != 1
@@ -96,12 +101,19 @@ module Mongo
             raise ArgumentError, "Argument must be a Message instance, but is a #{reply.class}: #{reply.inspect}"
           end
           @replies = [ reply ]
+          @connection_description = connection_description
         end
       end
 
       # @return [ Array<Protocol::Message> ] replies The wrapped wire protocol replies.
       attr_reader :replies
 
+      # @return [ Server::Description ] Server description of the server that
+      #   the operation was performed on that this result is for.
+      #
+      # @api private
+      attr_reader :connection_description
+
       # @api private
       def_delegators :parser,
         :not_master?, :node_recovering?, :node_shutting_down?
@@ -291,11 +303,14 @@ module Mongo
           self,
           code: parser.code,
           code_name: parser.code_name,
-          write_concern_error: parser.write_concern_error?,
+          write_concern_error_document: parser.write_concern_error_document,
           write_concern_error_code: parser.write_concern_error_code,
           write_concern_error_code_name: parser.write_concern_error_code_name,
+          write_concern_error_labels: parser.write_concern_error_labels,
           labels: parser.labels,
-          wtimeout: parser.wtimeout)
+          wtimeout: parser.wtimeout,
+          connection_description: connection_description,
+        )
       end
 
       # Raises a Mongo::OperationFailure exception corresponding to the
@@ -307,6 +322,15 @@ module Mongo
       end
       private :raise_operation_failure
 
+      # @return [ TopologyVersion | nil ] The topology version.
+      def topology_version
+        unless defined?(@topology_version)
+          @topology_version = first_document['topologyVersion'] &&
+            TopologyVersion.new(first_document['topologyVersion'])
+        end
+        @topology_version
+      end
+
       # Get the number of documents written by the server.
       #
       # @example Get the number of documents written.

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

@@ -1,4 +1,4 @@
-# Copyright (C) 2018-2019 MongoDB, Inc.
+# Copyright (C) 2018-2020 MongoDB Inc.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -22,12 +22,18 @@ module Mongo
 
       private
 
-      def command(server)
+      def command(connection)
+        if Lint.enabled?
+          unless connection.is_a?(Server::Connection)
+            raise Error::LintError, "Connection is not a Connection instance: #{connection}"
+          end
+        end
+
         sel = super
-        add_bypass_document_validation(sel, server)
+        add_bypass_document_validation(sel)
       end
 
-      def add_bypass_document_validation(sel, server)
+      def add_bypass_document_validation(sel)
         return sel unless bypass_document_validation
         sel.merge(bypassDocumentValidation: true)
       end

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

@@ -1,4 +1,4 @@
-# Copyright (C) 2018-2019 MongoDB, Inc.
+# Copyright (C) 2018-2020 MongoDB Inc.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -33,8 +33,8 @@ module Mongo
       # In order for the override to work correctly the
       # CausalConsistencySupported module must be included after
       # SessionsSupported module in target classes.
-      def apply_causal_consistency!(selector, server)
-        apply_causal_consistency_if_possible(selector, server)
+      def apply_causal_consistency!(selector, connection)
+        apply_causal_consistency_if_possible(selector, connection)
       end
     end
   end

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

@@ -0,0 +1,56 @@
+# Copyright (C) 2020 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 CollectionsInfoOrListCollections
+      include PolymorphicLookup
+
+      # Execute the operation.
+      #
+      # @example
+      #   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.
+      def execute(server, client:)
+        server.with_connection do |connection|
+          operation = final_operation(connection)
+          operation.execute(connection, client: client)
+        end
+      end
+
+      private
+
+      def final_operation(connection)
+         op_class = if connection.features.list_collections_enabled?
+          if connection.features.op_msg_enabled?
+            ListCollections::OpMsg
+          else
+            ListCollections::Command
+          end
+        else
+          CollectionsInfo::Command
+        end
+
+        op_class.new(spec)
+      end
+    end
+  end
+end

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

@@ -1,4 +1,4 @@
-# Copyright (C) 2015-2019 MongoDB, Inc.
+# Copyright (C) 2015-2020 MongoDB Inc.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -22,21 +22,27 @@ module Mongo
 
       include ResponseHandling
 
-      def do_execute(server, client, options = {})
+      def do_execute(connection, client, options = {})
         unpin_maybe(session) do
-          add_error_labels do
-            add_server_diagnostics(server) do
-              get_result(server, client, options).tap do |result|
-                process_result(result, server)
+          add_error_labels(client, connection, session) do
+            add_server_diagnostics(connection) do
+              get_result(connection, client, options).tap do |result|
+                process_result(result, connection)
               end
             end
           end
         end
       end
 
-      def execute(server, client:, options: {})
-        do_execute(server, client, options).tap do |result|
-          validate_result(result, server)
+      def execute(connection, client:, options: {})
+        if Lint.enabled?
+          unless connection.is_a?(Mongo::Server::Connection)
+            raise Error::LintError, "Connection argument is of wrong type: #{connection}"
+          end
+        end
+
+        do_execute(connection, client, options).tap do |result|
+          validate_result(result, client, connection)
         end
       end
 
@@ -46,41 +52,53 @@ module Mongo
         Result
       end
 
-      def get_result(server, client, options = {})
-        result_class.new(dispatch_message(server, client, options))
+      def get_result(connection, client, options = {})
+        result_class.new(*dispatch_message(connection, client, options))
       end
 
       # Returns a Protocol::Message or nil as reply.
-      def dispatch_message(server, client, options = {})
-        server.with_connection do |connection|
-          message = build_message(server)
-          message = message.maybe_encrypt(server, client)
-          connection.dispatch([ message ], operation_id, client, options)
-        end
+      def dispatch_message(connection, client, options = {})
+        message = build_message(connection)
+        message = message.maybe_encrypt(connection, client)
+        reply = connection.dispatch([ message ], operation_id, client, options)
+        [reply, connection.description]
       end
 
-      def build_message(server)
-        message(server)
+      def build_message(connection)
+        message(connection)
       end
 
-      def process_result(result, server)
-        server.update_cluster_time(result)
+      def process_result(result, connection)
+        connection.server.update_cluster_time(result)
+
+        process_result_for_sdam(result, connection)
+
+        if session
+          session.process(result)
+        end
+
+        result
+      end
 
-        if result.not_master? || result.node_recovering?
+      def process_result_for_sdam(result, connection)
+        if (result.not_master? || result.node_recovering?) &&
+          connection.generation >= connection.server.pool.generation
+        then
           if result.node_shutting_down?
             keep_pool = false
           else
             # Max wire version needs to be examined while the server is known
-            keep_pool = server.description.server_version_gte?('4.2')
+            keep_pool = connection.description.server_version_gte?('4.2')
           end
 
-          server.unknown!(keep_connection_pool: keep_pool)
+          connection.server.unknown!(
+            keep_connection_pool: keep_pool,
+            generation: connection.generation,
+            topology_version: result.topology_version,
+          )
 
-          server.scan_semaphore.signal
+          connection.server.scan_semaphore.signal
         end
-
-        session.process(result) if session
-        result
       end
     end
   end

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

@@ -1,4 +1,4 @@
-# Copyright (C) 2018-2019 MongoDB, Inc.
+# Copyright (C) 2018-2020 MongoDB Inc.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -21,8 +21,8 @@ module Mongo
     # @api private
     module ExecutableNoValidate
 
-      def execute(server, client:)
-        do_execute(server, client)
+      def execute(connection, client:)
+        do_execute(connection, client)
       end
     end
   end

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

@@ -1,4 +1,4 @@
-# Copyright (C) 2018-2019 MongoDB, Inc.
+# Copyright (C) 2018-2020 MongoDB Inc.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

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

@@ -1,4 +1,4 @@
-# Copyright (C) 2018-2019 MongoDB, Inc.
+# Copyright (C) 2018-2020 MongoDB Inc.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

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

@@ -1,4 +1,4 @@
-# Copyright (C) 2015-2019 MongoDB, Inc.
+# Copyright (C) 2015-2020 MongoDB Inc.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -22,7 +22,15 @@ module Mongo
 
       private
 
-      def options(server = nil)
+      # Get the options for executing the operation on a particular connection.
+      #
+      # @param [ Server::Connection ] connection The connection that the
+      #   operation will be executed on.
+      #
+      # @return [ Hash ] The options.
+      #
+      # @since 2.0.0
+      def options(connection)
         super.merge(limit: -1)
       end
     end

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

@@ -1,4 +1,4 @@
-# Copyright (C) 2015-2019 MongoDB, Inc.
+# Copyright (C) 2015-2020 MongoDB Inc.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

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

@@ -1,4 +1,4 @@
-# Copyright (C) 2018-2019 MongoDB, Inc.
+# Copyright (C) 2018-2020 MongoDB Inc.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -23,14 +23,16 @@ module Mongo
       include PolymorphicLookup
 
       def execute(server, client:, options: {})
-        operation = final_operation(server)
-        operation.execute(server, client: client, options: options)
+        server.with_connection do |connection|
+          operation = final_operation(connection)
+          operation.execute(connection, client: client, options: options)
+        end
       end
 
       private
 
-      def final_operation(server)
-        cls = if server.features.op_msg_enabled?
+      def final_operation(connection)
+        cls = if connection.features.op_msg_enabled?
           polymorphic_class(self.class.name, :OpMsg)
         else
           polymorphic_class(self.class.name, :Command)

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

@@ -1,4 +1,4 @@
-# Copyright (C) 2018-2019 MongoDB, Inc.
+# Copyright (C) 2018-2020 MongoDB Inc.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -24,16 +24,18 @@ module Mongo
       include PolymorphicLookup
 
       def execute(server, client:)
-        operation = final_operation(server)
-        operation.execute(server, client: client)
+        server.with_connection do |connection|
+          operation = final_operation(connection)
+          operation.execute(connection, client: client)
+        end
       end
 
       private
 
-      def final_operation(server)
-        cls = if server.features.op_msg_enabled?
+      def final_operation(connection)
+        cls = if connection.features.op_msg_enabled?
           polymorphic_class(self.class.name, :OpMsg)
-        elsif server.features.find_command_enabled?
+        elsif connection.features.find_command_enabled?
           polymorphic_class(self.class.name, :Command)
         else
           polymorphic_class(self.class.name, :Legacy)

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

@@ -1,4 +1,4 @@
-# Copyright (C) 2018-2019 MongoDB, Inc.
+# Copyright (C) 2018-2020 MongoDB Inc.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -24,16 +24,18 @@ module Mongo
       include PolymorphicLookup
 
       def execute(server, client:)
-        operation = final_operation(server)
-        operation.execute(server, client: client)
+        server.with_connection do |connection|
+          operation = final_operation(connection)
+          operation.execute(connection, client: client)
+        end
       end
 
       private
 
-      def final_operation(server)
-        cls = if server.features.op_msg_enabled?
+      def final_operation(connection)
+        cls = if connection.features.op_msg_enabled?
           polymorphic_class(self.class.name, :OpMsg)
-        elsif server.features.list_indexes_enabled?
+        elsif connection.features.list_indexes_enabled?
           polymorphic_class(self.class.name, :Command)
         else
           polymorphic_class(self.class.name, :Legacy)