Alamofire源码学习之旅：使用RequestInterceptor拦截请求，让网络访问更稳健

IOS

2023-09-03 17:41:07

RequestInterceptor：网络请求的守护神

在网络请求的世界中，请求拦截器就好比一位经验丰富的守门员，它能够在请求发出之前或收到响应之后对其进行拦截，并根据需要进行处理。例如，您可以使用请求拦截器来：

在请求中添加或修改请求头信息
检查请求参数的合法性，防止不必要的请求发出
在请求中加入额外的安全信息，增强安全性
在请求失败时自动进行重试，提高网络访问的容错性

实现自己的请求拦截器

Alamofire提供了RequestInterceptor协议，使您能够轻松实现自己的请求拦截器。该协议定义了两个方法：

requestAdapter(URLRequest, Session) -> URLRequest：在请求发出之前调用，您可以通过修改参数来对请求进行调整。
retryPolicy(for: Error, Session) -> RetryPolicy：在请求失败时调用，您可以指定重试策略来决定是否重试请求。

实战：打造稳健的网络访问

下面是一个使用请求拦截器来实现自动重试的例子：

struct RetryInterceptor: RequestInterceptor {
    func requestAdapter(_ request: URLRequest, session: Session) -> URLRequest {
        return request
    }

    func retryPolicy(for error: Error, session: Session) -> RetryPolicy {
        if let response = error.asAFError?.response {
            if response.statusCode == 401 {
                // 401 Unauthorized，说明token过期，需要重新登录
                return .retryWithDelay(0.5)
            }
        }

        // 其他错误，不重试
        return .doNotRetry
    }
}

然后，您可以在SessionConfiguration中注册请求拦截器：

let configuration = URLSessionConfiguration.default
configuration.httpAdditionalHeaders = ["Content-Type": "application/json"]
configuration.requestInterceptors = [RetryInterceptor()]

这样，当网络请求失败时，请求拦截器会自动根据错误信息进行重试，提高网络访问的容错性。