Consistently avoid non-local returns

connectrpc · Oct 2, 2023 · 47b9af9 · 47b9af9
1 parent a910957
commit 47b9af9
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/library/src/main/kotlin/com/connectrpc/protocols/GRPCInterceptor.kt b/library/src/main/kotlin/com/connectrpc/protocols/GRPCInterceptor.kt
@@ -142,14 +142,16 @@ internal class GRPCInterceptor(
                         val completion = completionParser.parse(headers, emptyMap())
                         if (completion != null) {
                             val exception = completion.toConnectExceptionOrNull(serializationStrategy)
-                            return@fold StreamResult.Complete(
+                            StreamResult.Complete(
                                 code = exception?.code ?: Code.OK,
                                 cause = exception,
                                 trailers = headers,
                             )
+                        } else {
+                            responseCompressionPool = clientConfig
+                                .compressionPool(headers[GRPC_ENCODING]?.first())
+                            StreamResult.Headers(headers)
                         }
-                        responseCompressionPool = clientConfig.compressionPool(headers[GRPC_ENCODING]?.first())
-                        StreamResult.Headers(headers)
                     },
                     onMessage = { result ->
                         val (_, unpackedMessage) = Envelope.unpackWithHeaderByte(