add retry to middleware

vetrek · Mar 26, 2024 · 1f45fa7 · 1f45fa7
1 parent 39ce44f
commit 1f45fa7
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/Source/Core/ApiClient+Closure.swift b/Source/Core/ApiClient+Closure.swift
@@ -251,8 +251,11 @@ extension ApiClient {
     request: URLRequest,
     queue: DispatchQueue = .main,
     progressHUD: SNProgressHUD? = nil,
+    retryCount: Int = 0,
     completion: @escaping (Response<Data>) -> Void
   ) -> NetworkCancellable? {
+    guard retryCount < 2 else { return nil }
+
     let task = session?.dataTask(
       with: request
     ) { (data, response, error) in
@@ -311,6 +314,7 @@ extension ApiClient {
                 request: request,
                 queue: queue,
                 progressHUD: progressHUD,
+                retryCount: retryCount + 1,
                 completion: completion
               )
               return
@@ -328,6 +332,7 @@ extension ApiClient {
                 request: request,
                 queue: queue,
                 progressHUD: progressHUD,
+                retryCount: retryCount + 1,
                 completion: completion
               )
               return