@nxtedition/shared 2.0.13 → 2.0.16

package/index.js

@@ -66,6 +66,18 @@ export function reader(options, cb) {
   }
 }
 
+function arrayWrite(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
+}
+
 export function writer({ sharedState, sharedBuffer }) {
   const state = new Int32Array(sharedState)
   const buffer = Buffer.from(sharedBuffer)
@@ -74,96 +86,59 @@ export function writer({ sharedState, sharedBuffer }) {
   let readPos = 0
   let writePos = 0
 
-  function _tryWrite(maxLen, fn, opaque) {
-    assert(maxLen <= size)
+  return function write(len, fn, opaque) {
+    if (!len) {
+      return
+    }
 
-    readPos = Atomics.load(state, READ_INDEX)
+    if (Number.isInteger(len)) {
+      assert(len >= 0, `len: ${len} >= 0`)
+      assert(typeof fn === 'function', `fn: ${typeof fn} === 'function`)
+    } else {
+      opaque = Array.isArray(len) ? len : arguments
 
-    if (size - writePos < maxLen) {
-      if (readPos < maxLen) {
-        return false
+      len = 0
+      for (const buf of opaque) {
+        len += Buffer.byteLength(buf)
       }
 
+      fn = arrayWrite
+    }
+
+    assert(len <= size)
+
+    readPos = Atomics.load(state, READ_INDEX)
+
+    if (size - writePos < len) {
       buffer.writeInt32LE(-1, writePos)
+      writePos += 4
+      buffer.subarray(writePos).fill(-4)
       writePos = 0
-    } else {
-      const available = writePos >= readPos ? size - writePos : readPos - writePos
+    }
 
-      if (available < maxLen) {
-        return false
+    while (true) {
+      const available = writePos >= readPos ? size - writePos : readPos - writePos
+      if (available >= len) {
+        break
       }
+      Atomics.wait(state, READ_INDEX, readPos)
     }
 
     buffer.writeInt32LE(-2, writePos)
     writePos += 4
-
     buffer.writeInt32LE(-3, writePos)
-    const len = fn(buffer.subarray(writePos + 4, writePos + 4 + maxLen), opaque)
-    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
+    const pos = fn(buffer.subarray(writePos + 4, writePos + 4 + len), opaque)
+
+    assert(pos > 0, `pos: ${pos} > 0`)
+    assert(pos <= len, `pos: ${pos} <= len: ${len}`)
+    assert(pos <= size, `pos: ${pos} <= size: ${size}`)
+
+    buffer.writeInt32LE(pos, writePos)
+    writePos += 4 + pos
     buffer.writeInt32LE(-4, writePos)
 
     Atomics.store(state, WRITE_INDEX, writePos)
     Atomics.notify(state, WRITE_INDEX)
-
-    return true
   }
-
-  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) {
-    if (!args.length) {
-      return
-    }
-
-    let maxLen
-    let fn
-    let opaque
-
-    if (Number.isInteger(args[0])) {
-      maxLen = args[0]
-      fn = args[1]
-      opaque = args[2]
-
-      assert(maxLen >= 0, `maxLen: ${maxLen} >= 0`)
-      assert(typeof fn === 'function', `fn: ${typeof fn} === 'function`)
-    } else {
-      if (Array.isArray(args[0])) {
-        args = args[0]
-      }
-
-      maxLen = 0
-      for (const buf of args) {
-        maxLen += Buffer.byteLength(buf)
-      }
-
-      fn = defaultWrite
-
-      opaque = args
-    }
-
-    while (!_tryWrite(maxLen, fn, opaque)) {
-      // TODO (fix): Async? Warn? Timeout?
-      Atomics.wait(state, READ_INDEX, readPos, 1e3)
-    }
-  }
-
-  write.write = write
-  write.flush = () => {}
-
-  return write
 }

package/package.json

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