chore(services): suppress goaways

Signed-off-by: Brandon McAnsh <[email protected]>

Author: bmc08gt

services/flipcash/src/main/kotlin/com/flipcash/services/internal/network/services/AccountService.kt

@@ -1,6 +1,7 @@
 package com.flipcash.services.internal.network.services
 
 import com.flipcash.services.internal.network.api.AccountApi
+import com.getcode.opencode.internal.network.extensions.foldWithSuppression
 import com.flipcash.services.models.GetUserFlagsError
 import com.flipcash.services.models.LoginError
 import com.flipcash.services.models.RegisterError
@@ -15,7 +16,7 @@ internal class AccountService @Inject constructor(
     suspend fun register(owner: KeyPair): Result<ID> {
         return runCatching {
             api.register(owner)
-        }.fold(
+        }.foldWithSuppression(
             onSuccess = { response ->
                 when (response.result) {
                     RpcAccountService.RegisterResponse.Result.OK -> {
@@ -46,7 +47,7 @@ internal class AccountService @Inject constructor(
     suspend fun login(owner: KeyPair): Result<ID> {
         return runCatching {
             api.login(owner)
-        }.fold(
+        }.foldWithSuppression(
             onSuccess = { response ->
                 when (response.result) {
                     RpcAccountService.LoginResponse.Result.OK -> {
@@ -77,7 +78,7 @@ internal class AccountService @Inject constructor(
     suspend fun getUserFlags(owner: KeyPair, userId: ID): Result<RpcAccountService.UserFlags> {
         return runCatching {
             api.getUserFlags(userId, owner)
-        }.fold(
+        }.foldWithSuppression(
             onSuccess = { response ->
                 when (response.result) {
                     RpcAccountService.GetUserFlagsResponse.Result.OK -> Result.success(response.userFlags)

services/flipcash/src/main/kotlin/com/flipcash/services/internal/network/services/ActivityFeedService.kt

@@ -3,6 +3,7 @@ package com.flipcash.services.internal.network.services
 import com.codeinc.flipcash.gen.activity.v1.ActivityFeedService
 import com.codeinc.flipcash.gen.activity.v1.Model
 import com.flipcash.services.internal.network.api.ActivityFeedApi
+import com.getcode.opencode.internal.network.extensions.foldWithSuppression
 import com.flipcash.services.models.ActivityFeedType
 import com.flipcash.services.models.GetActivityFeedMessagesError
 import com.flipcash.services.models.QueryOptions
@@ -20,7 +21,7 @@ internal class ActivityFeedService @Inject constructor(
     ): Result<List<Model.Notification>> {
         return runCatching {
             api.getLatestNotifications(owner, type, maxItems)
-        }.fold(
+        }.foldWithSuppression(
             onSuccess = { response ->
                 when (response.result) {
                     ActivityFeedService.GetLatestNotificationsResponse.Result.OK -> Result.success(response.notificationsList)
@@ -42,7 +43,7 @@ internal class ActivityFeedService @Inject constructor(
     ): Result<List<Model.Notification>> {
         return runCatching {
             api.getNotificationsPage(owner, type, queryOptions)
-        }.fold(
+        }.foldWithSuppression(
             onSuccess = { response ->
                 when (response.result) {
                     ActivityFeedService.GetPagedNotificationsResponse.Result.OK -> Result.success(response.notificationsList)
@@ -63,7 +64,7 @@ internal class ActivityFeedService @Inject constructor(
     ): Result<List<Model.Notification>> {
         return runCatching {
             api.getNotificationsByIds(owner, ids)
-        }.fold(
+        }.foldWithSuppression(
             onSuccess = { response ->
                 when (response.result) {
                     ActivityFeedService.GetBatchNotificationsResponse.Result.OK -> Result.success(response.notificationsList)

services/flipcash/src/main/kotlin/com/flipcash/services/internal/network/services/PurchaseService.kt

@@ -4,6 +4,7 @@ import com.codeinc.flipcash.gen.iap.v1.IapService
 import com.flipcash.services.internal.model.billing.IapMetadata
 import com.flipcash.services.internal.model.billing.Receipt
 import com.flipcash.services.internal.network.api.PurchaseApi
+import com.getcode.opencode.internal.network.extensions.foldWithSuppression
 import com.flipcash.services.models.PurchaseAckError
 import com.getcode.ed25519.Ed25519.KeyPair
 import javax.inject.Inject
@@ -14,7 +15,7 @@ internal class PurchaseService @Inject constructor(
     suspend fun onPurchaseCompleted(owner: KeyPair, receipt: Receipt, metadata: IapMetadata): Result<Unit> {
         return runCatching {
             api.onPurchaseCompleted(owner, receipt, metadata)
-        }.fold(
+        }.foldWithSuppression(
             onSuccess = { response ->
                 when (response.result) {
                     IapService.OnPurchaseCompletedResponse.Result.OK -> Result.success(Unit)

services/flipcash/src/main/kotlin/com/flipcash/services/internal/network/services/PushService.kt

@@ -1,6 +1,7 @@
 package com.flipcash.services.internal.network.services
 
 import com.flipcash.services.internal.network.api.PushApi
+import com.getcode.opencode.internal.network.extensions.foldWithSuppression
 import com.flipcash.services.models.AddTokenError
 import com.flipcash.services.models.DeleteTokenError
 import com.getcode.ed25519.Ed25519.KeyPair
@@ -17,7 +18,7 @@ internal class PushService @Inject constructor(
     ): Result<Unit> {
         return runCatching {
             api.addToken(owner, token, installationId)
-        }.fold(
+        }.foldWithSuppression(
             onSuccess = { response ->
                 when (response.result) {
                     RpcPushService.AddTokenResponse.Result.OK -> Result.success(Unit)
@@ -38,7 +39,7 @@ internal class PushService @Inject constructor(
     ): Result<Unit> {
         return runCatching {
             api.deleteTokens(owner, installationId)
-        }.fold(
+        }.foldWithSuppression(
             onSuccess = { response ->
                 when (response.result) {
                     RpcPushService.DeleteTokensResponse.Result.OK -> Result.success(Unit)

services/opencode/src/main/kotlin/com/getcode/opencode/internal/network/core/GrpcApi.kt

@@ -1,18 +1,5 @@
 package com.getcode.opencode.internal.network.core
 
 import io.grpc.ManagedChannel
-import io.grpc.stub.ClientCallStreamObserver
-import io.grpc.stub.ClientResponseObserver
-import io.grpc.stub.StreamObserver
-import kotlinx.coroutines.channels.awaitClose
-import kotlinx.coroutines.flow.Flow
-import kotlinx.coroutines.flow.callbackFlow
-import kotlinx.coroutines.flow.cancellable
-import kotlinx.coroutines.flow.channelFlow
-import kotlinx.coroutines.flow.onCompletion
-import okhttp3.Request
-import okhttp3.Response
-import java.util.concurrent.atomic.AtomicBoolean
-import kotlin.reflect.KFunction2
 
 abstract class GrpcApi(protected val managedChannel: ManagedChannel)

services/opencode/src/main/kotlin/com/getcode/opencode/internal/network/extensions/FoldWithSuppression.kt

@@ -0,0 +1,28 @@
+package com.getcode.opencode.internal.network.extensions
+
+import com.getcode.utils.SuppressibleException
+import io.grpc.Status
+import io.grpc.StatusException
+
+fun <T, R> Result<T>.foldWithSuppression(
+    onSuccess: (T) -> R,
+    onFailure: (Throwable) -> R
+): R = fold(
+    onSuccess = onSuccess,
+    onFailure = { throwable ->
+        val shouldSuppress = when {
+            didReceiveGoAway(throwable) -> true
+            else -> false
+        }
+        if (shouldSuppress) {
+            return@fold onFailure(SuppressibleException(throwable))
+        }
+
+        onFailure(throwable)
+    }
+)
+
+private fun didReceiveGoAway(throwable: Throwable): Boolean =
+    (throwable is StatusException &&
+            throwable.status == Status.UNAVAILABLE &&
+            throwable.cause?.message?.contains("Goaway") == true)

services/opencode/src/main/kotlin/com/getcode/opencode/internal/network/services/AccountService.kt

@@ -3,6 +3,7 @@ package com.getcode.opencode.internal.network.services
 import com.codeinc.opencode.gen.account.v1.AccountService
 import com.getcode.ed25519.Ed25519.KeyPair
 import com.getcode.opencode.internal.network.api.AccountApi
+import com.getcode.opencode.internal.network.extensions.foldWithSuppression
 import com.getcode.opencode.model.accounts.AccountInfo
 import com.getcode.opencode.model.core.errors.CodeAccountCheckError
 import com.getcode.opencode.model.core.errors.GetAccountsError
@@ -40,7 +41,7 @@ internal class AccountService @Inject constructor(
     ): Result<Map<PublicKey, AccountInfo>> {
         return runCatching {
             api.getTokenAccounts(accountOwner, requestingOwner)
-        }.fold(
+        }.foldWithSuppression(
             onSuccess = { response ->
                 when (response.result) {
                     AccountService.GetTokenAccountInfosResponse.Result.OK -> {

services/opencode/src/main/kotlin/com/getcode/opencode/internal/network/services/CurrencyService.kt

@@ -2,6 +2,7 @@ package com.getcode.opencode.internal.network.services
 
 import com.codeinc.opencode.gen.currency.v1.CurrencyService
 import com.getcode.opencode.internal.network.api.CurrencyApi
+import com.getcode.opencode.internal.network.extensions.foldWithSuppression
 import com.getcode.opencode.model.core.errors.GetRatesError
 import com.getcode.opencode.model.financial.CurrencyCode
 import com.getcode.opencode.model.financial.Rate
@@ -16,7 +17,7 @@ internal class CurrencyService @Inject constructor(
     ): Result<Map<CurrencyCode, Rate>> {
         return runCatching {
             api.getAllRates(from?.toEpochMilliseconds())
-        }.fold(
+        }.foldWithSuppression(
             onSuccess = { response ->
                 when (response.result) {
                     CurrencyService.GetAllRatesResponse.Result.OK -> {

services/opencode/src/main/kotlin/com/getcode/opencode/internal/network/services/MessagingService.kt

@@ -6,6 +6,7 @@ import com.getcode.opencode.internal.bidi.BidirectionalStreamReference
 import com.getcode.opencode.internal.bidi.openBidirectionalStream
 import com.getcode.opencode.internal.network.api.MessagingApi
 import com.getcode.opencode.internal.network.extensions.clientPongWith
+import com.getcode.opencode.internal.network.extensions.foldWithSuppression
 import com.getcode.opencode.internal.network.extensions.openMessageStreamRequest
 import com.getcode.opencode.internal.network.extensions.toPublicKey
 import com.getcode.opencode.model.core.errors.AckMessagesError
@@ -107,7 +108,7 @@ internal class MessagingService @Inject constructor(
                     }
                 }
             }
-        ).fold(
+        ).foldWithSuppression(
             onFailure = { onEvent(Result.failure(it)) },
             onSuccess = { onEvent(Result.success(it)) }
         )
@@ -117,12 +118,12 @@ internal class MessagingService @Inject constructor(
         rendezvous: KeyPair,
     ): Result<List<MessagingService.Message>> {
         return runCatching { api.pollMessages(rendezvous) }
-            .fold(
+            .foldWithSuppression(
                 onSuccess = { response ->
                     Result.success(response.messagesList)
                 },
                 onFailure = {
-                    return Result.failure(PollMessagesError.Other(cause = it))
+                    Result.failure(PollMessagesError.Other(cause = it))
                 }
             )
     }
@@ -132,7 +133,7 @@ internal class MessagingService @Inject constructor(
         messageIds: List<MessagingService.MessageId> = emptyList(),
     ): Result<Unit> {
         return runCatching { api.ackMessages(rendezvous, messageIds) }
-            .fold(
+            .foldWithSuppression(
                 onSuccess = { response ->
                     when (response.result) {
                         RpcMessagingService.AckMesssagesResponse.Result.OK -> Result.success(Unit)
@@ -154,7 +155,7 @@ internal class MessagingService @Inject constructor(
         message: RpcMessagingService.Message.Builder,
     ): Result<PublicKey> {
         return runCatching { api.sendMessage(rendezvous = rendezvous, message = message) }
-            .fold(
+            .foldWithSuppression(
                 onSuccess = { response ->
                     when (response.result) {
                         RpcMessagingService.SendMessageResponse.Result.OK -> {

services/opencode/src/main/kotlin/com/getcode/opencode/internal/network/services/TransactionService.kt

@@ -9,6 +9,7 @@ import com.getcode.opencode.internal.bidi.BidirectionalStreamReference
 import com.getcode.opencode.internal.bidi.openBidirectionalStream
 import com.getcode.opencode.internal.domain.mapping.TransactionMetadataMapper
 import com.getcode.opencode.internal.network.api.TransactionApi
+import com.getcode.opencode.internal.network.extensions.foldWithSuppression
 import com.getcode.opencode.internal.network.extensions.toModel
 import com.getcode.opencode.model.core.errors.AirdropError
 import com.getcode.opencode.model.core.errors.GetIntentMetadataError
@@ -80,7 +81,7 @@ internal class TransactionService @Inject constructor(
     ): Result<TransactionMetadata> {
         return runCatching {
             api.getIntentMetadata(intentId, owner)
-        }.fold(
+        }.foldWithSuppression(
             onSuccess = { response ->
                 when (response.result) {
                     TransactionService.GetIntentMetadataResponse.Result.OK -> {
@@ -110,7 +111,7 @@ internal class TransactionService @Inject constructor(
     ): Result<Limits> {
         return runCatching {
             api.getLimits(owner, consumedSince.toEpochMilliseconds())
-        }.fold(
+        }.foldWithSuppression(
             onSuccess = { response ->
                 when (response.result) {
                     TransactionService.GetLimitsResponse.Result.OK -> {
@@ -141,7 +142,7 @@ internal class TransactionService @Inject constructor(
     ): Result<WithdrawalAvailability> {
         return runCatching {
             api.canWithdrawToAccount(destination)
-        }.fold(
+        }.foldWithSuppression(
             onSuccess = { response ->
                 val availability = WithdrawalAvailability.newInstance(
                     destination = destination,
@@ -166,7 +167,7 @@ internal class TransactionService @Inject constructor(
     ): Result<ExchangeData.WithRate> {
         return runCatching {
             api.airdrop(type, destination)
-        }.fold(
+        }.foldWithSuppression(
             onSuccess = { response ->
                 when (response.result) {
                     TransactionService.AirdropResponse.Result.OK -> Result.success(response.exchangeData.toModel())
@@ -188,7 +189,7 @@ internal class TransactionService @Inject constructor(
     ): Result<Unit> {
         return runCatching {
             api.voidGiftCard(owner, giftCardVault)
-        }.fold(
+        }.foldWithSuppression(
             onSuccess = { response ->
                 when (response.result) {
                     TransactionService.VoidGiftCardResponse.Result.OK -> Result.success(Unit)