Also call stream.close() if cancelled

smithy-lang · Aug 1, 2024 · e43b4e0 · e43b4e0
1 parent c906b75
commit e43b4e0
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/...-engine-crt/jvm/src/aws/smithy/kotlin/runtime/http/engine/crt/SdkStreamResponseHandler.kt b/...-engine-crt/jvm/src/aws/smithy/kotlin/runtime/http/engine/crt/SdkStreamResponseHandler.kt
@@ -149,7 +149,11 @@ internal class SdkStreamResponseHandler(
         }
 
         // short circuit, stop buffering data and discard remaining incoming bytes
-        if (isCancelled) return bodyBytesIn.len
+        if (isCancelled) {
+            crtStream?.close()
+            stream.close()
+            return bodyBytesIn.len
+        }
 
         val buffer = SdkBuffer().apply {
             val bytes = bodyBytesIn.readAll()