@nxtedition/shared 2.0.9 → 2.0.12

package/index.js

@@ -38,11 +38,16 @@ async function* _reader({ sharedState, sharedBuffer }, cb) {
       const len = buffer.readInt32LE(readPos)
       readPos += 4
 
+      assert(len > 0, `len: ${len} > 0`)
+
       const raw = buffer.subarray(readPos, readPos + len)
       readPos += len
 
       if (cb) {
-        await cb(raw)
+        const thenable = cb(raw)
+        if (thenable) {
+          await thenable
+        }
       } else {
         yield raw
       }
@@ -70,8 +75,6 @@ export function writer({ sharedState, sharedBuffer }) {
   let writePos = 0
 
   function _tryWrite(maxLen, fn, opaque) {
-    maxLen += 4
-
     assert(maxLen <= size)
 
     readPos = Atomics.load(state, READ_INDEX)
@@ -94,13 +97,15 @@ export function writer({ sharedState, sharedBuffer }) {
     buffer.writeInt32LE(-2, writePos)
     writePos += 4
 
+    buffer.writeInt32LE(-3, writePos)
     const len = fn(buffer.subarray(writePos + 4, writePos + 4 + maxLen), opaque)
-    assert(len <= maxLen - 4, `len: ${len} <= maxLen: ${maxLen - 4}`)
+    assert(len > 0, `len: ${len} > 0`)
+    assert(len <= maxLen, `len: ${len} <= maxLen: ${maxLen}`)
     assert(len <= size, `len: ${len} <= size: ${size}`)
 
     buffer.writeInt32LE(len, writePos)
     writePos += 4 + len
-    buffer.writeInt32LE(-3, writePos)
+    buffer.writeInt32LE(-4, writePos)
 
     Atomics.store(state, WRITE_INDEX, writePos)
     Atomics.notify(state, WRITE_INDEX)
@@ -108,18 +113,16 @@ export function writer({ sharedState, sharedBuffer }) {
     return true
   }
 
-  async function _write(maxLen, fn, opaque) {
-    assert(maxLen <= size)
-
-    const buf = Buffer.allocUnsafe(maxLen)
-    const len = fn(buf, opaque)
-
-    while (!_tryWrite(len, (dst, buf) => buf.copy(dst, 0, 0, len))) {
-      const { async, value } = Atomics.waitAsync(state, READ_INDEX, readPos)
-      if (async) {
-        await value
+  function defaultWrite(dst, data) {
+    let pos = 0
+    for (const buf of data) {
+      if (typeof buf === 'string') {
+        pos += dst.write(buf, pos)
+      } else {
+        pos += buf.copy(dst, pos)
       }
     }
+    return pos
   }
 
   function write(...args) {
@@ -143,23 +146,20 @@ export function writer({ sharedState, sharedBuffer }) {
         args = args[0]
       }
 
-      maxLen = args.reduce((len, buf) => len + Buffer.byteLength(buf), 0)
-      fn = (dst, data) => {
-        let pos = 0
-        for (const buf of data) {
-          if (typeof buf === 'string') {
-            pos += dst.write(buf, pos)
-          } else {
-            pos += buf.copy(dst, pos)
-          }
-        }
-        return pos
+      maxLen = 0
+      for (const buf of args) {
+        maxLen += Buffer.byteLength(buf)
       }
+
+      fn = defaultWrite
+
       opaque = args
     }
 
-    if (!_tryWrite(maxLen, fn, opaque)) {
-      return _write(maxLen, fn, opaque)
+    while (!_tryWrite(maxLen, fn, opaque)) {
+      // TODO (fix): Async? Warn? Timeout?
+      const result = Atomics.wait(state, READ_INDEX, readPos, 1e3)
+      assert(result !== 'timed-out')
     }
   }
 

package/package.json

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