Alamofire：如何全局处理错误

Z时代
2024-01-10
分类：问答

我的问题与这一问题非常相似，但对于Alamofire：AFNetworking：全局处理错误并重复请求

如何能够全局捕获错误（通常为401）并在发出其他请求之前进行处理（如果不进行管理则最终失败）？

我当时想链接一个自定义响应处理程序，但这对应用程序的每个请求都很愚蠢。

也许是子类化，但是我应该子类化处理该类？

回答：

鉴于NSURLSession的并行性质，在oauth流中处理401响应的刷新非常复杂。我花了很多时间来建立对我们非常有效的内部解决方案。以下是对如何实现它的一般概念的非常高级的提取。

import Foundation
import Alamofire
public class AuthorizationManager: Manager {
    public typealias NetworkSuccessHandler = (AnyObject?) -> Void
    public typealias NetworkFailureHandler = (NSHTTPURLResponse?, AnyObject?, NSError) -> Void
    private typealias CachedTask = (NSHTTPURLResponse?, AnyObject?, NSError?) -> Void
    private var cachedTasks = Array<CachedTask>()
    private var isRefreshing = false
    public func startRequest(
        method method: Alamofire.Method,
        URLString: URLStringConvertible,
        parameters: [String: AnyObject]?,
        encoding: ParameterEncoding,
        success: NetworkSuccessHandler?,
        failure: NetworkFailureHandler?) -> Request?
    {
        let cachedTask: CachedTask = { [weak self] URLResponse, data, error in
            guard let spanSelf = self else { return }
            if let error = error {
                failure?(URLResponse, data, error)
            } else {
                spanSelf.startRequest(
                    method: method,
                    URLString: URLString,
                    parameters: parameters,
                    encoding: encoding,
                    success: success,
                    failure: failure
                )
            }
        }
        if self.isRefreshing {
            self.cachedTasks.append(cachedTask)
            return nil
        }
        // Append your auth tokens here to your parameters
        let request = self.request(method, URLString, parameters: parameters, encoding: encoding)
        request.response { [weak self] request, response, data, error in
            guard let spanSelf = self else { return }
            if let response = response where response.statusCode == 401 {
                spanSelf.cachedTasks.append(cachedTask)
                spanSelf.refreshTokens()
                return
            }
            if let error = error {
                failure?(response, data, error)
            } else {
                success?(data)
            }
        }
        return request
    }
    func refreshTokens() {
        self.isRefreshing = true
        // Make the refresh call and run the following in the success closure to restart the cached tasks
        let cachedTaskCopy = self.cachedTasks
        self.cachedTasks.removeAll()
        cachedTaskCopy.map { $0(nil, nil, nil) }
        self.isRefreshing = false
    }
}