1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258 |
- import Foundation
- open class Session {
-
- public static let `default` = Session()
-
-
-
-
-
-
- public let session: URLSession
-
- public let delegate: SessionDelegate
-
- public let rootQueue: DispatchQueue
-
- public let startRequestsImmediately: Bool
-
-
-
- public let requestQueue: DispatchQueue
-
-
-
- public let serializationQueue: DispatchQueue
-
-
- public let interceptor: RequestInterceptor?
-
- public let serverTrustManager: ServerTrustManager?
-
- public let redirectHandler: RedirectHandler?
-
- public let cachedResponseHandler: CachedResponseHandler?
-
- public let eventMonitor: CompositeEventMonitor
-
- public let defaultEventMonitors: [EventMonitor] = [AlamofireNotifications()]
-
- var requestTaskMap = RequestTaskMap()
-
- var activeRequests: Set<Request> = []
-
- var waitingCompletions: [URLSessionTask: () -> Void] = [:]
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
- public init(session: URLSession,
- delegate: SessionDelegate,
- rootQueue: DispatchQueue,
- startRequestsImmediately: Bool = true,
- requestQueue: DispatchQueue? = nil,
- serializationQueue: DispatchQueue? = nil,
- interceptor: RequestInterceptor? = nil,
- serverTrustManager: ServerTrustManager? = nil,
- redirectHandler: RedirectHandler? = nil,
- cachedResponseHandler: CachedResponseHandler? = nil,
- eventMonitors: [EventMonitor] = []) {
- precondition(session.configuration.identifier == nil,
- "Alamofire does not support background URLSessionConfigurations.")
- precondition(session.delegateQueue.underlyingQueue === rootQueue,
- "Session(session:) initializer must be passed the DispatchQueue used as the delegateQueue's underlyingQueue as rootQueue.")
- self.session = session
- self.delegate = delegate
- self.rootQueue = rootQueue
- self.startRequestsImmediately = startRequestsImmediately
- self.requestQueue = requestQueue ?? DispatchQueue(label: "\(rootQueue.label).requestQueue", target: rootQueue)
- self.serializationQueue = serializationQueue ?? DispatchQueue(label: "\(rootQueue.label).serializationQueue", target: rootQueue)
- self.interceptor = interceptor
- self.serverTrustManager = serverTrustManager
- self.redirectHandler = redirectHandler
- self.cachedResponseHandler = cachedResponseHandler
- eventMonitor = CompositeEventMonitor(monitors: defaultEventMonitors + eventMonitors)
- delegate.eventMonitor = eventMonitor
- delegate.stateProvider = self
- }
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
- public convenience init(configuration: URLSessionConfiguration = URLSessionConfiguration.af.default,
- delegate: SessionDelegate = SessionDelegate(),
- rootQueue: DispatchQueue = DispatchQueue(label: "org.alamofire.session.rootQueue"),
- startRequestsImmediately: Bool = true,
- requestQueue: DispatchQueue? = nil,
- serializationQueue: DispatchQueue? = nil,
- interceptor: RequestInterceptor? = nil,
- serverTrustManager: ServerTrustManager? = nil,
- redirectHandler: RedirectHandler? = nil,
- cachedResponseHandler: CachedResponseHandler? = nil,
- eventMonitors: [EventMonitor] = []) {
- precondition(configuration.identifier == nil, "Alamofire does not support background URLSessionConfigurations.")
- let delegateQueue = OperationQueue(maxConcurrentOperationCount: 1, underlyingQueue: rootQueue, name: "org.alamofire.session.sessionDelegateQueue")
- let session = URLSession(configuration: configuration, delegate: delegate, delegateQueue: delegateQueue)
- self.init(session: session,
- delegate: delegate,
- rootQueue: rootQueue,
- startRequestsImmediately: startRequestsImmediately,
- requestQueue: requestQueue,
- serializationQueue: serializationQueue,
- interceptor: interceptor,
- serverTrustManager: serverTrustManager,
- redirectHandler: redirectHandler,
- cachedResponseHandler: cachedResponseHandler,
- eventMonitors: eventMonitors)
- }
- deinit {
- finishRequestsForDeinit()
- session.invalidateAndCancel()
- }
-
-
-
-
-
-
-
-
-
-
- public func withAllRequests(perform action: @escaping (Set<Request>) -> Void) {
- rootQueue.async {
- action(self.activeRequests)
- }
- }
-
-
-
-
-
-
-
-
-
- public func cancelAllRequests(completingOnQueue queue: DispatchQueue = .main, completion: (() -> Void)? = nil) {
- withAllRequests { requests in
- requests.forEach { $0.cancel() }
- queue.async {
- completion?()
- }
- }
- }
-
-
- public typealias RequestModifier = (inout URLRequest) throws -> Void
- struct RequestConvertible: URLRequestConvertible {
- let url: URLConvertible
- let method: HTTPMethod
- let parameters: Parameters?
- let encoding: ParameterEncoding
- let headers: HTTPHeaders?
- let requestModifier: RequestModifier?
- func asURLRequest() throws -> URLRequest {
- var request = try URLRequest(url: url, method: method, headers: headers)
- try requestModifier?(&request)
- return try encoding.encode(request, with: parameters)
- }
- }
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
- open func request(_ convertible: URLConvertible,
- method: HTTPMethod = .get,
- parameters: Parameters? = nil,
- encoding: ParameterEncoding = URLEncoding.default,
- headers: HTTPHeaders? = nil,
- interceptor: RequestInterceptor? = nil,
- requestModifier: RequestModifier? = nil) -> DataRequest {
- let convertible = RequestConvertible(url: convertible,
- method: method,
- parameters: parameters,
- encoding: encoding,
- headers: headers,
- requestModifier: requestModifier)
- return request(convertible, interceptor: interceptor)
- }
- struct RequestEncodableConvertible<Parameters: Encodable>: URLRequestConvertible {
- let url: URLConvertible
- let method: HTTPMethod
- let parameters: Parameters?
- let encoder: ParameterEncoder
- let headers: HTTPHeaders?
- let requestModifier: RequestModifier?
- func asURLRequest() throws -> URLRequest {
- var request = try URLRequest(url: url, method: method, headers: headers)
- try requestModifier?(&request)
- return try parameters.map { try encoder.encode($0, into: request) } ?? request
- }
- }
-
-
-
-
-
-
-
-
-
-
-
-
-
- open func request<Parameters: Encodable>(_ convertible: URLConvertible,
- method: HTTPMethod = .get,
- parameters: Parameters? = nil,
- encoder: ParameterEncoder = URLEncodedFormParameterEncoder.default,
- headers: HTTPHeaders? = nil,
- interceptor: RequestInterceptor? = nil,
- requestModifier: RequestModifier? = nil) -> DataRequest {
- let convertible = RequestEncodableConvertible(url: convertible,
- method: method,
- parameters: parameters,
- encoder: encoder,
- headers: headers,
- requestModifier: requestModifier)
- return request(convertible, interceptor: interceptor)
- }
-
-
-
-
-
-
-
- open func request(_ convertible: URLRequestConvertible, interceptor: RequestInterceptor? = nil) -> DataRequest {
- let request = DataRequest(convertible: convertible,
- underlyingQueue: rootQueue,
- serializationQueue: serializationQueue,
- eventMonitor: eventMonitor,
- interceptor: interceptor,
- delegate: self)
- perform(request)
- return request
- }
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
- open func streamRequest<Parameters: Encodable>(_ convertible: URLConvertible,
- method: HTTPMethod = .get,
- parameters: Parameters? = nil,
- encoder: ParameterEncoder = URLEncodedFormParameterEncoder.default,
- headers: HTTPHeaders? = nil,
- automaticallyCancelOnStreamError: Bool = false,
- interceptor: RequestInterceptor? = nil,
- requestModifier: RequestModifier? = nil) -> DataStreamRequest {
- let convertible = RequestEncodableConvertible(url: convertible,
- method: method,
- parameters: parameters,
- encoder: encoder,
- headers: headers,
- requestModifier: requestModifier)
- return streamRequest(convertible,
- automaticallyCancelOnStreamError: automaticallyCancelOnStreamError,
- interceptor: interceptor)
- }
-
-
-
-
-
-
-
-
-
-
-
-
-
-
- open func streamRequest(_ convertible: URLConvertible,
- method: HTTPMethod = .get,
- headers: HTTPHeaders? = nil,
- automaticallyCancelOnStreamError: Bool = false,
- interceptor: RequestInterceptor? = nil,
- requestModifier: RequestModifier? = nil) -> DataStreamRequest {
- let convertible = RequestEncodableConvertible(url: convertible,
- method: method,
- parameters: Empty?.none,
- encoder: URLEncodedFormParameterEncoder.default,
- headers: headers,
- requestModifier: requestModifier)
- return streamRequest(convertible,
- automaticallyCancelOnStreamError: automaticallyCancelOnStreamError,
- interceptor: interceptor)
- }
-
-
-
-
-
-
-
-
-
-
- open func streamRequest(_ convertible: URLRequestConvertible,
- automaticallyCancelOnStreamError: Bool = false,
- interceptor: RequestInterceptor? = nil) -> DataStreamRequest {
- let request = DataStreamRequest(convertible: convertible,
- automaticallyCancelOnStreamError: automaticallyCancelOnStreamError,
- underlyingQueue: rootQueue,
- serializationQueue: serializationQueue,
- eventMonitor: eventMonitor,
- interceptor: interceptor,
- delegate: self)
- perform(request)
- return request
- }
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
- open func download(_ convertible: URLConvertible,
- method: HTTPMethod = .get,
- parameters: Parameters? = nil,
- encoding: ParameterEncoding = URLEncoding.default,
- headers: HTTPHeaders? = nil,
- interceptor: RequestInterceptor? = nil,
- requestModifier: RequestModifier? = nil,
- to destination: DownloadRequest.Destination? = nil) -> DownloadRequest {
- let convertible = RequestConvertible(url: convertible,
- method: method,
- parameters: parameters,
- encoding: encoding,
- headers: headers,
- requestModifier: requestModifier)
- return download(convertible, interceptor: interceptor, to: destination)
- }
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
- open func download<Parameters: Encodable>(_ convertible: URLConvertible,
- method: HTTPMethod = .get,
- parameters: Parameters? = nil,
- encoder: ParameterEncoder = URLEncodedFormParameterEncoder.default,
- headers: HTTPHeaders? = nil,
- interceptor: RequestInterceptor? = nil,
- requestModifier: RequestModifier? = nil,
- to destination: DownloadRequest.Destination? = nil) -> DownloadRequest {
- let convertible = RequestEncodableConvertible(url: convertible,
- method: method,
- parameters: parameters,
- encoder: encoder,
- headers: headers,
- requestModifier: requestModifier)
- return download(convertible, interceptor: interceptor, to: destination)
- }
-
-
-
-
-
-
-
-
-
- open func download(_ convertible: URLRequestConvertible,
- interceptor: RequestInterceptor? = nil,
- to destination: DownloadRequest.Destination? = nil) -> DownloadRequest {
- let request = DownloadRequest(downloadable: .request(convertible),
- underlyingQueue: rootQueue,
- serializationQueue: serializationQueue,
- eventMonitor: eventMonitor,
- interceptor: interceptor,
- delegate: self,
- destination: destination ?? DownloadRequest.defaultDestination)
- perform(request)
- return request
- }
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
- open func download(resumingWith data: Data,
- interceptor: RequestInterceptor? = nil,
- to destination: DownloadRequest.Destination? = nil) -> DownloadRequest {
- let request = DownloadRequest(downloadable: .resumeData(data),
- underlyingQueue: rootQueue,
- serializationQueue: serializationQueue,
- eventMonitor: eventMonitor,
- interceptor: interceptor,
- delegate: self,
- destination: destination ?? DownloadRequest.defaultDestination)
- perform(request)
- return request
- }
-
- struct ParameterlessRequestConvertible: URLRequestConvertible {
- let url: URLConvertible
- let method: HTTPMethod
- let headers: HTTPHeaders?
- let requestModifier: RequestModifier?
- func asURLRequest() throws -> URLRequest {
- var request = try URLRequest(url: url, method: method, headers: headers)
- try requestModifier?(&request)
- return request
- }
- }
- struct Upload: UploadConvertible {
- let request: URLRequestConvertible
- let uploadable: UploadableConvertible
- func createUploadable() throws -> UploadRequest.Uploadable {
- try uploadable.createUploadable()
- }
- func asURLRequest() throws -> URLRequest {
- try request.asURLRequest()
- }
- }
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
- open func upload(_ data: Data,
- to convertible: URLConvertible,
- method: HTTPMethod = .post,
- headers: HTTPHeaders? = nil,
- interceptor: RequestInterceptor? = nil,
- fileManager: FileManager = .default,
- requestModifier: RequestModifier? = nil) -> UploadRequest {
- let convertible = ParameterlessRequestConvertible(url: convertible,
- method: method,
- headers: headers,
- requestModifier: requestModifier)
- return upload(data, with: convertible, interceptor: interceptor, fileManager: fileManager)
- }
-
-
-
-
-
-
-
-
-
-
- open func upload(_ data: Data,
- with convertible: URLRequestConvertible,
- interceptor: RequestInterceptor? = nil,
- fileManager: FileManager = .default) -> UploadRequest {
- upload(.data(data), with: convertible, interceptor: interceptor, fileManager: fileManager)
- }
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
- open func upload(_ fileURL: URL,
- to convertible: URLConvertible,
- method: HTTPMethod = .post,
- headers: HTTPHeaders? = nil,
- interceptor: RequestInterceptor? = nil,
- fileManager: FileManager = .default,
- requestModifier: RequestModifier? = nil) -> UploadRequest {
- let convertible = ParameterlessRequestConvertible(url: convertible,
- method: method,
- headers: headers,
- requestModifier: requestModifier)
- return upload(fileURL, with: convertible, interceptor: interceptor, fileManager: fileManager)
- }
-
-
-
-
-
-
-
-
-
-
-
- open func upload(_ fileURL: URL,
- with convertible: URLRequestConvertible,
- interceptor: RequestInterceptor? = nil,
- fileManager: FileManager = .default) -> UploadRequest {
- upload(.file(fileURL, shouldRemove: false), with: convertible, interceptor: interceptor, fileManager: fileManager)
- }
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
- open func upload(_ stream: InputStream,
- to convertible: URLConvertible,
- method: HTTPMethod = .post,
- headers: HTTPHeaders? = nil,
- interceptor: RequestInterceptor? = nil,
- fileManager: FileManager = .default,
- requestModifier: RequestModifier? = nil) -> UploadRequest {
- let convertible = ParameterlessRequestConvertible(url: convertible,
- method: method,
- headers: headers,
- requestModifier: requestModifier)
- return upload(stream, with: convertible, interceptor: interceptor, fileManager: fileManager)
- }
-
-
-
-
-
-
-
-
-
-
-
- open func upload(_ stream: InputStream,
- with convertible: URLRequestConvertible,
- interceptor: RequestInterceptor? = nil,
- fileManager: FileManager = .default) -> UploadRequest {
- upload(.stream(stream), with: convertible, interceptor: interceptor, fileManager: fileManager)
- }
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
- open func upload(multipartFormData: @escaping (MultipartFormData) -> Void,
- to url: URLConvertible,
- usingThreshold encodingMemoryThreshold: UInt64 = MultipartFormData.encodingMemoryThreshold,
- method: HTTPMethod = .post,
- headers: HTTPHeaders? = nil,
- interceptor: RequestInterceptor? = nil,
- fileManager: FileManager = .default,
- requestModifier: RequestModifier? = nil) -> UploadRequest {
- let convertible = ParameterlessRequestConvertible(url: url,
- method: method,
- headers: headers,
- requestModifier: requestModifier)
- let formData = MultipartFormData(fileManager: fileManager)
- multipartFormData(formData)
- return upload(multipartFormData: formData,
- with: convertible,
- usingThreshold: encodingMemoryThreshold,
- interceptor: interceptor,
- fileManager: fileManager)
- }
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
- open func upload(multipartFormData: @escaping (MultipartFormData) -> Void,
- with request: URLRequestConvertible,
- usingThreshold encodingMemoryThreshold: UInt64 = MultipartFormData.encodingMemoryThreshold,
- interceptor: RequestInterceptor? = nil,
- fileManager: FileManager = .default) -> UploadRequest {
- let formData = MultipartFormData(fileManager: fileManager)
- multipartFormData(formData)
- return upload(multipartFormData: formData,
- with: request,
- usingThreshold: encodingMemoryThreshold,
- interceptor: interceptor,
- fileManager: fileManager)
- }
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
- open func upload(multipartFormData: MultipartFormData,
- to url: URLConvertible,
- usingThreshold encodingMemoryThreshold: UInt64 = MultipartFormData.encodingMemoryThreshold,
- method: HTTPMethod = .post,
- headers: HTTPHeaders? = nil,
- interceptor: RequestInterceptor? = nil,
- fileManager: FileManager = .default,
- requestModifier: RequestModifier? = nil) -> UploadRequest {
- let convertible = ParameterlessRequestConvertible(url: url,
- method: method,
- headers: headers,
- requestModifier: requestModifier)
- let multipartUpload = MultipartUpload(encodingMemoryThreshold: encodingMemoryThreshold,
- request: convertible,
- multipartFormData: multipartFormData)
- return upload(multipartUpload, interceptor: interceptor, fileManager: fileManager)
- }
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
- open func upload(multipartFormData: MultipartFormData,
- with request: URLRequestConvertible,
- usingThreshold encodingMemoryThreshold: UInt64 = MultipartFormData.encodingMemoryThreshold,
- interceptor: RequestInterceptor? = nil,
- fileManager: FileManager = .default) -> UploadRequest {
- let multipartUpload = MultipartUpload(encodingMemoryThreshold: encodingMemoryThreshold,
- request: request,
- multipartFormData: multipartFormData)
- return upload(multipartUpload, interceptor: interceptor, fileManager: fileManager)
- }
-
-
- func upload(_ uploadable: UploadRequest.Uploadable,
- with convertible: URLRequestConvertible,
- interceptor: RequestInterceptor?,
- fileManager: FileManager) -> UploadRequest {
- let uploadable = Upload(request: convertible, uploadable: uploadable)
- return upload(uploadable, interceptor: interceptor, fileManager: fileManager)
- }
- func upload(_ upload: UploadConvertible, interceptor: RequestInterceptor?, fileManager: FileManager) -> UploadRequest {
- let request = UploadRequest(convertible: upload,
- underlyingQueue: rootQueue,
- serializationQueue: serializationQueue,
- eventMonitor: eventMonitor,
- interceptor: interceptor,
- fileManager: fileManager,
- delegate: self)
- perform(request)
- return request
- }
-
-
-
-
- func perform(_ request: Request) {
- rootQueue.async {
- guard !request.isCancelled else { return }
- self.activeRequests.insert(request)
- self.requestQueue.async {
-
- switch request {
- case let r as UploadRequest: self.performUploadRequest(r)
- case let r as DataRequest: self.performDataRequest(r)
- case let r as DownloadRequest: self.performDownloadRequest(r)
- case let r as DataStreamRequest: self.performDataStreamRequest(r)
- default: fatalError("Attempted to perform unsupported Request subclass: \(type(of: request))")
- }
- }
- }
- }
- func performDataRequest(_ request: DataRequest) {
- dispatchPrecondition(condition: .onQueue(requestQueue))
- performSetupOperations(for: request, convertible: request.convertible)
- }
- func performDataStreamRequest(_ request: DataStreamRequest) {
- dispatchPrecondition(condition: .onQueue(requestQueue))
- performSetupOperations(for: request, convertible: request.convertible)
- }
- func performUploadRequest(_ request: UploadRequest) {
- dispatchPrecondition(condition: .onQueue(requestQueue))
- performSetupOperations(for: request, convertible: request.convertible) {
- do {
- let uploadable = try request.upload.createUploadable()
- self.rootQueue.async { request.didCreateUploadable(uploadable) }
- return true
- } catch {
- self.rootQueue.async { request.didFailToCreateUploadable(with: error.asAFError(or: .createUploadableFailed(error: error))) }
- return false
- }
- }
- }
- func performDownloadRequest(_ request: DownloadRequest) {
- dispatchPrecondition(condition: .onQueue(requestQueue))
- switch request.downloadable {
- case let .request(convertible):
- performSetupOperations(for: request, convertible: convertible)
- case let .resumeData(resumeData):
- rootQueue.async { self.didReceiveResumeData(resumeData, for: request) }
- }
- }
- func performSetupOperations(for request: Request,
- convertible: URLRequestConvertible,
- shouldCreateTask: @escaping () -> Bool = { true })
- {
- dispatchPrecondition(condition: .onQueue(requestQueue))
- let initialRequest: URLRequest
- do {
- initialRequest = try convertible.asURLRequest()
- try initialRequest.validate()
- } catch {
- rootQueue.async { request.didFailToCreateURLRequest(with: error.asAFError(or: .createURLRequestFailed(error: error))) }
- return
- }
- rootQueue.async { request.didCreateInitialURLRequest(initialRequest) }
- guard !request.isCancelled else { return }
- guard let adapter = adapter(for: request) else {
- guard shouldCreateTask() else { return }
- rootQueue.async { self.didCreateURLRequest(initialRequest, for: request) }
- return
- }
- adapter.adapt(initialRequest, for: self) { result in
- do {
- let adaptedRequest = try result.get()
- try adaptedRequest.validate()
- self.rootQueue.async { request.didAdaptInitialRequest(initialRequest, to: adaptedRequest) }
- guard shouldCreateTask() else { return }
- self.rootQueue.async { self.didCreateURLRequest(adaptedRequest, for: request) }
- } catch {
- self.rootQueue.async { request.didFailToAdaptURLRequest(initialRequest, withError: .requestAdaptationFailed(error: error)) }
- }
- }
- }
-
- func didCreateURLRequest(_ urlRequest: URLRequest, for request: Request) {
- dispatchPrecondition(condition: .onQueue(rootQueue))
- request.didCreateURLRequest(urlRequest)
- guard !request.isCancelled else { return }
- let task = request.task(for: urlRequest, using: session)
- requestTaskMap[request] = task
- request.didCreateTask(task)
- updateStatesForTask(task, request: request)
- }
- func didReceiveResumeData(_ data: Data, for request: DownloadRequest) {
- dispatchPrecondition(condition: .onQueue(rootQueue))
- guard !request.isCancelled else { return }
- let task = request.task(forResumeData: data, using: session)
- requestTaskMap[request] = task
- request.didCreateTask(task)
- updateStatesForTask(task, request: request)
- }
- func updateStatesForTask(_ task: URLSessionTask, request: Request) {
- dispatchPrecondition(condition: .onQueue(rootQueue))
- request.withState { state in
- switch state {
- case .initialized, .finished:
-
- break
- case .resumed:
- task.resume()
- rootQueue.async { request.didResumeTask(task) }
- case .suspended:
- task.suspend()
- rootQueue.async { request.didSuspendTask(task) }
- case .cancelled:
-
- task.resume()
- task.cancel()
- rootQueue.async { request.didCancelTask(task) }
- }
- }
- }
-
- func adapter(for request: Request) -> RequestAdapter? {
- if let requestInterceptor = request.interceptor, let sessionInterceptor = interceptor {
- return Interceptor(adapters: [requestInterceptor, sessionInterceptor])
- } else {
- return request.interceptor ?? interceptor
- }
- }
- func retrier(for request: Request) -> RequestRetrier? {
- if let requestInterceptor = request.interceptor, let sessionInterceptor = interceptor {
- return Interceptor(retriers: [requestInterceptor, sessionInterceptor])
- } else {
- return request.interceptor ?? interceptor
- }
- }
-
- func finishRequestsForDeinit() {
- requestTaskMap.requests.forEach { request in
- rootQueue.async {
- request.finish(error: AFError.sessionDeinitialized)
- }
- }
- }
- }
- extension Session: RequestDelegate {
- public var sessionConfiguration: URLSessionConfiguration {
- session.configuration
- }
- public var startImmediately: Bool { startRequestsImmediately }
- public func cleanup(after request: Request) {
- activeRequests.remove(request)
- }
- public func retryResult(for request: Request, dueTo error: AFError, completion: @escaping (RetryResult) -> Void) {
- guard let retrier = retrier(for: request) else {
- rootQueue.async { completion(.doNotRetry) }
- return
- }
- retrier.retry(request, for: self, dueTo: error) { retryResult in
- self.rootQueue.async {
- guard let retryResultError = retryResult.error else { completion(retryResult); return }
- let retryError = AFError.requestRetryFailed(retryError: retryResultError, originalError: error)
- completion(.doNotRetryWithError(retryError))
- }
- }
- }
- public func retryRequest(_ request: Request, withDelay timeDelay: TimeInterval?) {
- rootQueue.async {
- let retry: () -> Void = {
- guard !request.isCancelled else { return }
- request.prepareForRetry()
- self.perform(request)
- }
- if let retryDelay = timeDelay {
- self.rootQueue.after(retryDelay) { retry() }
- } else {
- retry()
- }
- }
- }
- }
- extension Session: SessionStateProvider {
- func request(for task: URLSessionTask) -> Request? {
- dispatchPrecondition(condition: .onQueue(rootQueue))
- return requestTaskMap[task]
- }
- func didGatherMetricsForTask(_ task: URLSessionTask) {
- dispatchPrecondition(condition: .onQueue(rootQueue))
- let didDisassociate = requestTaskMap.disassociateIfNecessaryAfterGatheringMetricsForTask(task)
- if didDisassociate {
- waitingCompletions[task]?()
- waitingCompletions[task] = nil
- }
- }
- func didCompleteTask(_ task: URLSessionTask, completion: @escaping () -> Void) {
- dispatchPrecondition(condition: .onQueue(rootQueue))
- let didDisassociate = requestTaskMap.disassociateIfNecessaryAfterCompletingTask(task)
- if didDisassociate {
- completion()
- } else {
- waitingCompletions[task] = completion
- }
- }
- func credential(for task: URLSessionTask, in protectionSpace: URLProtectionSpace) -> URLCredential? {
- dispatchPrecondition(condition: .onQueue(rootQueue))
- return requestTaskMap[task]?.credential ??
- session.configuration.urlCredentialStorage?.defaultCredential(for: protectionSpace)
- }
- func cancelRequestsForSessionInvalidation(with error: Error?) {
- dispatchPrecondition(condition: .onQueue(rootQueue))
- requestTaskMap.requests.forEach { $0.finish(error: AFError.sessionInvalidated(error: error)) }
- }
- }
|