feat(GiniCaptureSDK): retry logic for feedback in DocumentService

PP-615
gini · Nov 20, 2024 · 9002004 · 9002004
1 parent e21771e
commit 9002004
Showing 1 changed file with 25 additions and 5 deletions.
diff --git a/CaptureSDK/GiniCaptureSDK/Sources/GiniCaptureSDK/Networking/DocumentService.swift b/CaptureSDK/GiniCaptureSDK/Sources/GiniCaptureSDK/Networking/DocumentService.swift
@@ -149,18 +149,38 @@ public final class DocumentService: DocumentServiceProtocol {
             Log(message: "Cannot send feedback: no document", event: .error)
             return
         }
-        captureNetworkService.sendFeedback(document: document, 
+        attemptFeedback(document: document,
+                        updatedExtractions: updatedExtractions,
+                        updatedCompoundExtractions: updatedCompoundExtractions,
+                        retryCount: 3)
+    }
+
+    private func attemptFeedback(document: Document,
+                                 updatedExtractions: [Extraction],
+                                 updatedCompoundExtractions: [String: [[Extraction]]]?,
+                                 retryCount: Int) {
+        captureNetworkService.sendFeedback(document: document,
                                            updatedExtractions: updatedExtractions,
                                            updatedCompoundExtractions: updatedCompoundExtractions) { result in
             switch result {
             case .success:
                 Log(message: "Feedback sent with \(updatedExtractions.count) extractions and \(updatedCompoundExtractions?.count ?? 0) compound extractions",
                     event: "🚀")
             case .failure(let error):
-                let message = "Error sending feedback for document with id: \(document.id) error: \(error)"
-                Log(message: message, event: .error)
-                let errorLog = ErrorLog(description: message, error: error)
-                GiniConfiguration.shared.errorLogger.handleErrorLog(error: errorLog)
+                if retryCount > 0 {
+                    Log(message: "Retrying feedback due to error: \(error). Remaining retries: \(retryCount - 1)", event: .warning)
+                    DispatchQueue.global().asyncAfter(deadline: .now() + 5) {
+                        self.attemptFeedback(document: document,
+                                             updatedExtractions: updatedExtractions,
+                                             updatedCompoundExtractions: updatedCompoundExtractions,
+                                             retryCount: retryCount - 1)
+                    }
+                } else {
+                    let message = "Error sending feedback for document with id: \(document.id) error: \(error)"
+                    Log(message: message, event: .error)
+                    let errorLog = ErrorLog(description: message, error: error)
+                    GiniConfiguration.shared.errorLogger.handleErrorLog(error: errorLog)
+                }
             }
         }
     }