mirror of
https://github.com/swift-server/async-http-client.git
synced 2026-06-02 07:37:34 +00:00
986dc47c11
Users should be able to use swift-configuration to create the http client configuration object Changes - duplicate package.swift to create a separate version for 6.0 and 6.1, since Configuration is 6.2+ - add helper to create http client configuration using ConfigReader
141 lines
7.0 KiB
Swift
141 lines
7.0 KiB
Swift
//===----------------------------------------------------------------------===//
|
|
//
|
|
// This source file is part of the AsyncHTTPClient open source project
|
|
//
|
|
// Copyright (c) 2026 Apple Inc. and the AsyncHTTPClient project authors
|
|
// Licensed under Apache License v2.0
|
|
//
|
|
// See LICENSE.txt for license information
|
|
// See CONTRIBUTORS.txt for the list of AsyncHTTPClient project authors
|
|
//
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#if compiler(>=6.2)
|
|
import Configuration
|
|
import NIOCore
|
|
|
|
@available(macOS 15.0, iOS 18.0, watchOS 11.0, tvOS 18.0, visionOS 2.0, *)
|
|
extension HTTPClient.Configuration {
|
|
/// Initializes HTTPClient configuration from a ConfigReader.
|
|
///
|
|
/// ## Configuration keys:
|
|
/// - `dnsOverrides` (string array, optional): Colon-separated host:IP pairs for DNS overrides (e.g., "localhost:127.0.0.1").
|
|
/// - `redirect` (scoped): Redirect handling configuration read by ``RedirectConfiguration/init(configReader:)``.
|
|
/// - `timeout` (scoped): Timeout configuration read by ``Timeout/init(configReader:)``.
|
|
/// - `connectionPool` (scoped): Connection pool configuration read by ``ConnectionPool/init(configReader:)``.
|
|
/// - `httpVersion` (string, optional, default: automatic): HTTP version to use ( "automatic" or "http1Only").
|
|
/// - `maximumUsesPerConnection` (int, optional, default: nil, no limit): Maximum uses per connection.
|
|
///
|
|
/// - Throws: `HTTPClientError.invalidRedirectConfiguration` if redirect mode is invalid.
|
|
/// - Throws: `HTTPClientError.invalidHTTPVersionConfiguration` if httpVersion is specified but invalid.
|
|
public init(configReader: ConfigReader) throws {
|
|
self.init()
|
|
|
|
// Each entry in the list should be a colon separated pair e.g. localhost:127.0.0.1 or localhost:::1
|
|
if let dnsOverridesList = configReader.stringArray(forKey: "dnsOverrides") {
|
|
for entry in dnsOverridesList {
|
|
guard let separatorIndex = entry.firstIndex(of: ":") else {
|
|
throw HTTPClientError.invalidDNSOverridesConfiguration
|
|
}
|
|
let key = entry.prefix(upTo: separatorIndex)
|
|
let value = entry.suffix(from: entry.index(after: separatorIndex))
|
|
if key.isEmpty || value.isEmpty {
|
|
throw HTTPClientError.invalidDNSOverridesConfiguration
|
|
}
|
|
self.dnsOverride[String(key)] = String(value.filter { !$0.isWhitespace })
|
|
}
|
|
}
|
|
|
|
self.redirectConfiguration = try .init(configReader: configReader.scoped(to: "redirect"))
|
|
self.timeout = .init(configReader: configReader.scoped(to: "timeout"))
|
|
self.connectionPool = .init(configReader: configReader.scoped(to: "connectionPool"))
|
|
if let version = try HTTPVersion(configReader: configReader) {
|
|
self.httpVersion = version
|
|
}
|
|
self.maximumUsesPerConnection = configReader.int(forKey: "maximumUsesPerConnection")
|
|
}
|
|
}
|
|
|
|
@available(macOS 15.0, iOS 18.0, watchOS 11.0, tvOS 18.0, visionOS 2.0, *)
|
|
extension HTTPClient.Configuration.RedirectConfiguration {
|
|
/// Initializes redirect configuration from a ConfigReader.
|
|
///
|
|
/// ## Configuration keys:
|
|
/// - `mode` (string, optional, default: "follow"): Redirect handling mode ("follow" or "disallow").
|
|
/// - `maxRedirects` (int, optional, default: 5): Maximum allowed redirects when mode is "follow".
|
|
/// - `allowCycles` (bool, optional, default: false): Allow cyclic redirects when mode is "follow".
|
|
///
|
|
/// - Throws: `HTTPClientError.invalidRedirectConfiguration` if mode is specified but invalid.
|
|
public init(configReader: ConfigReader) throws {
|
|
guard let mode = configReader.string(forKey: "mode") else {
|
|
// default
|
|
self = .follow(max: 5, allowCycles: false)
|
|
return
|
|
}
|
|
if mode == "follow" {
|
|
let maxRedirects = configReader.int(forKey: "maxRedirects", default: 5)
|
|
let allowCycles = configReader.bool(forKey: "allowCycles", default: false)
|
|
self = .follow(max: maxRedirects, allowCycles: allowCycles)
|
|
} else if mode == "disallow" {
|
|
self = .disallow
|
|
} else {
|
|
throw HTTPClientError.invalidRedirectConfiguration
|
|
}
|
|
}
|
|
}
|
|
|
|
@available(macOS 15.0, iOS 18.0, watchOS 11.0, tvOS 18.0, visionOS 2.0, *)
|
|
extension HTTPClient.Configuration.Timeout {
|
|
/// Initializes timeout configuration from a ConfigReader.
|
|
///
|
|
/// ## Configuration keys:
|
|
/// - `connectionMs` (int, optional, default: 10,000): Connection timeout in milliseconds.
|
|
/// - `readMs` (int, optional): Read timeout in milliseconds.
|
|
/// - `writeMs` (int, optional): Write timeout in milliseconds.
|
|
public init(configReader: ConfigReader) {
|
|
self.init()
|
|
self.connect = configReader.int(forKey: "connectionMs").map { TimeAmount.milliseconds(Int64($0)) }
|
|
self.read = configReader.int(forKey: "readMs").map { TimeAmount.milliseconds(Int64($0)) }
|
|
self.write = configReader.int(forKey: "writeMs").map { TimeAmount.milliseconds(Int64($0)) }
|
|
}
|
|
}
|
|
|
|
@available(macOS 15.0, iOS 18.0, watchOS 11.0, tvOS 18.0, visionOS 2.0, *)
|
|
extension HTTPClient.Configuration.ConnectionPool {
|
|
/// Initializes connection pool configuration from a ConfigReader.
|
|
///
|
|
/// ## Configuration keys:
|
|
/// - `idleTimeoutMs` (int, optional, default: 60,000): Connection idle timeout in milliseconds.
|
|
/// - `concurrentHTTP1ConnectionsPerHostSoftLimit` (int, optional, default: 8): Soft limit for concurrent HTTP/1.1 connections per host.
|
|
/// - `retryConnectionEstablishment` (bool, optional, default: true): Retry failed connection establishment.
|
|
/// - `preWarmedHTTP1ConnectionCount` (int, optional, default: 0): Number of pre-warmed HTTP/1.1 connections per host.
|
|
public init(configReader: ConfigReader) {
|
|
self.init()
|
|
self.idleTimeout = TimeAmount.milliseconds(Int64(configReader.int(forKey: "idleTimeoutMs", default: 60000)))
|
|
self.concurrentHTTP1ConnectionsPerHostSoftLimit = configReader.int(
|
|
forKey: "concurrentHTTP1ConnectionsPerHostSoftLimit",
|
|
default: 8
|
|
)
|
|
self.retryConnectionEstablishment = configReader.bool(forKey: "retryConnectionEstablishment", default: true)
|
|
self.preWarmedHTTP1ConnectionCount = configReader.int(forKey: "preWarmedHTTP1ConnectionCount", default: 0)
|
|
}
|
|
}
|
|
|
|
@available(macOS 15.0, iOS 18.0, watchOS 11.0, tvOS 18.0, visionOS 2.0, *)
|
|
extension HTTPClient.Configuration.HTTPVersion {
|
|
fileprivate init?(configReader: ConfigReader) throws {
|
|
guard let rawValue = configReader.string(forKey: "httpVersion") else {
|
|
// Unspecified is not an error. It's an optional prop.
|
|
return nil
|
|
}
|
|
// Specified but invalid IS an error
|
|
guard let base = Self.Configuration(rawValue: rawValue) else {
|
|
throw HTTPClientError.invalidHTTPVersionConfiguration
|
|
}
|
|
self = .init(configuration: base)
|
|
}
|
|
}
|
|
#endif
|