commit 4b6764f66237b0c6f97258d9d84de1bf4665cf2e
author Treehugger Robot <android-test-infra-autosubmit@system.gserviceaccount.com> Wed Mar 12 23:50:47 2025 -0700
committer Gerrit Code Review <noreply-gerritcodereview@google.com> Wed Mar 12 23:50:47 2025 -0700
tree c347702201f28d3e001169db1e205923e56dd0c2
parent 6f7ad334d7feb231ff46c0b870c7d40e3f265702
parent ea72cfbcf78f47ce2dd673b9f59428dbd78435f2

Merge "Handle gRPC canellation case" into main

android/TerminalApp/java/com/android/virtualization/terminal/DebianServiceImpl.kt

diff --git a/android/TerminalApp/java/com/android/virtualization/terminal/DebianServiceImpl.kt b/android/TerminalApp/java/com/android/virtualization/terminal/DebianServiceImpl.kt
index 887ae02..e035ad4 100644
--- a/android/TerminalApp/java/com/android/virtualization/terminal/DebianServiceImpl.kt
+++ b/android/TerminalApp/java/com/android/virtualization/terminal/DebianServiceImpl.kt
@@ -28,6 +28,7 @@
 import com.android.virtualization.terminal.proto.ReportVmActivePortsResponse
 import com.android.virtualization.terminal.proto.ShutdownQueueOpeningRequest
 import com.android.virtualization.terminal.proto.ShutdownRequestItem
+import io.grpc.stub.ServerCallStreamObserver
 import io.grpc.stub.StreamObserver
 
 internal class DebianServiceImpl(context: Context) : DebianServiceImplBase() {
@@ -79,8 +80,15 @@
         request: ShutdownQueueOpeningRequest?,
         responseObserver: StreamObserver<ShutdownRequestItem?>,
     ) {
+        val serverCallStreamObserver = responseObserver as ServerCallStreamObserver<ShutdownRequestItem?>
+        serverCallStreamObserver.setOnCancelHandler {
+            shutdownRunnable = null
+        }
         Log.d(TAG, "openShutdownRequestQueue")
         shutdownRunnable = Runnable {
+            if (serverCallStreamObserver.isCancelled()) {
+                return@Runnable
+            }
             responseObserver.onNext(ShutdownRequestItem.newBuilder().build())
             responseObserver.onCompleted()
             shutdownRunnable = null