chore: trace gRPC connection state changes; properly handle cancellation gRPC exceptions when backgrounding app

Signed-off-by: Brandon McAnsh <git@bmcreations.dev>

Author: bmc08gt

libs/logging/src/main/kotlin/com/getcode/utils/ErrorUtils.kt

@@ -38,9 +38,10 @@ object ErrorUtils {
             else throwable
 
         if (throwableCause is StatusRuntimeException) {
-            when (throwableCause.status) {
-                Status.UNAVAILABLE -> return
-                Status.CANCELLED -> return
+            when (throwableCause.status.code) {
+                Status.Code.UNAVAILABLE -> return
+                Status.Code.CANCELLED -> return
+                else -> { /* fall through */ }
             }
         }
 

services/flipcash/src/main/kotlin/com/flipcash/services/inject/FlipcashModule.kt

@@ -40,6 +40,9 @@ import dagger.Provides
 import dagger.hilt.InstallIn
 import dagger.hilt.android.qualifiers.ApplicationContext
 import dagger.hilt.components.SingletonComponent
+import com.getcode.utils.TraceType
+import com.getcode.utils.trace
+import io.grpc.ConnectivityState
 import io.grpc.ManagedChannel
 import io.grpc.android.AndroidChannelBuilder
 import io.grpc.okhttp.OkHttpChannelBuilder
@@ -87,6 +90,21 @@ internal object FlipcashModule {
                 }
             }
             .build()
+            .also { observeChannelState("flipcash", it) }
+    }
+
+    private fun observeChannelState(name: String, channel: ManagedChannel) {
+        val state = channel.getState(false)
+        trace(
+            tag = "gRPC",
+            message = "$name => $state",
+            type = TraceType.StateChange,
+        )
+        if (state != ConnectivityState.SHUTDOWN) {
+            channel.notifyWhenStateChanged(state) {
+                observeChannelState(name, channel)
+            }
+        }
     }
 
     @Provides

services/opencode/src/main/kotlin/com/getcode/opencode/inject/OpenCodeModule.kt

@@ -41,6 +41,9 @@ import dagger.Provides
 import dagger.hilt.InstallIn
 import dagger.hilt.android.qualifiers.ApplicationContext
 import dagger.hilt.components.SingletonComponent
+import com.getcode.utils.TraceType
+import com.getcode.utils.trace
+import io.grpc.ConnectivityState
 import io.grpc.ManagedChannel
 import io.grpc.android.AndroidChannelBuilder
 import io.grpc.okhttp.OkHttpChannelBuilder
@@ -101,6 +104,7 @@ object OpenCodeModule {
                 }
             }
             .build()
+            .also { observeChannelState("opencode", it) }
     }
 
     @Singleton
@@ -122,6 +126,21 @@ object OpenCodeModule {
                 }
             }
             .build()
+            .also { observeChannelState("opencode-stream", it) }
+    }
+
+    private fun observeChannelState(name: String, channel: ManagedChannel) {
+        val state = channel.getState(false)
+        trace(
+            tag = "gRPC",
+            message = "$name => $state",
+            type = TraceType.StateChange,
+        )
+        if (state != ConnectivityState.SHUTDOWN) {
+            channel.notifyWhenStateChanged(state) {
+                observeChannelState(name, channel)
+            }
+        }
     }
 
     @Provides

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

@@ -14,7 +14,6 @@ abstract class GrpcApi(protected val managedChannels: List<ManagedChannel>): Def
         ProcessLifecycleOwner.get().lifecycle.addObserver(this)
     }
 
-
     override fun onStart(owner: LifecycleOwner) {
         super.onStart(owner)
         warmUp()
@@ -23,4 +22,4 @@ abstract class GrpcApi(protected val managedChannels: List<ManagedChannel>): Def
     private fun warmUp() {
         managedChannels.onEach { it.enterIdle() }
     }
-}
+}

services/opencode/src/main/kotlin/com/getcode/opencode/utils/logging/LoggingClientCallListener.kt

@@ -22,22 +22,31 @@ class LoggingClientCallListener<ReqT, ResT>(
         val requestLog = requestQueue.joinToString(separator = ", ", prefix = "[", postfix = "]") { getObjectDetails(it) }
         val responseLog = responseQueue.joinToString(separator = ", ", prefix = "[", postfix = "]") { getObjectDetails(it) }
 
-        if (status.isOk) {
-            trace(tag = "RpcLogging", message = "Request: $requestLog", type = TraceType.Network)
-            trace(tag = "RpcLogging", message = "Response: $responseLog", type = TraceType.Network)
-            trace(
-                tag = "RpcLogging",
-                message = "The request was processed successfully",
-                type = TraceType.Network
-            )
-        } else {
-            trace(tag = "RpcLogging", message = "Request: $requestLog", type = TraceType.Network)
-            trace(
-                tag = "RpcLogging",
-                message = "An error occurred while processing the request",
-                type = TraceType.Error,
-                error = status.asRuntimeException(trailers)
-            )
+        trace(tag = "RpcLogging", message = "Request: $requestLog", type = TraceType.Network)
+        when {
+            status.isOk -> {
+                trace(tag = "RpcLogging", message = "Response: $responseLog", type = TraceType.Network)
+                trace(
+                    tag = "RpcLogging",
+                    message = "The request was processed successfully",
+                    type = TraceType.Network
+                )
+            }
+            status.code == Status.Code.CANCELLED -> {
+                trace(
+                    tag = "RpcLogging",
+                    message = "Request cancelled",
+                    type = TraceType.Log,
+                )
+            }
+            else -> {
+                trace(
+                    tag = "RpcLogging",
+                    message = "An error occurred while processing the request",
+                    type = TraceType.Error,
+                    error = status.asRuntimeException(trailers)
+                )
+            }
         }
 
         super.onClose(status, trailers)