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

…#47130)
airbytehq · Oct 21, 2024 · d122510 · d122510
1 parent cfb6c5c
commit d122510
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() }
             }
         }