PWN-999 - Retry on socket referral exception (#2207)

app/src/main/java/org/p2p/wallet/referral/repository/ReferralRemoteRepository.kt

@@ -3,9 +3,11 @@ package org.p2p.wallet.referral.repository
 import com.google.gson.Gson
 import org.near.borshj.BorshBuffer
 import timber.log.Timber
+import java.net.SocketTimeoutException
 import java.net.URI
 import java.util.Optional
 import kotlin.time.Duration.Companion.milliseconds
+import kotlin.time.Duration.Companion.seconds
 import kotlinx.coroutines.withContext
 import org.p2p.core.dispatchers.CoroutineDispatchers
 import org.p2p.core.rpc.JsonRpc
@@ -17,6 +19,7 @@ import org.p2p.solanaj.utils.SolanaMessageSigner
 import org.p2p.wallet.auth.gateway.repository.mapper.write
 import org.p2p.wallet.common.feature_toggles.toggles.remote.ReferralProgramEnabledFeatureToggle
 import org.p2p.wallet.infrastructure.network.provider.TokenKeyProvider
+import org.p2p.wallet.utils.retryOnException
 
 class ReferralRemoteRepository(
     private val api: RpcApi,
@@ -29,9 +32,23 @@ class ReferralRemoteRepository(
 
     private val url = URI("https://referral.key.app/")
 
+    /**
+     * Sometimes the service responds with the timeout error `operation timed out`
+     *
+     */
+    private class ReferralServiceTimedOut : Exception()
+
     override suspend fun registerReferent() {
         if (referralEnabledFt.isFeatureEnabled) {
-            registerReferentInternal()
+            retryOnException(
+                exceptionTypes = setOf(
+                    ReferralServiceTimedOut::class,
+                    SocketTimeoutException::class
+                ),
+                maxAttempts = 10,
+                delayMillis = 5.seconds.inWholeMilliseconds,
+                block = ::registerReferentInternal
+            )
         }
     }
 
@@ -57,11 +74,19 @@ class ReferralRemoteRepository(
             request.parseResponse(response, gson)
         } catch (rpcError: JsonRpc.ResponseError) {
             val message = rpcError.message ?: return
-            if (message.contains("duplicate key value violates unique constraint")) {
-                Timber.i("User already registered")
-            } else {
-                Timber.e(rpcError, "Failed to register referent")
+            when {
+                message.contains("duplicate key value violates unique constraint") -> {
+                    Timber.i("User already registered")
+                }
+                message.contains("timed out") -> {
+                    throw ReferralServiceTimedOut()
+                }
+                else -> {
+                    Timber.e(rpcError, "Failed to register referent")
+                }
             }
+        } catch (error: SocketTimeoutException) {
+            throw error
         } catch (error: Throwable) {
             Timber.e(error, "Failed to register referent")
         }