Response.swift 19 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399
  1. //
  2. // Response.swift
  3. //
  4. // Copyright (c) 2014-2018 Alamofire Software Foundation (http://alamofire.org/)
  5. //
  6. // Permission is hereby granted, free of charge, to any person obtaining a copy
  7. // of this software and associated documentation files (the "Software"), to deal
  8. // in the Software without restriction, including without limitation the rights
  9. // to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
  10. // copies of the Software, and to permit persons to whom the Software is
  11. // furnished to do so, subject to the following conditions:
  12. //
  13. // The above copyright notice and this permission notice shall be included in
  14. // all copies or substantial portions of the Software.
  15. //
  16. // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  17. // IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  18. // FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  19. // AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  20. // LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  21. // OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
  22. // THE SOFTWARE.
  23. //
  24. import Foundation
  25. /// Default type of `DataResponse` returned by Alamofire, with an `AFError` `Failure` type.
  26. public typealias AFDataResponse<Success> = DataResponse<Success, AFError>
  27. /// Default type of `DownloadResponse` returned by Alamofire, with an `AFError` `Failure` type.
  28. public typealias AFDownloadResponse<Success> = DownloadResponse<Success, AFError>
  29. /// Type used to store all values associated with a serialized response of a `DataRequest` or `UploadRequest`.
  30. public struct DataResponse<Success, Failure: Error> {
  31. /// The URL request sent to the server.
  32. public let request: URLRequest?
  33. /// The server's response to the URL request.
  34. public let response: HTTPURLResponse?
  35. /// The data returned by the server.
  36. public let data: Data?
  37. /// The final metrics of the response.
  38. public let metrics: URLSessionTaskMetrics?
  39. /// The time taken to serialize the response.
  40. public let serializationDuration: TimeInterval
  41. /// The result of response serialization.
  42. public let result: Result<Success, Failure>
  43. /// Returns the associated value of the result if it is a success, `nil` otherwise.
  44. public var value: Success? { return result.success }
  45. /// Returns the associated error value if the result if it is a failure, `nil` otherwise.
  46. public var error: Failure? { return result.failure }
  47. /// Creates a `DataResponse` instance with the specified parameters derived from the response serialization.
  48. ///
  49. /// - Parameters:
  50. /// - request: The `URLRequest` sent to the server.
  51. /// - response: The `HTTPURLResponse` from the server.
  52. /// - data: The `Data` returned by the server.
  53. /// - metrics: The `URLSessionTaskMetrics` of the `DataRequest` or `UploadRequest`.
  54. /// - serializationDuration: The duration taken by serialization.
  55. /// - result: The `Result` of response serialization.
  56. public init(request: URLRequest?,
  57. response: HTTPURLResponse?,
  58. data: Data?,
  59. metrics: URLSessionTaskMetrics?,
  60. serializationDuration: TimeInterval,
  61. result: Result<Success, Failure>) {
  62. self.request = request
  63. self.response = response
  64. self.data = data
  65. self.metrics = metrics
  66. self.serializationDuration = serializationDuration
  67. self.result = result
  68. }
  69. }
  70. // MARK: -
  71. extension DataResponse: CustomStringConvertible, CustomDebugStringConvertible {
  72. /// The textual representation used when written to an output stream, which includes whether the result was a
  73. /// success or failure.
  74. public var description: String {
  75. return "\(result)"
  76. }
  77. /// The debug textual representation used when written to an output stream, which includes the URL request, the URL
  78. /// response, the server data, the duration of the network and serialization actions, and the response serialization
  79. /// result.
  80. public var debugDescription: String {
  81. let requestDescription = request.map { "\($0.httpMethod!) \($0)" } ?? "nil"
  82. let requestBody = request?.httpBody.map { String(decoding: $0, as: UTF8.self) } ?? "None"
  83. let responseDescription = response.map { response in
  84. let sortedHeaders = response.headers.sorted()
  85. return """
  86. [Status Code]: \(response.statusCode)
  87. [Headers]:
  88. \(sortedHeaders)
  89. """
  90. } ?? "nil"
  91. let responseBody = data.map { String(decoding: $0, as: UTF8.self) } ?? "None"
  92. let metricsDescription = metrics.map { "\($0.taskInterval.duration)s" } ?? "None"
  93. return """
  94. [Request]: \(requestDescription)
  95. [Request Body]: \n\(requestBody)
  96. [Response]: \n\(responseDescription)
  97. [Response Body]: \n\(responseBody)
  98. [Data]: \(data?.description ?? "None")
  99. [Network Duration]: \(metricsDescription)
  100. [Serialization Duration]: \(serializationDuration)s
  101. [Result]: \(result)
  102. """
  103. }
  104. }
  105. // MARK: -
  106. extension DataResponse {
  107. /// Evaluates the specified closure when the result of this `DataResponse` is a success, passing the unwrapped
  108. /// result value as a parameter.
  109. ///
  110. /// Use the `map` method with a closure that does not throw. For example:
  111. ///
  112. /// let possibleData: DataResponse<Data> = ...
  113. /// let possibleInt = possibleData.map { $0.count }
  114. ///
  115. /// - parameter transform: A closure that takes the success value of the instance's result.
  116. ///
  117. /// - returns: A `DataResponse` whose result wraps the value returned by the given closure. If this instance's
  118. /// result is a failure, returns a response wrapping the same failure.
  119. public func map<NewSuccess>(_ transform: (Success) -> NewSuccess) -> DataResponse<NewSuccess, Failure> {
  120. return DataResponse<NewSuccess, Failure>(request: request,
  121. response: response,
  122. data: data,
  123. metrics: metrics,
  124. serializationDuration: serializationDuration,
  125. result: result.map(transform))
  126. }
  127. /// Evaluates the given closure when the result of this `DataResponse` is a success, passing the unwrapped result
  128. /// value as a parameter.
  129. ///
  130. /// Use the `tryMap` method with a closure that may throw an error. For example:
  131. ///
  132. /// let possibleData: DataResponse<Data> = ...
  133. /// let possibleObject = possibleData.tryMap {
  134. /// try JSONSerialization.jsonObject(with: $0)
  135. /// }
  136. ///
  137. /// - parameter transform: A closure that takes the success value of the instance's result.
  138. ///
  139. /// - returns: A success or failure `DataResponse` depending on the result of the given closure. If this instance's
  140. /// result is a failure, returns the same failure.
  141. public func tryMap<NewSuccess>(_ transform: (Success) throws -> NewSuccess) -> DataResponse<NewSuccess, Error> {
  142. return DataResponse<NewSuccess, Error>(request: request,
  143. response: response,
  144. data: data,
  145. metrics: metrics,
  146. serializationDuration: serializationDuration,
  147. result: result.tryMap(transform))
  148. }
  149. /// Evaluates the specified closure when the `DataResponse` is a failure, passing the unwrapped error as a parameter.
  150. ///
  151. /// Use the `mapError` function with a closure that does not throw. For example:
  152. ///
  153. /// let possibleData: DataResponse<Data> = ...
  154. /// let withMyError = possibleData.mapError { MyError.error($0) }
  155. ///
  156. /// - Parameter transform: A closure that takes the error of the instance.
  157. ///
  158. /// - Returns: A `DataResponse` instance containing the result of the transform.
  159. public func mapError<NewFailure: Error>(_ transform: (Failure) -> NewFailure) -> DataResponse<Success, NewFailure> {
  160. return DataResponse<Success, NewFailure>(request: request,
  161. response: response,
  162. data: data,
  163. metrics: metrics,
  164. serializationDuration: serializationDuration,
  165. result: result.mapError(transform))
  166. }
  167. /// Evaluates the specified closure when the `DataResponse` is a failure, passing the unwrapped error as a parameter.
  168. ///
  169. /// Use the `tryMapError` function with a closure that may throw an error. For example:
  170. ///
  171. /// let possibleData: DataResponse<Data> = ...
  172. /// let possibleObject = possibleData.tryMapError {
  173. /// try someFailableFunction(taking: $0)
  174. /// }
  175. ///
  176. /// - Parameter transform: A throwing closure that takes the error of the instance.
  177. ///
  178. /// - Returns: A `DataResponse` instance containing the result of the transform.
  179. public func tryMapError<NewFailure: Error>(_ transform: (Failure) throws -> NewFailure) -> DataResponse<Success, Error> {
  180. return DataResponse<Success, Error>(request: request,
  181. response: response,
  182. data: data,
  183. metrics: metrics,
  184. serializationDuration: serializationDuration,
  185. result: result.tryMapError(transform))
  186. }
  187. }
  188. // MARK: -
  189. /// Used to store all data associated with a serialized response of a download request.
  190. public struct DownloadResponse<Success, Failure: Error> {
  191. /// The URL request sent to the server.
  192. public let request: URLRequest?
  193. /// The server's response to the URL request.
  194. public let response: HTTPURLResponse?
  195. /// The final destination URL of the data returned from the server after it is moved.
  196. public let fileURL: URL?
  197. /// The resume data generated if the request was cancelled.
  198. public let resumeData: Data?
  199. /// The final metrics of the response.
  200. public let metrics: URLSessionTaskMetrics?
  201. /// The time taken to serialize the response.
  202. public let serializationDuration: TimeInterval
  203. /// The result of response serialization.
  204. public let result: Result<Success, Failure>
  205. /// Returns the associated value of the result if it is a success, `nil` otherwise.
  206. public var value: Success? { return result.success }
  207. /// Returns the associated error value if the result if it is a failure, `nil` otherwise.
  208. public var error: Failure? { return result.failure }
  209. /// Creates a `DownloadResponse` instance with the specified parameters derived from response serialization.
  210. ///
  211. /// - Parameters:
  212. /// - request: The `URLRequest` sent to the server.
  213. /// - response: The `HTTPURLResponse` from the server.
  214. /// - temporaryURL: The temporary destination `URL` of the data returned from the server.
  215. /// - destinationURL: The final destination `URL` of the data returned from the server, if it was moved.
  216. /// - resumeData: The resume `Data` generated if the request was cancelled.
  217. /// - metrics: The `URLSessionTaskMetrics` of the `DownloadRequest`.
  218. /// - serializationDuration: The duration taken by serialization.
  219. /// - result: The `Result` of response serialization.
  220. public init(request: URLRequest?,
  221. response: HTTPURLResponse?,
  222. fileURL: URL?,
  223. resumeData: Data?,
  224. metrics: URLSessionTaskMetrics?,
  225. serializationDuration: TimeInterval,
  226. result: Result<Success, Failure>) {
  227. self.request = request
  228. self.response = response
  229. self.fileURL = fileURL
  230. self.resumeData = resumeData
  231. self.metrics = metrics
  232. self.serializationDuration = serializationDuration
  233. self.result = result
  234. }
  235. }
  236. // MARK: -
  237. extension DownloadResponse: CustomStringConvertible, CustomDebugStringConvertible {
  238. /// The textual representation used when written to an output stream, which includes whether the result was a
  239. /// success or failure.
  240. public var description: String {
  241. return "\(result)"
  242. }
  243. /// The debug textual representation used when written to an output stream, which includes the URL request, the URL
  244. /// response, the temporary and destination URLs, the resume data, the durations of the network and serialization
  245. /// actions, and the response serialization result.
  246. public var debugDescription: String {
  247. let requestDescription = request.map { "\($0.httpMethod!) \($0)" } ?? "nil"
  248. let requestBody = request?.httpBody.map { String(decoding: $0, as: UTF8.self) } ?? "None"
  249. let responseDescription = response.map { response in
  250. let sortedHeaders = response.headers.sorted()
  251. return """
  252. [Status Code]: \(response.statusCode)
  253. [Headers]:
  254. \(sortedHeaders)
  255. """
  256. } ?? "nil"
  257. let metricsDescription = metrics.map { "\($0.taskInterval.duration)s" } ?? "None"
  258. let resumeDataDescription = resumeData.map { "\($0)" } ?? "None"
  259. return """
  260. [Request]: \(requestDescription)
  261. [Request Body]: \n\(requestBody)
  262. [Response]: \n\(responseDescription)
  263. [File URL]: \(fileURL?.path ?? "nil")
  264. [ResumeData]: \(resumeDataDescription)
  265. [Network Duration]: \(metricsDescription)
  266. [Serialization Duration]: \(serializationDuration)s
  267. [Result]: \(result)
  268. """
  269. }
  270. }
  271. // MARK: -
  272. extension DownloadResponse {
  273. /// Evaluates the given closure when the result of this `DownloadResponse` is a success, passing the unwrapped
  274. /// result value as a parameter.
  275. ///
  276. /// Use the `map` method with a closure that does not throw. For example:
  277. ///
  278. /// let possibleData: DownloadResponse<Data> = ...
  279. /// let possibleInt = possibleData.map { $0.count }
  280. ///
  281. /// - parameter transform: A closure that takes the success value of the instance's result.
  282. ///
  283. /// - returns: A `DownloadResponse` whose result wraps the value returned by the given closure. If this instance's
  284. /// result is a failure, returns a response wrapping the same failure.
  285. public func map<NewSuccess>(_ transform: (Success) -> NewSuccess) -> DownloadResponse<NewSuccess, Failure> {
  286. return DownloadResponse<NewSuccess, Failure>(request: request,
  287. response: response,
  288. fileURL: fileURL,
  289. resumeData: resumeData,
  290. metrics: metrics,
  291. serializationDuration: serializationDuration,
  292. result: result.map(transform))
  293. }
  294. /// Evaluates the given closure when the result of this `DownloadResponse` is a success, passing the unwrapped
  295. /// result value as a parameter.
  296. ///
  297. /// Use the `tryMap` method with a closure that may throw an error. For example:
  298. ///
  299. /// let possibleData: DownloadResponse<Data> = ...
  300. /// let possibleObject = possibleData.tryMap {
  301. /// try JSONSerialization.jsonObject(with: $0)
  302. /// }
  303. ///
  304. /// - parameter transform: A closure that takes the success value of the instance's result.
  305. ///
  306. /// - returns: A success or failure `DownloadResponse` depending on the result of the given closure. If this
  307. /// instance's result is a failure, returns the same failure.
  308. public func tryMap<NewSuccess>(_ transform: (Success) throws -> NewSuccess) -> DownloadResponse<NewSuccess, Error> {
  309. return DownloadResponse<NewSuccess, Error>(request: request,
  310. response: response,
  311. fileURL: fileURL,
  312. resumeData: resumeData,
  313. metrics: metrics,
  314. serializationDuration: serializationDuration,
  315. result: result.tryMap(transform))
  316. }
  317. /// Evaluates the specified closure when the `DownloadResponse` is a failure, passing the unwrapped error as a parameter.
  318. ///
  319. /// Use the `mapError` function with a closure that does not throw. For example:
  320. ///
  321. /// let possibleData: DownloadResponse<Data> = ...
  322. /// let withMyError = possibleData.mapError { MyError.error($0) }
  323. ///
  324. /// - Parameter transform: A closure that takes the error of the instance.
  325. ///
  326. /// - Returns: A `DownloadResponse` instance containing the result of the transform.
  327. public func mapError<NewFailure: Error>(_ transform: (Failure) -> NewFailure) -> DownloadResponse<Success, NewFailure> {
  328. return DownloadResponse<Success, NewFailure>(request: request,
  329. response: response,
  330. fileURL: fileURL,
  331. resumeData: resumeData,
  332. metrics: metrics,
  333. serializationDuration: serializationDuration,
  334. result: result.mapError(transform))
  335. }
  336. /// Evaluates the specified closure when the `DownloadResponse` is a failure, passing the unwrapped error as a parameter.
  337. ///
  338. /// Use the `tryMapError` function with a closure that may throw an error. For example:
  339. ///
  340. /// let possibleData: DownloadResponse<Data> = ...
  341. /// let possibleObject = possibleData.tryMapError {
  342. /// try someFailableFunction(taking: $0)
  343. /// }
  344. ///
  345. /// - Parameter transform: A throwing closure that takes the error of the instance.
  346. ///
  347. /// - Returns: A `DownloadResponse` instance containing the result of the transform.
  348. public func tryMapError<NewFailure: Error>(_ transform: (Failure) throws -> NewFailure) -> DownloadResponse<Success, Error> {
  349. return DownloadResponse<Success, Error>(request: request,
  350. response: response,
  351. fileURL: fileURL,
  352. resumeData: resumeData,
  353. metrics: metrics,
  354. serializationDuration: serializationDuration,
  355. result: result.tryMapError(transform))
  356. }
  357. }