class Async::HTTP::Client
def self.open(*arguments, **options, &block)
def self.open(*arguments, **options, &block) client = self.new(*arguments, **options) return client unless block_given? begin yield client ensure client.close end end
def call(request)
def call(request) request.scheme ||= self.scheme request.authority ||= self.authority attempt = 0 # We may retry the request if it is possible to do so. https://tools.ietf.org/html/draft-nottingham-httpbis-retry-01 is a good guide for how retrying requests should work. begin attempt += 1 # As we cache pool, it's possible these pool go bad (e.g. closed by remote host). In this case, we need to try again. It's up to the caller to impose a timeout on this. If this is the last attempt, we force a new connection. connection = @pool.acquire response = request.call(connection) # The connection won't be released until the body is completely read/released. ::Protocol::HTTP::Body::Streamable.wrap(response) do @pool.release(connection) end return response rescue Protocol::RequestFailed # This is a specific case where the entire request wasn't sent before a failure occurred. So, we can even resend non-idempotent requests. @pool.release(connection) if connection if attempt < @retries retry else raise end rescue @pool.release(connection) if connection if request.idempotent? and attempt < @retries retry else raise end end end
def close
def close while @pool.busy? Async.logger.warn(self) {"Waiting for pool to drain: #{@pool}"} @pool.wait end @pool.close end
def initialize(endpoint, protocol = endpoint.protocol, scheme = endpoint.scheme, authority = endpoint.authority, retries: DEFAULT_RETRIES, connection_limit: DEFAULT_CONNECTION_LIMIT)
-
authority
(String
) -- The default authority to set to requests. -
scheme
(String
) -- The default scheme to set to requests. -
protocol
(Protocol::HTTP1 | Protocol::HTTP2 | Protocol::HTTPS
) -- the protocol to use. -
endpoint
(Endpoint
) -- the endpoint to connnect to.
def initialize(endpoint, protocol = endpoint.protocol, scheme = endpoint.scheme, authority = endpoint.authority, retries: DEFAULT_RETRIES, connection_limit: DEFAULT_CONNECTION_LIMIT) @endpoint = endpoint @protocol = protocol @retries = retries @pool = make_pool(connection_limit) @scheme = scheme @authority = authority end
def make_pool(connection_limit)
def make_pool(connection_limit) Async::Pool::Controller.wrap(limit: connection_limit) do Async.logger.debug(self) {"Making connection to #{@endpoint.inspect}"} @protocol.client(@endpoint.connect) end end
def secure?
def secure? @endpoint.secure? end