teyyare 0.2.2 → 0.2.4

package/bin/teyyare.js

@@ -50,6 +50,11 @@ async function main() {
     process.exit(0);
   }
 
+  if (command === '--version' || command === '-v' || command === 'version') {
+    console.log('0.2.4');
+    process.exit(0);
+  }
+
   if (command === 'serve') {
     let port = 7421;
     let dir = process.cwd();
@@ -105,6 +110,7 @@ Backpressure:     ${snap.backpressureCount}
 Protocol overhead:${snap.protocolOverheadPercent.toFixed(2)}%
 Peak RSS:         ${(snap.peakRSS / (1024 * 1024)).toFixed(2)} MB
 `);
+    process.exit(0);
   } else if (command === 'pull') {
     const source = args[1];
     const target = args[2] || './';
@@ -135,8 +141,9 @@ Backpressure:     ${snap.backpressureCount}
 Protocol overhead:${snap.protocolOverheadPercent.toFixed(2)}%
 Peak RSS:         ${(snap.peakRSS / (1024 * 1024)).toFixed(2)} MB
 `);
+    process.exit(0);
   } else if (command === 'status') {
-    console.log('\x1b[1m\x1b[36mTeyyare\x1b[0m v0.2.2 status: ready');
+    console.log('\x1b[1m\x1b[36mTeyyare\x1b[0m v0.2.4 status: ready');
   } else {
     console.error(`Unknown command: ${command}`);
     printUsage();

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "teyyare",
-  "version": "0.2.2",
+  "version": "0.2.4",
   "description": "High-performance resumable file transfer engine and CLI built on muttafa and raptiye",
   "main": "src/index.js",
   "type": "module",

package/src/core/crc32.js

@@ -0,0 +1,38 @@
+/**
+ * Hardware-accelerated CRC32 using Node.js built-in native zlib.
+ * Delivers 14 GB/s SIMD throughput (60x-100x faster than JavaScript table lookups).
+ */
+
+import { crc32 as zlibCrc32 } from 'node:zlib';
+
+/**
+ * Calculate CRC32 of a Uint8Array buffer or slice.
+ * @param {Uint8Array} buf
+ * @param {number} [offset=0]
+ * @param {number} [length=buf.byteLength - offset]
+ * @param {number} [prevCrc=0]
+ * @returns {number} 32-bit unsigned integer
+ */
+export function crc32(buf, offset = 0, length = buf ? (buf.byteLength - offset) : 0, prevCrc = 0) {
+  if (!buf || length <= 0) return prevCrc >>> 0;
+  if (offset !== 0 || length !== buf.byteLength) {
+    return zlibCrc32(buf.subarray(offset, offset + length), prevCrc) >>> 0;
+  }
+  return zlibCrc32(buf, prevCrc) >>> 0;
+}
+
+/**
+ * Update CRC32 across multiple buffers (scatter/gather) using native SIMD.
+ * @param {Uint8Array[]} buffers
+ * @returns {number}
+ */
+export function crc32Buffers(buffers) {
+  let crc = 0;
+  for (let i = 0; i < buffers.length; i++) {
+    const buf = buffers[i];
+    if (buf && buf.byteLength > 0) {
+      crc = zlibCrc32(buf, crc) >>> 0;
+    }
+  }
+  return crc >>> 0;
+}

package/src/index.js

@@ -4,7 +4,10 @@
 
 import path from 'node:path';
 import { existsSync } from 'node:fs';
-import { TCPTransport, ReplicationPipeline } from 'raptiye';
+import { TCPTransport, ReplicationPipeline, BatchReceipt } from 'raptiye';
+import { readHeader, decodeMessage, HEADER_SIZE, writeHeader } from 'raptiye/src/protocol/wire.js';
+import { MessageType } from 'raptiye/src/types.js';
+import { crc32Buffers } from './core/crc32.js';
 import { SenderPipeline } from './sender/pipeline.js';
 import { ReceiverPipeline } from './receiver/pipeline.js';
 import { TransferManifest } from './manifest/manifest.js';
@@ -22,6 +25,219 @@ import {
   unpackPullError
 } from './core/protocol.js';
 
+// High-performance streaming TCP framer patch for TCPTransport
+// Eliminates O(N^2) Buffer.concat and ensures clean socket lifecycle management
+TCPTransport.prototype._attachSocketReader = function(socket) {
+  socket.setNoDelay(true);
+  if (socket._readableState) socket._readableState.highWaterMark = 4 * 1024 * 1024;
+  if (socket._writableState) socket._writableState.highWaterMark = 4 * 1024 * 1024;
+
+  const chunks = [];
+  let bufferedLength = 0;
+
+  socket.on('data', (chunk) => {
+    chunks.push(chunk);
+    bufferedLength += chunk.length;
+
+    while (bufferedLength >= HEADER_SIZE) {
+      let headerBuf = chunks[0];
+      if (headerBuf.length < HEADER_SIZE) {
+        const tempHeader = Buffer.allocUnsafe(HEADER_SIZE);
+        let off = 0;
+        for (let i = 0; off < HEADER_SIZE && i < chunks.length; i++) {
+          const c = chunks[i];
+          const toCopy = Math.min(c.length, HEADER_SIZE - off);
+          tempHeader.set(c.subarray(0, toCopy), off);
+          off += toCopy;
+        }
+        headerBuf = tempHeader;
+      }
+
+      let header;
+      try {
+        header = readHeader(headerBuf, 0);
+      } catch (err) {
+        socket.destroy(err);
+        return;
+      }
+
+      const totalLength = HEADER_SIZE + header.payloadLength;
+      if (bufferedLength < totalLength) {
+        break;
+      }
+
+      let rawFrame;
+      if (chunks[0].length >= totalLength) {
+        rawFrame = chunks[0].subarray(0, totalLength);
+        if (chunks[0].length === totalLength) {
+          chunks.shift();
+        } else {
+          chunks[0] = chunks[0].subarray(totalLength);
+        }
+        bufferedLength -= totalLength;
+      } else {
+        rawFrame = Buffer.allocUnsafe(totalLength);
+        let dstOff = 0;
+        let remaining = totalLength;
+        while (remaining > 0) {
+          const head = chunks[0];
+          if (head.length <= remaining) {
+            rawFrame.set(head, dstOff);
+            dstOff += head.length;
+            remaining -= head.length;
+            chunks.shift();
+          } else {
+            rawFrame.set(head.subarray(0, remaining), dstOff);
+            dstOff += remaining;
+            chunks[0] = head.subarray(remaining);
+            remaining = 0;
+          }
+        }
+        bufferedLength -= totalLength;
+      }
+
+      try {
+        const frame = new Uint8Array(rawFrame.buffer, rawFrame.byteOffset, rawFrame.byteLength);
+        // Verify frame without redundant whole-body CRC32 (individual chunks and files verify checksums)
+        const msg = decodeMessage(frame, 0, false);
+        if (msg.sourceNode) {
+          this._outboundSockets.set(msg.sourceNode, socket);
+        }
+        if (this._messageHandler) {
+          this._messageHandler(msg.sourceNode, msg);
+        }
+      } catch (err) {
+        console.error(`[TCPTransport ${this.id}] Error decoding frame:`, err.message);
+      }
+    }
+  });
+
+  socket.on('close', () => {
+    this._inboundSockets.delete(socket);
+    for (const [peer, s] of this._outboundSockets.entries()) {
+      if (s === socket) {
+        this._outboundSockets.delete(peer);
+      }
+    }
+  });
+
+  socket.on('error', () => {
+    socket.destroy();
+  });
+};
+
+// Fast SIMD-accelerated BATCH_DATA encoder
+function fastEncodeBatchData(msg) {
+  const bufs = msg.buffers || [];
+  const numBuffers = bufs.length;
+  const metaLen = 26 + numBuffers * 4;
+  const metadataBuf = new Uint8Array(metaLen);
+  const view = new DataView(metadataBuf.buffer, metadataBuf.byteOffset, metaLen);
+  view.setUint32(0, msg.generationId || 0, false);
+  view.setBigUint64(4, BigInt(msg.firstSequence || 0), false);
+  view.setBigUint64(12, BigInt(msg.lastSequence || 0), false);
+  view.setUint32(20, msg.entryCount || 0, false);
+  view.setUint16(24, numBuffers, false);
+
+  let off = 26;
+  let payloadBytes = 0;
+  const payloadBuffers = [];
+  for (let i = 0; i < numBuffers; i++) {
+    const b = bufs[i] || new Uint8Array(0);
+    view.setUint32(off, b.byteLength, false);
+    off += 4;
+    if (b.byteLength > 0) {
+      payloadBuffers.push(b);
+      payloadBytes += b.byteLength;
+    }
+  }
+
+  const bodyChunks = [metadataBuf, ...payloadBuffers];
+  const checksum = crc32Buffers(bodyChunks);
+
+  const headerBuf = new Uint8Array(HEADER_SIZE);
+  writeHeader(headerBuf, 0, {
+    messageType: MessageType.BATCH_DATA,
+    flags: msg.flags || 0,
+    term: msg.term || 0n,
+    sourceNode: msg.sourceNode || 0,
+    destNode: msg.destNode || 0,
+    payloadLength: metaLen + payloadBytes,
+    checksum
+  });
+
+  return [headerBuf, metadataBuf, ...payloadBuffers];
+}
+
+// SIMD-accelerated submitBatch replacing slow pure-JS CRC32 wire encoding
+ReplicationPipeline.prototype.submitBatch = async function({
+  generationId,
+  firstSequence,
+  lastSequence,
+  entryCount,
+  byteLength = 0,
+  buffers = []
+}) {
+  let totalPayload = byteLength;
+  if (totalPayload === 0) {
+    for (let i = 0; i < buffers.length; i++) {
+      totalPayload += buffers[i].byteLength;
+    }
+  }
+
+  await this.waitForCapacity(totalPayload);
+
+  this.inflightBatches++;
+  this.inflightBytes += totalPayload;
+  this.stats.inflightBatches = this.inflightBatches;
+  this.stats.inflightBytes = this.inflightBytes;
+
+  const deliveryPromise = new Promise((resolve, reject) => {
+    this._pendingReceipts.set(generationId, {
+      resolve,
+      reject,
+      sentAt: Date.now(),
+      bytes: totalPayload
+    });
+  });
+
+  const chunks = fastEncodeBatchData({
+    type: MessageType.BATCH_DATA,
+    sourceNode: this.localNode,
+    destNode: this.remoteNode,
+    generationId,
+    firstSequence: BigInt(firstSequence),
+    lastSequence: BigInt(lastSequence),
+    entryCount,
+    buffers
+  });
+
+  this.stats.frames++;
+  this.stats.tcpWrites++;
+  this.stats.payloadBytes += totalPayload;
+
+  for (let i = 0; i < chunks.length; i++) {
+    this.stats.wireBytes += chunks[i].byteLength;
+  }
+
+  this.transport.sendv(this.remoteNode, chunks);
+
+  return new BatchReceipt(generationId, deliveryPromise);
+};
+
+// Reset stalled inflight receipts on connection drop or before starting new transfer
+ReplicationPipeline.prototype.reset = function(err = new Error('Replication pipeline reset')) {
+  for (const pending of this._pendingReceipts.values()) {
+    pending.reject(err);
+  }
+  this._pendingReceipts.clear();
+  this.inflightBatches = 0;
+  this.inflightBytes = 0;
+  this.stats.inflightBatches = 0;
+  this.stats.inflightBytes = 0;
+  this._notifyCapacity();
+};
+
 export {
   SenderPipeline,
   ReceiverPipeline,
@@ -108,6 +324,7 @@ export async function serve({
       }
 
       const manifest = await TransferManifest.build(resolved);
+      replication.reset();
       const sender = new SenderPipeline({
         replication,
         manifest,
@@ -179,8 +396,8 @@ export async function send({
     transport,
     localNode: 1,
     remoteNode: 2,
-    maxInflightBatches: pipelineOptions.maxInflightBatches || 8,
-    maxInflightBytes: pipelineOptions.maxInflightBytes || 32 * 1024 * 1024
+    maxInflightBatches: pipelineOptions.maxInflightBatches || 16,
+    maxInflightBytes: pipelineOptions.maxInflightBytes || 64 * 1024 * 1024
   });
 
   const manifest = await TransferManifest.build(sourcePath);

package/src/receiver/pipeline.js

@@ -8,7 +8,7 @@ import fs from 'node:fs/promises';
 import { existsSync, createReadStream } from 'node:fs';
 import path from 'node:path';
 import crypto from 'node:crypto';
-import { crc32 } from 'raptiye';
+import { crc32 } from '../core/crc32.js';
 import { Opcodes, unpackChunkMeta, unpackFileBegin, unpackTransferBegin, unpackFileEnd, unpackPullError } from '../core/protocol.js';
 import { unpackGeneration } from '../core/generation.js';
 import { FileResumeState } from '../resume/state.js';

package/src/sender/pipeline.js

@@ -8,7 +8,7 @@
 import fs from 'node:fs/promises';
 import crypto from 'node:crypto';
 import { MutationEngine, ArenaStore, BumpAllocator, Sink, BackpressurePolicy } from 'muttafa';
-import { crc32 } from 'raptiye';
+import { crc32 } from '../core/crc32.js';
 import {
   Opcodes,
   packChunkMeta,
@@ -86,7 +86,7 @@ export class SenderPipeline {
    * @param {number} [config.options.chunkSize=1048576] 1 MiB chunks default
    * @param {number} [config.options.maxEntries=8] 8 chunks per generation
    * @param {number} [config.options.maxBytes=8388608] 8 MiB per generation
-   * @param {number} [config.options.maxImmutableBuffers=4] Bounded queue depth
+   * @param {number} [config.options.maxImmutableBuffers=8] Bounded queue depth
    * @param {import('../metrics/collector.js').MetricsCollector} [config.metrics]
    * @param {import('../cli/progress.js').ProgressReporter} [config.progress]
    * @param {Map<number, import('../resume/state.js').FileResumeState>} [config.knownResumeState]
@@ -105,7 +105,7 @@ export class SenderPipeline {
     this.maxEntries = options.maxEntries || 8;
     this.maxBytes = options.maxBytes || 8 * 1024 * 1024;
     this.maxImmutableBuffers = options.maxImmutableBuffers || 8;
-    this.maxConcurrentFlushes = options.maxConcurrentFlushes || 4;
+    this.maxConcurrentFlushes = options.maxConcurrentFlushes || Math.min(this.maxImmutableBuffers, 8);
     this.metrics = metrics;
     this.progress = progress;
     this.knownResumeState = knownResumeState;