fix(AWSCore): call response interceptors in HTTP response handlers (#…

…5495)
aws-amplify · Jan 21, 2025 · 2d6ac35 · 2d6ac35
1 parent d9823c6
commit 2d6ac35
Showing 1 changed file with 15 additions and 0 deletions.
diff --git 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(id<AWSNetworkingHTTPResponseInterceptor>interceptor 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(id<AWSNetworkingHTTPResponseInterceptor>interceptor 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