Bulk Load CDK: Fix transient failure due to multiple channel closures

airbytehq · Oct 20, 2024 · 09e6b1f · 09e6b1f
1 parent 7b70a70
commit 09e6b1f
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/...dk/bulk/toolkits/load-s3/src/main/kotlin/io/airbyte/cdk/load/file/s3/S3MultipartUpload.kt b/...dk/bulk/toolkits/load-s3/src/main/kotlin/io/airbyte/cdk/load/file/s3/S3MultipartUpload.kt
@@ -81,10 +81,8 @@ class S3MultipartUpload<T : OutputStream>(
 
     inner class UploadStream : OutputStream() {
         override fun close() = runBlocking {
-            workQueue.send {
-                if (closeOnce.setOnce()) {
-                    workQueue.close()
-                }
+            if (closeOnce.setOnce()) {
+                workQueue.send { workQueue.close() }
             }
         }