@nxtedition/shared 1.1.2 → 1.2.0

package/index.js

@@ -68,26 +68,17 @@ 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 maxLen = 4
-    for (const buf of raw) {
-      maxLen += buf.byteLength ?? buf.length * 3
-    }
-
-    assert(maxLen < size / 2)
+    assert(len < size / 2)
 
-    if (size - writePos < maxLen + 4) {
-      if (readPos < maxLen + 4) {
+    if (size - writePos < len + 4) {
+      if (readPos < len + 4) {
         return false
       }
 
@@ -96,31 +87,64 @@ export function writer({ sharedState, sharedBuffer }) {
     } else {
       const available = writePos >= readPos ? size - writePos : readPos - writePos
 
-      if (available < maxLen + 4) {
+      if (available < len + 4) {
         return false
       }
     }
 
-    const lenPos = writePos
-    writePos += 4
+    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 + 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
       }
     }
 
-    const len = writePos - lenPos
+    assert(pos <= maxLen)
 
-    assert(len <= maxLen)
-
-    buffer.writeInt32LE(len, lenPos)
-
-    Atomics.store(state, WRITE_INDEX, writePos)
-    Atomics.notify(state, WRITE_INDEX)
+    release(pos)
 
     return true
   }
@@ -149,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.2",
+  "version": "1.2.0",
   "description": "Ring Buffer for NodeJS cross Worker communication",
   "main": "index.js",
   "repository": {