Class: Mongo::Server

# File 'lib/mongo/server.rb', line 104

attr_reader :cluster

# File 'lib/mongo/server.rb', line 256

def connectable?; end

# File 'lib/mongo/server.rb', line 311

def connected?
  @connected
end

# File 'lib/mongo/server.rb', line 118

attr_reader :description

# File 'lib/mongo/server.rb', line 125

def force_load_balancer?
  options[:connect] == :load_balanced
end

# File 'lib/mongo/server.rb', line 108

attr_reader :monitor

# File 'lib/mongo/server.rb', line 114

attr_reader :monitoring

# File 'lib/mongo/server.rb', line 111

attr_reader :options

# File 'lib/mongo/server.rb', line 529

def retry_reads?
  !!logical_session_timeout
end

# File 'lib/mongo/server.rb', line 549

def retry_writes?
  !!((logical_session_timeout && !standalone?) || load_balancer?)
end

# File 'lib/mongo/server.rb', line 227

attr_reader :round_trip_time_calculator

# File 'lib/mongo/server.rb', line 223

attr_reader :scan_semaphore

# File 'lib/mongo/server.rb', line 642

def clear_connection_pool(service_id: nil, interrupt_in_use_connections: false)
  @pool_lock.synchronize do
    # A server being marked unknown after it is closed is technically
    # incorrect but it does not meaningfully alter any state.
    # Because historically the driver permitted servers to be marked
    # unknown at any time, continue doing so even if the pool is closed.
    if @pool && !@pool.closed?
      @pool.disconnect!(service_id: service_id, interrupt_in_use_connections: interrupt_in_use_connections)
    end
  end
end

# File 'lib/mongo/server.rb', line 632

def clear_description
  @description = Mongo::Server::Description.new(address, {})
end

# File 'lib/mongo/server.rb', line 287

def close
  monitor.stop! if monitor

  @connected = false

  _pool = nil
  @pool_lock.synchronize do
    _pool, @pool = @pool, nil
  end

  # TODO: change this to _pool.close in RUBY-3174.
  # Clear the pool. If the server is not unknown then the
  # pool will stay ready. Stop the background populator thread.
  _pool&.close(stay_ready: true)

  nil
end

# File 'lib/mongo/server.rb', line 175

def compressor
  return unless monitor

  monitor.compressor
end

# File 'lib/mongo/server.rb', line 270

def disconnect!
  monitor.stop! if monitor

  @connected = false

  # The current CMAP spec requires a pool to be mostly unusable
  # if its server is unknown (or, therefore, disconnected).
  # However any outstanding operations should continue to completion,
  # and their connections need to be checked into the pool to be
  # torn down. Because of this cleanup requirement we cannot just
  # close the pool and set it to nil here, to be recreated the next
  # time the server is discovered.
  pool_internal&.clear

  true
end

# File 'lib/mongo/server.rb', line 510

def handle_auth_failure!
  yield
rescue Mongo::Error::SocketTimeoutError
  # possibly cluster is slow, do not give up on it
  raise
rescue Mongo::Error::SocketError, Auth::Unauthorized => e
  # non-timeout network error or auth error, clear the pool and mark the
  # topology as unknown
  unknown!(
    generation: e.generation,
    service_id: e.service_id,
    stop_push_monitor: true
  )
  raise
end

# File 'lib/mongo/server.rb', line 485

def handle_handshake_failure!
  yield
rescue Mongo::Error::SocketError, Mongo::Error::SocketTimeoutError => e
  unless e.label?('SystemOverloadedError')
    unknown!(
      generation: e.generation,
      service_id: e.service_id,
      stop_push_monitor: true
    )
  end
  raise
end

# File 'lib/mongo/server.rb', line 153

def heartbeat_frequency
  cluster.heartbeat_interval
end

# File 'lib/mongo/server.rb', line 158

alias heartbeat_frequency_seconds heartbeat_frequency

# File 'lib/mongo/server.rb', line 346

def inspect
  "#<Mongo::Server:0x#{object_id} address=#{address.host}:#{address.port} #{status}>"
end

# File 'lib/mongo/server.rb', line 133

def last_scan
  if description && !description.config.empty?
    description.last_update_time
  else
    @last_scan
  end
end

# File 'lib/mongo/server.rb', line 144

def last_scan_monotime
  if description && !description.config.empty?
    description.last_update_monotime
  else
    @last_scan_monotime
  end
end

# File 'lib/mongo/server.rb', line 443

def matches_tag_set?(tag_set)
  tag_set.keys.all? do |k|
    tags[k] && tags[k] == tag_set[k]
  end
end

# File 'lib/mongo/server.rb', line 655

def next_connection_id
  @connection_id_gen.next_id
end

# File 'lib/mongo/server.rb', line 408

def pool
  raise Error::ServerNotUsable, address if unknown?

  @pool_lock.synchronize do
    opts = connected? ? options : options.merge(populator_io: false)
    @pool ||= ConnectionPool.new(self, opts).tap do |pool|
      pool.ready
    end
  end
end

# File 'lib/mongo/server.rb', line 427

def pool_internal
  @pool_lock.synchronize do
    @pool
  end
end

# File 'lib/mongo/server.rb', line 331

def publish_opening_event
  publish_sdam_event(
    Monitoring::SERVER_OPENING,
    Monitoring::Event::ServerOpening.new(address, cluster.topology)
  )
end

# File 'lib/mongo/server.rb', line 457

def reconnect!
  monitor.restart! if options[:monitoring_io] != false
  @connected = true
end

# File 'lib/mongo/server.rb', line 321

def start_monitoring
  publish_opening_event
  return unless options[:monitoring_io] != false

  monitor.run!
end

# File 'lib/mongo/server.rb', line 353

def status
  if load_balancer?
    'LB'
  elsif primary?
    'PRIMARY'
  elsif secondary?
    'SECONDARY'
  elsif standalone?
    'STANDALONE'
  elsif arbiter?
    'ARBITER'
  elsif ghost?
    'GHOST'
  elsif other?
    'OTHER'
  elsif mongos?
    'MONGOS'
  elsif unknown?
    'UNKNOWN'
  else
    # Since the summary method is often used for debugging, do not raise
    # an exception in case none of the expected types matched
    nil
  end
end

# File 'lib/mongo/server.rb', line 383

def summary
  status = self.status || ''
  status += " replica_set=#{replica_set_name}" if replica_set_name

  status += ' NO-MONITORING' unless monitor&.running?

  status += " pool=#{@pool.summary}" if @pool

  address_bit = if address
                  "#{address.host}:#{address.port}"
                else
                  'nil'
                end

  "#<Server address=#{address_bit} #{status}>"
end

# File 'lib/mongo/server.rb', line 577

def unknown!(options = {})
  pool = pool_internal

  if load_balancer?
    # When the client is in load-balanced topology, servers (the one and
    # only that can be) starts out as a load balancer and stays as a
    # load balancer indefinitely. As such it is not marked unknown.
    #
    # However, this method also clears connection pool for the server
    # when the latter is marked unknown, and this part needs to happen
    # when the server is a load balancer.
    #
    # It is possible for a load balancer server to not have a service id,
    # for example if there haven't been any successful connections yet to
    # this server, but the server can still be marked unknown if one
    # of such connections failed midway through its establishment.
    if service_id = options[:service_id]
      pool&.disconnect!(service_id: service_id)
    end
    return
  end

  # NOTE: You cannot use safe navigation here because if pool is nil you end
  # up trying to evaluate Integer < nil which is invalid.
  return if options[:generation] && pool && options[:generation] < pool.generation

  if options[:topology_version] && description.topology_version &&
     !options[:topology_version].gt?(description.topology_version)
    return
  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 = {}
  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)
  cluster.run_sdam_flow(description, new_description, options)
end

# File 'lib/mongo/server.rb', line 622

def update_description(description)
  pool = pool_internal
  pool.ready if pool && !description.unknown?
  @description = description
end

# File 'lib/mongo/server.rb', line 660

def update_last_scan
  @last_scan = Time.now
  @last_scan_monotime = Utils.monotonic_time
end

# File 'lib/mongo/server.rb', line 473

def with_connection(connection_global_id: nil, context: nil, &block)
  pool.with_connection(
    connection_global_id: connection_global_id,
    context: context,
    &block
  )
end