可以使用以下代码示例来解决这个问题:
let sessionManager = Session.default
sessionManager.adapter = AccessTokenAdapter()
sessionManager.interceptor = Interceptor(retryPolicy: .retry, adapters: [BearerTokenInterceptor()])
class BearerTokenInterceptor: RequestInterceptor {
func adapt(_ urlRequest: URLRequest, for session: Session, completion: @escaping (Result) -> Void) {
// Check if the request needs to be authenticated
guard let urlString = urlRequest.url?.absoluteString else {
return completion(.failure(AFError.parameterEncodingFailed(reason: .missingURL)))
}
let urlStringWithoutParameters = urlString.components(separatedBy: "?").first ?? urlString
if !urlStringWithoutParameters.contains("/api/") {
// This is a public request - no need to add authentication headers
return completion(.success(urlRequest))
}
// This is a private request - add authentication headers
guard let accessToken = AuthManager.shared.accessToken else {
return completion(.failure(AFError.sessionDeinitialized))
}
var updatedRequest = urlRequest
updatedRequest.setValue("Bearer \(accessToken)", forHTTPHeaderField: "Authorization")
completion(.success(updatedRequest))
}
}
class AccessTokenAdapter: RequestAdapter {
func adapt(_ urlRequest: URLRequest, for session: Session, completion: @escaping (Result) -> Void) {
guard let accessToken = AuthManager.shared.accessToken else { return completion(.success(urlRequest)) }
var urlRequest = urlRequest
urlRequest.setValue("Bearer \(accessToken)", forHTTPHeaderField: "Authorization")
completion(.success(urlRequest))
}
}
上面的代码定义了一个 Interceptor
对象,包含了一个 BearerTokenInterceptor
对象和一个重试策略。BearerTokenInterceptor
实现了 RequestInterceptor
协议,它的 adapt
方法在检查请求是否需要身份验证之后,要么添加身份验证头部,要么返回未更改的请求对象。AccessTokenAdapter
实现了 RequestAdapter
协议,用于设置访问令牌头部。在这里,我们将 `