socketduct 0.1.17 → 0.1.19

package/src/client/virtual-socket.js

@@ -7,6 +7,7 @@ import {Duplex} from "node:stream"
  * @property {(streamId: number, callback: (error?: Error | null) => void) => void} _endStream - End the writable side.
  * @property {(streamId: number, error: Error | null) => void} _destroyStream - Destroy the virtual stream.
  * @property {(streamId: number) => void} _resumeReadable - Resume inbound flow.
+ * @property {(streamId: number, bytes: number) => void} [_consumeReadable] - Report bytes returned to the application.
  * @property {() => {address?: string, port?: number}} physicalAddress - Current physical endpoint.
  * @property {() => unknown} ref - Reference owner resources.
  * @property {() => unknown} unref - Unreference owner resources.
@@ -60,6 +61,9 @@ export class VirtualSocket extends Duplex {
   /** @type {boolean} */
   #remoteEnded = false
 
+  /** @type {number} */
+  #emittedDataBytes = 0
+
   /** @type {boolean} */
   connecting = true
 
@@ -123,6 +127,44 @@ export class VirtualSocket extends Duplex {
     this.#connection._resumeReadable(this.#streamId)
   }
 
+  /**
+   * Reports application consumption after Node has emitted flowing data or
+   * fulfilled a paused-mode read. Owners that need durable receive ACKs can
+   * therefore distinguish consumed bytes from process-local readable bytes.
+   * @param {number} [size] - Requested bytes.
+   * @returns {Buffer | string | null} Readable chunk.
+   * @override
+   */
+  read(size) {
+    const emittedBefore = this.#emittedDataBytes
+    const chunk = super.read(size)
+    if (chunk !== null) {
+      const bytes = typeof chunk === "string" ? Buffer.byteLength(chunk) : chunk.length
+      const emittedBytes = this.#emittedDataBytes - emittedBefore
+      if (bytes > emittedBytes) this.#connection._consumeReadable?.(this.#streamId, bytes - emittedBytes)
+    }
+    return chunk
+  }
+
+  /**
+   * A flowing Readable may emit directly from push() without returning a chunk
+   * through read(). Report DATA only after application listeners have run.
+   * @param {string | symbol} event - Event name.
+   * @param {...unknown} args - Event values.
+   * @returns {boolean} Whether the event had listeners.
+   * @override
+   */
+  emit(event, ...args) {
+    const emitted = super.emit(event, ...args)
+    const chunk = args[0]
+    if (event === "data" && (typeof chunk === "string" || Buffer.isBuffer(chunk))) {
+      const bytes = typeof chunk === "string" ? Buffer.byteLength(chunk) : chunk.length
+      this.#emittedDataBytes += bytes
+      this.#connection._consumeReadable?.(this.#streamId, bytes)
+    }
+    return emitted
+  }
+
   /**
    * @override
    * @param {Buffer | string | Uint8Array} chunk - Application bytes.