From 2d6ac3540b2069781d4254012e19e95c6f13e2bc Mon Sep 17 00:00:00 2001 From: Abhash Kumar Singh Date: Tue, 21 Jan 2025 09:27:11 -0800 Subject: [PATCH] fix(AWSCore): call response interceptors in HTTP response handlers (#5495) --- AWSCore/Networking/AWSURLSessionManager.m | 15 +++++++++++++++ 1 file changed, 15 insertions(+) diff --git a/AWSCore/Networking/AWSURLSessionManager.m b/AWSCore/Networking/AWSURLSessionManager.m index d3c7a8cdc25..24e84c11b89 100644 --- a/AWSCore/Networking/AWSURLSessionManager.m +++ b/AWSCore/Networking/AWSURLSessionManager.m @@ -279,6 +279,13 @@ - (void)URLSession:(NSURLSession *)session task:(NSURLSessionTask *)sessionTask if (!delegate.error && [sessionTask.response isKindOfClass:[NSHTTPURLResponse class]]) { NSHTTPURLResponse *httpResponse = (NSHTTPURLResponse *)sessionTask.response; + + for(idinterceptor in delegate.request.responseInterceptors) { + [interceptor interceptResponse:httpResponse + data:nil + originalRequest:sessionTask.originalRequest + currentRequest:sessionTask.currentRequest]; + } if (delegate.shouldWriteToFile) { NSError *error = nil; @@ -351,6 +358,14 @@ - (void)URLSession:(NSURLSession *)session task:(NSURLSessionTask *)sessionTask if (delegate.error && ([sessionTask.response isKindOfClass:[NSHTTPURLResponse class]] || sessionTask.response == nil) && delegate.request.retryHandler) { + + for(idinterceptor in delegate.request.responseInterceptors) { + [interceptor interceptResponse:(NSHTTPURLResponse *)sessionTask.response + data:nil + originalRequest:sessionTask.originalRequest + currentRequest:sessionTask.currentRequest]; + } + AWSNetworkingRetryType retryType = [delegate.request.retryHandler shouldRetry:delegate.currentRetryCount originalRequest:delegate.request response:(NSHTTPURLResponse *)sessionTask.response