fix: proteus send message error missing logic - WPB-16116 (#2535)

Co-authored-by: Jullian Mercier <[email protected]>
wireapp · Feb 13, 2025 · c26a23b · c26a23b
1 parent de14595
commit c26a23b
Showing 1 changed file with 7 additions and 6 deletions.
diff --git a/wire-ios-request-strategy/Sources/Message Sending/MessageSender.swift b/wire-ios-request-strategy/Sources/Message Sending/MessageSender.swift
@@ -173,15 +173,15 @@ public final class MessageSender: MessageSenderInterface {
                 .broadcastProteusMessage(message: messageData)
             await handleProteusSuccess(message: message, messageSendingStatus: messageStatus, response: response)
         } catch let networkError as NetworkError {
-            let retryOperation: () async throws -> Void = { [weak self] in
+            let operation: () async throws -> Void = { [weak self] in
                 try await self?.broadcastMessage(message: message)
             }
 
             try await handleNetworkError(
                 networkError,
                 message: message,
                 apiVersion: apiVersion,
-                retryOperation: retryOperation
+                operation: operation
             )
         }
     }
@@ -239,15 +239,15 @@ public final class MessageSender: MessageSenderInterface {
                 )
             await handleProteusSuccess(message: message, messageSendingStatus: messageStatus, response: response)
         } catch let networkError as NetworkError {
-            let retryOperation: () async throws -> Void = { [weak self] in
+            let operation: () async throws -> Void = { [weak self] in
                 try await self?.sendMessage(message: message)
             }
 
             try await handleNetworkError(
                 networkError,
                 message: message,
                 apiVersion: apiVersion,
-                retryOperation: retryOperation
+                operation: operation
             )
         }
     }
@@ -256,11 +256,12 @@ public final class MessageSender: MessageSenderInterface {
         _ networkError: NetworkError,
         message: any ProteusMessage,
         apiVersion: APIVersion,
-        retryOperation: () async throws -> Void
+        operation: () async throws -> Void
     ) async throws {
         do {
             let missingClients = try await handleProteusFailure(message: message, networkError)
             try await sessionEstablisher.establishSession(with: missingClients, apiVersion: apiVersion)
+            try await operation()
         } catch let error as MessageSendError {
             guard retryCount < maxRetryAttempts else {
                 retryCount = 0
@@ -269,7 +270,7 @@ public final class MessageSender: MessageSenderInterface {
 
             retryCount += 1
 
-            try await retryOperation()
+            try await operation()
         } catch {
             throw error
         }