RxSwift 对基于 AFNetworking 的网络请求组件的适配，在掌握两个框架的基础上，进一步提升工作效率

2023-11-03 14:29:26

作为 iOS 开发者，我们经常需要处理网络请求相关的工作。为了简化这一过程，我们可以使用一些网络请求框架，比如 AFNetworking。而 RxSwift 则是另一个流行的框架，它可以帮助我们以响应式的方式处理事件流。

RxSwift 和 AFNetworking 都非常强大，但它们的使用方式却截然不同。AFNetworking 是一个基于委托的框架，这意味着我们需要手动处理网络请求的回调。而 RxSwift 则是一个基于观察者的框架，它允许我们以更简洁的方式处理网络请求。

为了充分利用这两个框架的优势，我们可以将它们结合起来使用。具体来说，我们可以使用 RxSwift 来对 AFNetworking 的网络请求组件进行适配，以便以更简洁的方式处理网络请求。

步骤 1：将原本的网络封装请求类改造

原本的网络请求基类（IOPRequestManager）是返回为 void 的，改为返回 NSURLSessionDataTask 类型即可：

// IOPRequestManager.swift

import Alamofire
import RxSwift

class IOPRequestManager: NSObject {
    static let shared = IOPRequestManager()
    
    func request(_ url: String, method: HTTPMethod, parameters: [String: Any]?, encoding: ParameterEncoding = URLEncoding.default) -> Observable<Data> {
        return Observable.create { observer -> Disposable in
            let request = Alamofire.request(url, method: method, parameters: parameters, encoding: encoding).responseData { response in
                switch response.result {
                case .success(let data):
                    observer.onNext(data)
                    observer.onCompleted()
                case .failure(let error):
                    observer.onError(error)
                }
            }
            
            return Disposables.create {
                request.cancel()
            }
        }
    }
}

步骤 2：做简单二次封装

如下扩展：

// IOPRequestManager+Rx.swift

import RxSwift

extension IOPRequestManager: ReactiveCompatible {}

extension Reactive where Base: IOPRequestManager {
    func request(_ url: String, method: HTTPMethod, parameters: [String: Any]?, encoding: ParameterEncoding = URLEncoding.default) -> Observable<Data> {
        return base.request(url, method: method, parameters: parameters, encoding: encoding)
    }
}

现在，我们可以使用 RxSwift 来发送网络请求，并以响应式的方式处理网络请求的响应。例如：

// ViewController.swift

import RxSwift
import RxCocoa

class ViewController: UIViewController {
    let disposeBag = DisposeBag()
    
    override func viewDidLoad() {
        super.viewDidLoad()
        
        // 发送网络请求
        let request = IOPRequestManager.shared.rx.request("https://api.example.com/v1/users", method: .get)
        
        // 处理网络请求的响应
        request.subscribe(onNext: { data in
            // 处理数据
        }, onError: { error in
            // 处理错误
        }).disposed(by: disposeBag)
    }
}