Current File : //opt/puppetlabs/puppet/lib/ruby/vendor_ruby/puppet/http/resolver/srv.rb

# Resolve a service using DNS SRV records.
#
# @api public
class Puppet::HTTP::Resolver::SRV < Puppet::HTTP::Resolver

  # Create an DNS SRV resolver.
  #
  # @param [Puppet::HTTP::Client] client
  # @param [String] domain srv domain
  # @param [Resolv::DNS] dns
  #
  def initialize(client, domain:, dns: Resolv::DNS.new)
    @client = client
    @srv_domain = domain
    @delegate = Puppet::HTTP::DNS.new(dns)
  end

  # Walk the available srv records and return the first that successfully connects
  #
  # @param [Puppet::HTTP::Session] session
  # @param [Symbol] name the service being resolved
  # @param [Puppet::SSL::SSLContext] ssl_context
  # @param [Proc] canceled_handler optional callback allowing a resolver
  #   to cancel resolution.
  #
  # @return [Puppet::HTTP::Service] if an available service is found, return
  #   it. Return nil otherwise.
  #
  # @api public
  def resolve(session, name, ssl_context: nil, canceled_handler: nil)
    # Here we pass our HTTP service name as the DNS SRV service name
    # This is fine for :ca, but note that :puppet and :file are handled
    # specially in `each_srv_record`.
    @delegate.each_srv_record(@srv_domain, name) do |server, port|
      service = Puppet::HTTP::Service.create_service(@client, session, name, server, port)
      return service if check_connection?(session, service, ssl_context: ssl_context)
    end

    nil
  end
end