feat: 🎸 track number of written bytes

streamich · Jun 20, 2023 · b80f7b7 · b80f7b7
1 parent 69281ff
commit b80f7b7
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/src/fsa-to-node/FsaNodeWriteStream.ts b/src/fsa-to-node/FsaNodeWriteStream.ts
@@ -24,6 +24,7 @@ import type { IWriteStreamOptions } from '../node/types/options';
 export class FsaNodeWriteStream extends Writable implements IWriteStream {
   protected __pending__: boolean = true;
   protected __closed__: boolean = false;
+  protected __bytes__: number = 0;
   protected readonly __stream__: Promise<IFileSystemWritableFileStream>;
   protected readonly __mutex__ = concurrency(1);
 
@@ -52,6 +53,7 @@ export class FsaNodeWriteStream extends Writable implements IWriteStream {
       const writable = await this.__stream__;
       for (const buffer of buffers) {
         await writable.write(buffer);
+        this.__bytes__ += buffer.byteLength;
       }
     });
   }
@@ -77,7 +79,7 @@ export class FsaNodeWriteStream extends Writable implements IWriteStream {
   // ------------------------------------------------------------- IWriteStream
 
   public get bytesWritten(): number {
-    return 0;
+    return this.__bytes__;
   }
 
   public get pending(): boolean {

diff --git a/src/fsa-to-node/__tests__/FsaNodeFs.test.ts b/src/fsa-to-node/__tests__/FsaNodeFs.test.ts
@@ -641,6 +641,7 @@ describe('.createWriteStream()', () => {
     stream.write(Buffer.from('DEF'));
     stream.end();
     await new Promise(resolve => stream.once('close', resolve));
+    expect(stream.bytesWritten).toBe(6);
     expect(mfs.__vol.toJSON()).toStrictEqual({
       '/mountpoint/folder/file': 'test',
       '/mountpoint/folder/file2': 'ABCDEF',