主要通過擴展Request、DataRequest和DownloadRequest將服務器返回的響應序列化成我們想要的數據。
1. 輔助類型
1) DataResponseSerializerProtocol和DataResponseSerializer<Value>
// DataResponse序列化器協議，所有的DataResponse序列化器必須遵循這個協議
public protocol DataResponseSerializerProtocol {
// 序列化后的對象類型
associatedtype SerializedObject
// 這個閉包告訴我們如何序列化
var serializeResponse: (URLRequest?, HTTPURLResponse?, Data?, Error?) -> Result<SerializedObject> { get }
}
// 通用的DataResponse序列化器
public struct DataResponseSerializer<Value>: DataResponseSerializerProtocol {
// 遵循DataResponseSerializerProtocol協議
public typealias SerializedObject = Value
// 遵循DataResponseSerializerProtocol協議
public var serializeResponse: (URLRequest?, HTTPURLResponse?, Data?, Error?) -> Result<Value>
// 初始化方法
public init(serializeResponse: @escaping (URLRequest?, HTTPURLResponse?, Data?, Error?) -> Result<Value>) {
self.serializeResponse = serializeResponse
}
}
2) DownloadResponseSerializerProtocol和DownloadResponseSerializer<Value>
與上面的兩個類似，大家對照遠嗎自己看下。
2. Timeline屬性
通過擴展為Request添加了timeline屬性。
extension Request {
var timeline: Timeline {
let requestStartTime = self.startTime ?? CFAbsoluteTimeGetCurrent()
let requestCompletedTime = self.endTime ?? CFAbsoluteTimeGetCurrent()
let initialResponseTime = self.delegate.initialResponseTime ?? requestCompletedTime
return Timeline(
requestStartTime: requestStartTime,
initialResponseTime: initialResponseTime,
requestCompletedTime: requestCompletedTime,
serializationCompletedTime: CFAbsoluteTimeGetCurrent()
)
}
}
3. 為DataRequest添加默認的序列化方法
這兩個方法都沒有將響應結果序列化成一個具體的類型。
// 這個方法沒有把響應結果序列化成具體的類型
//
// queue：指定completionHandler在哪個隊列執行，默認在主隊列
// 把completionHandler添加到delegate的隊列，請求完成后執行隊列
// 在completionHandler中返回默認的數據響應DefaultDataResponse（沒有經過序列化的）
@discardableResult
public func response(queue: DispatchQueue? = nil, completionHandler: @escaping (DefaultDataResponse) -> Void) -> Self {
delegate.queue.addOperation {
(queue ?? DispatchQueue.main).async {
var dataResponse = DefaultDataResponse(
request: self.request,
response: self.response,
data: self.delegate.data,
error: self.delegate.error,
timeline: self.timeline
)
dataResponse.add(self.delegate.metrics)
completionHandler(dataResponse)
}
}
return self
}
// 這個方法需要一個序列化器參數，會把響應結果序列化成序列化器需要的具體類型
//
// queue：指定completionHandler在哪個隊列執行，默認在主隊列
// 把completionHandler添加到delegate的隊列，請求完成后執行隊列
// 在completionHandler中返回默認的數據響應DefaultDataResponse（經過序列化的）
@discardableResult
public func response<T: DataResponseSerializerProtocol>(
queue: DispatchQueue? = nil,
responseSerializer: T,
completionHandler: @escaping (DataResponse<T.SerializedObject>) -> Void)
-> Self
{
delegate.queue.addOperation {
let result = responseSerializer.serializeResponse(
self.request,
self.response,
self.delegate.data,
self.delegate.error
)
var dataResponse = DataResponse<T.SerializedObject>(
request: self.request,
response: self.response,
data: self.delegate.data,
result: result,
timeline: self.timeline
)
dataResponse.add(self.delegate.metrics)
(queue ?? DispatchQueue.main).async { completionHandler(dataResponse) }
}
return self
}
4. 為DownloadRequest添加默認的序列化方法
這兩個方法跟上面的兩個類似，不做過多說明。
@discardableResult
public func response(
queue: DispatchQueue? = nil,
completionHandler: @escaping (DefaultDownloadResponse) -> Void)
-> Self
{
// ...
}
@discardableResult
public func response<T: DownloadResponseSerializerProtocol>(
queue: DispatchQueue? = nil,
responseSerializer: T,
completionHandler: @escaping (DownloadResponse<T.SerializedObject>) -> Void)
-> Self
{
// ...
}
5. 把響應序列化成Data
extension Request {
// 返回序列化后的結果
public static func serializeResponseData(response: HTTPURLResponse?, data: Data?, error: Error?) -> Result<Data> {
guard error == nil else { return .failure(error!) }
// 如果返回的狀態碼是204(沒有數據返回)和205(重置數據)，返回空數據
if let response = response, emptyDataStatusCodes.contains(response.statusCode) { return .success(Data()) }
guard let validData = data else {
return .failure(AFError.responseSerializationFailed(reason: .inputDataNil))
}
return .success(validData)
}
}
extension DataRequest {
// 創建一個數據響應序列化器，把響應序列化成Data類型
public static func dataResponseSerializer() -> DataResponseSerializer<Data> {
// 這里其實是初始化DataResponseSerializer
// 本來是DataResponseSerializer(serializeResponse: closure)，
// 因為closure是語句中的最后一個closure，所以可以把小括號和serializeResponse省略
return DataResponseSerializer { _, response, data, error in
return Request.serializeResponseData(response: response, data: data, error: error)
}
}
// 使用上面創建的數據響應序列化器，把響應序列化成Data
@discardableResult
public func responseData(
queue: DispatchQueue? = nil,
completionHandler: @escaping (DataResponse<Data>) -> Void)
-> Self
{
return response(
queue: queue,
responseSerializer: DataRequest.dataResponseSerializer(),
completionHandler: completionHandler
)
}
}
extension DownloadRequest {
// 創建一個下載響應序列化器，把響應序列化成Data類型
public static func dataResponseSerializer() -> DownloadResponseSerializer<Data> {
return DownloadResponseSerializer { _, response, fileURL, error in
guard error == nil else { return .failure(error!) }
guard let fileURL = fileURL else {
return .failure(AFError.responseSerializationFailed(reason: .inputFileNil))
}
do {
let data = try Data(contentsOf: fileURL)
return Request.serializeResponseData(response: response, data: data, error: error)
} catch {
return .failure(AFError.responseSerializationFailed(reason: .inputFileReadFailed(at: fileURL)))
}
}
}
// 使用上面創建的下載響應序列化器，把響應序列化成Data
@discardableResult
public func responseData(
queue: DispatchQueue? = nil,
completionHandler: @escaping (DownloadResponse<Data>) -> Void)
-> Self
{
return response(
queue: queue,
responseSerializer: DownloadRequest.dataResponseSerializer(),
completionHandler: completionHandler
)
}
}
5. 把響應序列化成String & JSON & Property List
把響應序列化成這三種類型的思路，跟序列化成Data的思路完全一致，大家可以自己看下，如果有不懂的，歡迎留言，我會盡力解答。
有任何問題，歡迎大家留言！
歡迎加入我管理的Swift開發群：536353151，本群只討論Swift相關內容。
原創文章，轉載請注明出處。謝謝！
