mongo 2.22.0 → 2.24.0

data/lib/mongo/server.rb

@@ -1,5 +1,4 @@
 # frozen_string_literal: true
-# rubocop:todo all
 
 # Copyright (C) 2014-2020 MongoDB Inc.
 #
@@ -16,7 +15,6 @@
 # limitations under the License.
 
 module Mongo
-
   # Represents a single server on the server side that can be standalone, part of
   # a replica set, or a mongos.
   #
@@ -29,7 +27,7 @@ module Mongo
     # The default time in seconds to timeout a connection attempt.
     #
     # @since 2.4.3
-    CONNECT_TIMEOUT = 10.freeze
+    CONNECT_TIMEOUT = 10
 
     # Instantiate a new server object. Will start the background refresh and
     # subscribe to the appropriate events.
@@ -82,21 +80,18 @@ module Mongo
       @scan_semaphore = DistinguishingSemaphore.new
       @round_trip_time_calculator = RoundTripTimeCalculator.new
       @description = Description.new(address, {},
-        load_balancer: !!@options[:load_balancer],
-        force_load_balancer: force_load_balancer?,
-      )
+                                     load_balancer: !!@options[:load_balancer],
+                                     force_load_balancer: force_load_balancer?)
       @last_scan = nil
       @last_scan_monotime = nil
       unless options[:monitoring_io] == false
         @monitor = Monitor.new(self, event_listeners, monitoring,
-          options.merge(
-            app_metadata: cluster.monitor_app_metadata,
-            push_monitor_app_metadata: cluster.push_monitor_app_metadata,
-            heartbeat_interval: cluster.heartbeat_interval,
-        ))
-        unless _monitor == false
-          start_monitoring
-        end
+                               options.merge(
+                                 app_metadata: cluster.monitor_app_metadata,
+                                 push_monitor_app_metadata: cluster.push_monitor_app_metadata,
+                                 heartbeat_interval: cluster.heartbeat_interval
+                               ))
+        start_monitoring unless _monitor == false
       end
       @connected = true
       @pool_lock = Mutex.new
@@ -108,7 +103,7 @@ module Mongo
     # @return [ Cluster ] cluster The server cluster.
     attr_reader :cluster
 
-    # @return [ nil | Monitor ] monitor The server monitor. nil if the servenr
+    # @return [ nil | Monitor ] monitor The server monitor. nil if the server
     #   was created with monitoring_io: false option.
     attr_reader :monitor
 
@@ -154,14 +149,13 @@ module Mongo
       end
     end
 
-
     # @deprecated
     def heartbeat_frequency
       cluster.heartbeat_interval
     end
 
     # @deprecated
-    alias :heartbeat_frequency_seconds :heartbeat_frequency
+    alias heartbeat_frequency_seconds heartbeat_frequency
 
     # Performs an immediate, synchronous check of the server.
     #
@@ -170,8 +164,8 @@ module Mongo
 
     # The compressor negotiated by the server monitor, if any.
     #
-    # This attribute is nil if no server check has not yet completed, and if
-    # no compression was negatiated.
+    # This attribute is nil if no server check has yet completed, and if
+    # no compression was negotiated.
     #
     # @note Compression is negotiated for each connection separately.
     #
@@ -179,11 +173,9 @@ module Mongo
     #
     # @deprecated
     def compressor
-      if monitor
-        monitor.compressor
-      else
-        nil
-      end
+      return unless monitor
+
+      monitor.compressor
     end
 
     # Delegate convenience methods to the monitor description.
@@ -218,7 +210,8 @@ module Mongo
     # @api private
     def_delegators :cluster,
                    :monitor_app_metadata,
-                   :push_monitor_app_metadata
+                   :push_monitor_app_metadata,
+                   :tracer
 
     def_delegators :features,
                    :check_driver_support!
@@ -245,6 +238,7 @@ module Mongo
     # @since 2.0.0
     def ==(other)
       return false unless other.is_a?(Server)
+
       address == other.address
     end
 
@@ -274,9 +268,7 @@ module Mongo
     #
     # @since 2.0.0
     def disconnect!
-      if monitor
-        monitor.stop!
-      end
+      monitor.stop! if monitor
 
       @connected = false
 
@@ -293,9 +285,7 @@ module Mongo
     end
 
     def close
-      if monitor
-        monitor.stop!
-      end
+      monitor.stop! if monitor
 
       @connected = false
 
@@ -330,9 +320,9 @@ module Mongo
     # @api private
     def start_monitoring
       publish_opening_event
-      if options[:monitoring_io] != false
-        monitor.run!
-      end
+      return unless options[:monitoring_io] != false
+
+      monitor.run!
     end
 
     # Publishes the server opening event.
@@ -361,24 +351,23 @@ module Mongo
     #
     # @api private
     def status
-      case
-      when load_balancer?
+      if load_balancer?
         'LB'
-      when primary?
+      elsif primary?
         'PRIMARY'
-      when secondary?
+      elsif secondary?
         'SECONDARY'
-      when standalone?
+      elsif standalone?
         'STANDALONE'
-      when arbiter?
+      elsif arbiter?
         'ARBITER'
-      when ghost?
+      elsif ghost?
         'GHOST'
-      when other?
+      elsif other?
         'OTHER'
-      when mongos?
+      elsif mongos?
         'MONGOS'
-      when unknown?
+      elsif unknown?
         'UNKNOWN'
       else
         # Since the summary method is often used for debugging, do not raise
@@ -393,23 +382,17 @@ module Mongo
     # @since 2.7.0
     def summary
       status = self.status || ''
-      if replica_set_name
-        status += " replica_set=#{replica_set_name}"
-      end
+      status += " replica_set=#{replica_set_name}" if replica_set_name
 
-      unless monitor&.running?
-        status += " NO-MONITORING"
-      end
+      status += ' NO-MONITORING' unless monitor&.running?
 
-      if @pool
-        status += " pool=#{@pool.summary}"
-      end
+      status += " pool=#{@pool.summary}" if @pool
 
       address_bit = if address
-        "#{address.host}:#{address.port}"
-      else
-        'nil'
-      end
+                      "#{address.host}:#{address.port}"
+                    else
+                      'nil'
+                    end
 
       "#<Server address=#{address_bit} #{status}>"
     end
@@ -423,9 +406,7 @@ module Mongo
     #
     # @since 2.0.0
     def pool
-      if unknown?
-        raise Error::ServerNotUsable, address
-      end
+      raise Error::ServerNotUsable, address if unknown?
 
       @pool_lock.synchronize do
         opts = connected? ? options : options.merge(populator_io: false)
@@ -474,9 +455,7 @@ module Mongo
     #
     # @since 2.1.0
     def reconnect!
-      if options[:monitoring_io] != false
-        monitor.restart!
-      end
+      monitor.restart! if options[:monitoring_io] != false
       @connected = true
     end
 
@@ -506,11 +485,13 @@ module Mongo
     def handle_handshake_failure!
       yield
     rescue Mongo::Error::SocketError, Mongo::Error::SocketTimeoutError => e
-      unknown!(
-        generation: e.generation,
-        service_id: e.service_id,
-        stop_push_monitor: true,
-      )
+      unless e.label?('SystemOverloadedError')
+        unknown!(
+          generation: e.generation,
+          service_id: e.service_id,
+          stop_push_monitor: true
+        )
+      end
       raise
     end
 
@@ -537,7 +518,7 @@ module Mongo
       unknown!(
         generation: e.generation,
         service_id: e.service_id,
-        stop_push_monitor: true,
+        stop_push_monitor: true
       )
       raise
     end
@@ -546,7 +527,7 @@ module Mongo
     #
     # @api private
     def retry_reads?
-      !!(features.sessions_enabled? && logical_session_timeout)
+      !!logical_session_timeout
     end
 
     # Will writes sent to this server be retried.
@@ -556,18 +537,17 @@ module Mongo
     #
     # @return [ true, false ] If writes will be retried.
     #
-    # @note Retryable writes are only available on server versions 3.6+ and with
-    #   sharded clusters or replica sets.
+    # @note Retryable writes are only available with sharded clusters or replica sets.
     #
     # @note Some of the conditions in this method automatically return false for
-    #       for load balanced topologies. The conditions in this method should
+    #       load balanced topologies. The conditions in this method should
     #       always be true, since load-balanced topologies are only available on
     #       MongoDB 5.0+, and not for standalone topologies. Therefore, we can
     #       assume that retry writes are enabled.
     #
     # @since 2.5.0
     def retry_writes?
-      !!(features.sessions_enabled? && logical_session_timeout && !standalone?) || load_balancer?
+      !!((logical_session_timeout && !standalone?) || load_balancer?)
     end
 
     # Marks server unknown and publishes the associated SDAM event
@@ -618,42 +598,30 @@ module Mongo
 
       # NOTE: You cannot use safe navigation here because if pool is nil you end
       # up trying to evaluate Integer < nil which is invalid.
-      if options[:generation] && pool && options[:generation] < pool.generation
-        return
-      end
+      return if options[:generation] && pool && options[:generation] < pool.generation
 
       if options[:topology_version] && description.topology_version &&
-        !options[:topology_version].gt?(description.topology_version)
-      then
+         !options[:topology_version].gt?(description.topology_version)
         return
       end
 
-      if options[:stop_push_monitor]
-        monitor&.stop_push_monitor!
-      end
+      monitor&.stop_push_monitor! if options[:stop_push_monitor]
 
       # SDAM flow will update description on the server without in-place
       # mutations and invoke SDAM transitions as needed.
       config = {}
-      if options[:service_id]
-        config['serviceId'] = options[:service_id]
-      end
-      if options[:topology_version]
-        config['topologyVersion'] = options[:topology_version]
-      end
+      config['serviceId'] = options[:service_id] if options[:service_id]
+      config['topologyVersion'] = options[:topology_version] if options[:topology_version]
       new_description = Description.new(address, config,
-        load_balancer: load_balancer?,
-        force_load_balancer: options[:connect] == :load_balanced,
-      )
+                                        load_balancer: load_balancer?,
+                                        force_load_balancer: options[:connect] == :load_balanced)
       cluster.run_sdam_flow(description, new_description, options)
     end
 
     # @api private
     def update_description(description)
       pool = pool_internal
-      if pool && !description.unknown?
-        pool.ready
-      end
+      pool.ready if pool && !description.unknown?
       @description = description
     end