cleanup

connectrpc · Aug 10, 2023 · f4f978a · f4f978a
1 parent d6a0278
commit f4f978a
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/library/src/main/kotlin/build/buf/connect/impl/ProtocolClient.kt b/library/src/main/kotlin/build/buf/connect/impl/ProtocolClient.kt
@@ -193,13 +193,21 @@ class ProtocolClient(
         }
         continuation.invokeOnCancellation {
             httpStream.sendClose()
+            httpStream.receiveClose()
         }
         val stream = Stream(
             onSend = { buffer ->
                 httpStream.send(streamFunc.requestBodyFunction(buffer))
+            },
+            onReceiveClose = {
+                httpStream.receiveClose()
+            },
+            onSendClose = {
+                httpStream.sendClose()
             }
         )
         channel.invokeOnClose {
+            // Receive channel is closed so the stream's receive will be closed.
             stream.receiveClose()
         }
         continuation.resume(