final tweaks

connectrpc · Oct 18, 2023 · 0e76254 · 0e76254
1 parent fdcd426
commit 0e76254
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/library/src/main/kotlin/com/connectrpc/ClientOnlyStreamInterface.kt b/library/src/main/kotlin/com/connectrpc/ClientOnlyStreamInterface.kt
@@ -30,6 +30,7 @@ interface ClientOnlyStreamInterface<Input, Output> {
      * Receive a single response and close the stream.
      *
      * @return the single response [Output].
+     * @throws ConnectException If an error occurs making the call or processing the response.
      */
     suspend fun receiveAndClose(): Output
 

diff --git a/library/src/main/kotlin/com/connectrpc/impl/ClientOnlyStream.kt b/library/src/main/kotlin/com/connectrpc/impl/ClientOnlyStream.kt
@@ -34,7 +34,7 @@ internal class ClientOnlyStream<Input, Output>(
         try {
             messageStream.sendClose()
             val message = resultChannel.receive()
-            val additionalMessage = resultChannel.tryReceive()
+            val additionalMessage = resultChannel.receiveCatching()
             if (additionalMessage.isSuccess) {
                 throw ConnectException(code = Code.UNKNOWN, message = "unary stream has multiple messages")
             }