Add asyncDispose support to Readable and Writable (#19)

Author: yassernasc

index.js

@@ -65,6 +65,12 @@ exports.Readable = class Readable extends stream.Readable {
 
     super.unshift(chunk)
   }
+
+  async [Symbol.asyncDispose]() {
+    if (!this.destroyed) this.destroy()
+
+    await new Promise((resolve) => exports.finished(this, resolve))
+  }
 }
 
 exports.Writable = class Writable extends stream.Writable {
@@ -131,6 +137,12 @@ exports.Writable = class Writable extends stream.Writable {
 
     return result
   }
+
+  async [Symbol.asyncDispose]() {
+    if (!this.destroyed) this.destroy()
+
+    await new Promise((resolve) => exports.finished(this, resolve))
+  }
 }
 
 exports.Duplex = class Duplex extends stream.Duplex {

test/basic.js

@@ -720,4 +720,19 @@ test('duplexPair', (t) => {
   pairA.end()
 })
 
+test('async disposable', async (t) => {
+  t.plan(2)
+
+  const readable = new Readable()
+  const writable = new Writable()
+
+  {
+    await using disposableReadable = readable
+    await using disposableWritable = writable
+  }
+
+  t.ok(readable.destroyed)
+  t.ok(writable.destroyed)
+})
+
 function noop() {}