@nxtedition/shared 1.1.0 → 1.2.0

package/index.js

@@ -1,3 +1,5 @@
+import assert from 'node:assert'
+
 const WRITE_INDEX = 0
 const READ_INDEX = 1
 const END_OF_PACKET = -1
@@ -66,21 +68,14 @@ export function writer({ sharedState, sharedBuffer }) {
   let writePos = 0
   let flushing = null
 
-  function tryWrite(...raw) {
-    if (Array.isArray(raw[0])) {
-      raw = raw[0]
-    }
-
-    if (!raw.length) {
-      return true
+  function acquire(len) {
+    if (!len) {
+      return Buffer.alloc(0)
     }
 
     readPos = Atomics.load(state, READ_INDEX)
 
-    let len = 0
-    for (const buf of raw) {
-      len += buf.byteLength ?? Buffer.byteLength(buf)
-    }
+    assert(len < size / 2)
 
     if (size - writePos < len + 4) {
       if (readPos < len + 4) {
@@ -97,20 +92,59 @@ export function writer({ sharedState, sharedBuffer }) {
       }
     }
 
+    return buffer.subarray(writePos + 4, writePos + 4 + len)
+  }
+
+  function acquireSync (len) {
+    while (true) {
+      const buf = acquire(len)
+      if (buf) {
+        return buf
+      }
+      Atomics.wait(state, READ_INDEX, readPos)
+    }
+  }
+
+  function release(len) {
     buffer.writeInt32LE(len, writePos)
-    writePos += 4
+    writePos += 4 + len
+
+    Atomics.store(state, WRITE_INDEX, writePos)
+    Atomics.notify(state, WRITE_INDEX)
+  }
+
+  function tryWrite(...raw) {
+    if (Array.isArray(raw[0])) {
+      raw = raw[0]
+    }
+
+    if (!raw.length) {
+      return true
+    }
 
+    let maxLen = 4
+    for (const buf of raw) {
+      maxLen += buf.byteLength ?? buf.length * 3
+    }
+
+    const dst = acquire(maxLen)
+    if (!dst) {
+      return false
+    }
+
+    let pos = 0
     for (const buf of raw) {
       if (typeof buf === 'string') {
-        writePos += buffer.write(buf, writePos)
+        pos += buffer.write(buf, pos)
       } else {
-        buffer.set(buf, writePos)
-        writePos += buf.byteLength
+        buffer.set(buf, pos)
+        pos += buf.byteLength
       }
     }
 
-    Atomics.store(state, WRITE_INDEX, writePos)
-    Atomics.notify(state, WRITE_INDEX)
+    assert(pos <= maxLen)
+
+    release(pos)
 
     return true
   }
@@ -139,5 +173,10 @@ export function writer({ sharedState, sharedBuffer }) {
     return (flushing ??= flush())
   }
 
+  write.acquireSync = acquireSync
+  write.acquire = acquire
+  write.release = release
+  write.flush = flush
+
   return write
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@nxtedition/shared",
-  "version": "1.1.0",
+  "version": "1.2.0",
   "description": "Ring Buffer for NodeJS cross Worker communication",
   "main": "index.js",
   "repository": {