Close stream after sink

libp2p · Oct 31, 2022 · b58ee11 · b58ee11
1 parent f6b7f96
commit b58ee11
Showing 1 changed file with 10 additions and 6 deletions.
diff --git a/src/index.ts b/src/index.ts
@@ -148,14 +148,18 @@ async function webtransportBiDiStreamToStream (bidiStream: any, streamId: string
       }
     })(),
     sink: async function (source: Source<Uint8Array | Uint8ArrayList>) {
-      for await (const chunks of source) {
-        if (chunks.constructor === Uint8Array) {
-          await writer.write(chunks)
-        } else {
-          for (const buf of chunks) {
-            await writer.write(buf)
+      try {
+        for await (const chunks of source) {
+          if (chunks.constructor === Uint8Array) {
+            await writer.write(chunks)
+          } else {
+            for (const buf of chunks) {
+              await writer.write(buf)
+            }
           }
         }
+      } finally {
+        stream.closeWrite()
       }
     }
   }