knitting 0.1.50 → 0.1.52

package/src/memory/lock.d.ts

@@ -49,7 +49,13 @@ export declare enum PayloadBuffer {
     EnvelopeDynamicHeaderString = 43,
     ExternalPayload = 44,
     StaticExternalPayload = 45,
-    ProcessSharedBuffer = 46
+    ProcessSharedBuffer = 46,
+    BufferReference = 47,
+    SharedArrayBuffer = 48,
+    EnvelopeStaticHeaderExternal = 49,
+    EnvelopeDynamicHeaderExternal = 50,
+    EnvelopeStaticHeaderStringExternal = 51,
+    EnvelopeDynamicHeaderStringExternal = 52
 }
 export declare enum LockBound {
     paddingLock = 0,
@@ -70,15 +76,20 @@ export type Task = [
     number
 ] & {
     value: unknown;
+    finalize?: (() => void) | undefined;
     resolve: (value?: unknown) => void;
     reject: (reason?: unknown) => void;
 };
+export declare const PayloadTransportFinalizer: unique symbol;
 export declare const PromisePayloadMarker: unique symbol;
 export type PromisePayloadHandler = (task: Task, isRejected: boolean, value: unknown) => void;
 export declare const beginPromisePayload: (task: Task) => boolean;
 export declare const finishPromisePayload: (task: Task) => void;
 export declare const isPromisePayloadPending: (task: Task) => boolean;
 export declare const resetTaskLocalFlags: (task: Task) => void;
+export declare const addTaskFinalizer: (task: Task, finalizer: () => void) => void;
+export declare const attachPayloadTransportFinalizer: (task: Task, value: unknown) => void;
+export declare const runTaskFinalizers: (task: Task) => void;
 export declare enum TaskIndex {
     /**
      * Worker -> host response flags word.

package/src/memory/lock.js

@@ -54,6 +54,12 @@ export var PayloadBuffer;
     PayloadBuffer[PayloadBuffer["ExternalPayload"] = 44] = "ExternalPayload";
     PayloadBuffer[PayloadBuffer["StaticExternalPayload"] = 45] = "StaticExternalPayload";
     PayloadBuffer[PayloadBuffer["ProcessSharedBuffer"] = 46] = "ProcessSharedBuffer";
+    PayloadBuffer[PayloadBuffer["BufferReference"] = 47] = "BufferReference";
+    PayloadBuffer[PayloadBuffer["SharedArrayBuffer"] = 48] = "SharedArrayBuffer";
+    PayloadBuffer[PayloadBuffer["EnvelopeStaticHeaderExternal"] = 49] = "EnvelopeStaticHeaderExternal";
+    PayloadBuffer[PayloadBuffer["EnvelopeDynamicHeaderExternal"] = 50] = "EnvelopeDynamicHeaderExternal";
+    PayloadBuffer[PayloadBuffer["EnvelopeStaticHeaderStringExternal"] = 51] = "EnvelopeStaticHeaderStringExternal";
+    PayloadBuffer[PayloadBuffer["EnvelopeDynamicHeaderStringExternal"] = 52] = "EnvelopeDynamicHeaderStringExternal";
 })(PayloadBuffer || (PayloadBuffer = {}));
 export var LockBound;
 (function (LockBound) {
@@ -64,6 +70,7 @@ export var LockBound;
 })(LockBound || (LockBound = {}));
 export const LOCK_CACHE_LINE_BYTES = 64;
 export const LOCK_SECTOR_BYTES = 256;
+export const PayloadTransportFinalizer = Symbol.for("knitting.payloadCodec.transportFinalizer");
 export const PromisePayloadMarker = Symbol.for("knitting.promise.payload");
 const TASK_LOCAL_FLAGS_INDEX = 7;
 const TASK_LOCAL_PROMISE_PENDING_FLAG = 1 << 0;
@@ -72,7 +79,8 @@ export const beginPromisePayload = (task) => {
     const flags = task[TASK_LOCAL_FLAGS_INDEX];
     if ((flags & TASK_LOCAL_PROMISE_PENDING_FLAG) !== 0)
         return false;
-    task[TASK_LOCAL_FLAGS_INDEX] = (flags | TASK_LOCAL_PROMISE_PENDING_FLAG) >>> 0;
+    task[TASK_LOCAL_FLAGS_INDEX] = (flags | TASK_LOCAL_PROMISE_PENDING_FLAG) >>>
+        0;
     return true;
 };
 export const finishPromisePayload = (task) => {
@@ -83,6 +91,36 @@ export const isPromisePayloadPending = (task) => (task[TASK_LOCAL_FLAGS_INDEX] &
 export const resetTaskLocalFlags = (task) => {
     task[TASK_LOCAL_FLAGS_INDEX] = 0;
 };
+export const addTaskFinalizer = (task, finalizer) => {
+    const previous = task.finalize;
+    task.finalize = previous === undefined ? finalizer : () => {
+        try {
+            previous();
+        }
+        finally {
+            finalizer();
+        }
+    };
+};
+export const attachPayloadTransportFinalizer = (task, value) => {
+    if (task.finalize !== undefined || value === null || typeof value !== "object") {
+        return;
+    }
+    const finalizer = value[PayloadTransportFinalizer]?.();
+    if (typeof finalizer === "function")
+        addTaskFinalizer(task, finalizer);
+};
+export const runTaskFinalizers = (task) => {
+    const finalizer = task.finalize;
+    task.finalize = undefined;
+    if (finalizer !== undefined) {
+        try {
+            finalizer();
+        }
+        catch {
+        }
+    }
+};
 export var TaskIndex;
 (function (TaskIndex) {
     /**
@@ -186,6 +224,7 @@ const def = (_) => { };
 const createTaskShell = () => {
     const task = new Uint32Array(TaskIndex.Size);
     task.value = null;
+    task.finalize = undefined;
     task.resolve = def;
     task.reject = def;
     task[TASK_LOCAL_FLAGS_INDEX] = 0;
@@ -266,10 +305,12 @@ export const lock2 = ({ headers, headerSlotStrideU32, LockBoundSector, payload,
         lockSector: payloadLockRegion,
         textCompat: resolvedTextCompat,
         onPromise: (task, isRejected, value) => {
-            if ((task[TASK_LOCAL_FLAGS_INDEX] & TASK_LOCAL_PROMISE_TRACKED_FLAG) !== 0 &&
+            if ((task[TASK_LOCAL_FLAGS_INDEX] & TASK_LOCAL_PROMISE_TRACKED_FLAG) !==
+                0 &&
                 pendingPromiseCount > 0) {
                 task[TASK_LOCAL_FLAGS_INDEX] =
-                    (task[TASK_LOCAL_FLAGS_INDEX] & ~TASK_LOCAL_PROMISE_TRACKED_FLAG) >>> 0;
+                    (task[TASK_LOCAL_FLAGS_INDEX] & ~TASK_LOCAL_PROMISE_TRACKED_FLAG) >>>
+                        0;
                 pendingPromiseCount = (pendingPromiseCount - 1) | 0;
             }
             promiseHandler(task, isRejected, value);
@@ -322,7 +363,8 @@ export const lock2 = ({ headers, headerSlotStrideU32, LockBoundSector, payload,
         const flags = task[TASK_LOCAL_FLAGS_INDEX];
         if ((flags & TASK_LOCAL_PROMISE_TRACKED_FLAG) !== 0)
             return;
-        task[TASK_LOCAL_FLAGS_INDEX] = (flags | TASK_LOCAL_PROMISE_TRACKED_FLAG) >>> 0;
+        task[TASK_LOCAL_FLAGS_INDEX] = (flags | TASK_LOCAL_PROMISE_TRACKED_FLAG) >>>
+            0;
         pendingPromiseCount = (pendingPromiseCount + 1) | 0;
     };
     const encodeTaskValue = (task, slotIndex) => encodeTask(task, slotIndex);
@@ -564,171 +606,9 @@ export const lock2 = ({ headers, headerSlotStrideU32, LockBoundSector, payload,
                 return modified;
             };
         }
-        if (!shouldSettle) {
-            if (!onResolved) {
-                return () => {
-                    let diff = (a_load(hostBits, 0) ^ LastWorker) | 0;
-                    if (diff === 0)
-                        return 0;
-                    let modified = 0;
-                    let consumedBits = 0 | 0;
-                    let last = lastResolved;
-                    if (last === 32) {
-                        const idx = 31 - clz32(diff);
-                        const selectedBit = 1 << idx;
-                        const task = getTask(idx);
-                        decodeTask(task, idx);
-                        consumedBits = (consumedBits ^ selectedBit) | 0;
-                        settleTask(task);
-                        diff ^= selectedBit;
-                        modified++;
-                        if ((modified & 7) === 0 && consumedBits !== 0) {
-                            LastWorker = (LastWorker ^ consumedBits) | 0;
-                            a_store(workerBits, 0, LastWorker);
-                            consumedBits = 0 | 0;
-                        }
-                        last = idx;
-                    }
-                    while (diff !== 0) {
-                        const lowerMask = last === 31 ? 0x7fffffff : ((1 << last) - 1);
-                        let pick = diff & lowerMask;
-                        if (pick === 0)
-                            pick = diff;
-                        const idx = 31 - clz32(pick);
-                        const selectedBit = 1 << idx;
-                        const task = getTask(idx);
-                        decodeTask(task, idx);
-                        consumedBits = (consumedBits ^ selectedBit) | 0;
-                        settleTask(task);
-                        diff ^= selectedBit;
-                        modified++;
-                        if ((modified & 7) === 0 && consumedBits !== 0) {
-                            LastWorker = (LastWorker ^ consumedBits) | 0;
-                            a_store(workerBits, 0, LastWorker);
-                            consumedBits = 0 | 0;
-                        }
-                        last = idx;
-                    }
-                    if (consumedBits !== 0) {
-                        LastWorker = (LastWorker ^ consumedBits) | 0;
-                        a_store(workerBits, 0, LastWorker);
-                    }
-                    lastResolved = last;
-                    return modified;
-                };
-            }
-            const onResolvedTask = onResolved;
-            return () => {
-                let diff = (a_load(hostBits, 0) ^ LastWorker) | 0;
-                if (diff === 0)
-                    return 0;
-                let modified = 0;
-                let consumedBits = 0 | 0;
-                let last = lastResolved;
-                if (last === 32) {
-                    const idx = 31 - clz32(diff);
-                    const selectedBit = 1 << idx;
-                    const task = getTask(idx);
-                    decodeTask(task, idx);
-                    consumedBits = (consumedBits ^ selectedBit) | 0;
-                    settleTask(task);
-                    onResolvedTask(task);
-                    diff ^= selectedBit;
-                    modified++;
-                    if ((modified & 7) === 0 && consumedBits !== 0) {
-                        LastWorker = (LastWorker ^ consumedBits) | 0;
-                        a_store(workerBits, 0, LastWorker);
-                        consumedBits = 0 | 0;
-                    }
-                    last = idx;
-                }
-                while (diff !== 0) {
-                    const lowerMask = last === 31 ? 0x7fffffff : ((1 << last) - 1);
-                    let pick = diff & lowerMask;
-                    if (pick === 0)
-                        pick = diff;
-                    const idx = 31 - clz32(pick);
-                    const selectedBit = 1 << idx;
-                    const task = getTask(idx);
-                    decodeTask(task, idx);
-                    consumedBits = (consumedBits ^ selectedBit) | 0;
-                    settleTask(task);
-                    onResolvedTask(task);
-                    diff ^= selectedBit;
-                    modified++;
-                    if ((modified & 7) === 0 && consumedBits !== 0) {
-                        LastWorker = (LastWorker ^ consumedBits) | 0;
-                        a_store(workerBits, 0, LastWorker);
-                        consumedBits = 0 | 0;
-                    }
-                    last = idx;
-                }
-                if (consumedBits !== 0) {
-                    LastWorker = (LastWorker ^ consumedBits) | 0;
-                    a_store(workerBits, 0, LastWorker);
-                }
-                lastResolved = last;
-                return modified;
-            };
-        }
+        const hasOnResolved = onResolved !== undefined;
+        const onResolvedTask = onResolved ?? def;
         const shouldSettleTask = shouldSettle;
-        if (!onResolved) {
-            return () => {
-                let diff = (a_load(hostBits, 0) ^ LastWorker) | 0;
-                if (diff === 0)
-                    return 0;
-                let modified = 0;
-                let consumedBits = 0 | 0;
-                let last = lastResolved;
-                if (last === 32) {
-                    const idx = 31 - clz32(diff);
-                    const selectedBit = 1 << idx;
-                    const task = getTask(idx);
-                    decodeTask(task, idx);
-                    consumedBits = (consumedBits ^ selectedBit) | 0;
-                    if (shouldSettleTask(task)) {
-                        settleTask(task);
-                    }
-                    diff ^= selectedBit;
-                    modified++;
-                    if ((modified & 7) === 0 && consumedBits !== 0) {
-                        LastWorker = (LastWorker ^ consumedBits) | 0;
-                        a_store(workerBits, 0, LastWorker);
-                        consumedBits = 0 | 0;
-                    }
-                    last = idx;
-                }
-                while (diff !== 0) {
-                    const lowerMask = last === 31 ? 0x7fffffff : ((1 << last) - 1);
-                    let pick = diff & lowerMask;
-                    if (pick === 0)
-                        pick = diff;
-                    const idx = 31 - clz32(pick);
-                    const selectedBit = 1 << idx;
-                    const task = getTask(idx);
-                    decodeTask(task, idx);
-                    consumedBits = (consumedBits ^ selectedBit) | 0;
-                    if (shouldSettleTask(task)) {
-                        settleTask(task);
-                    }
-                    diff ^= selectedBit;
-                    modified++;
-                    if ((modified & 7) === 0 && consumedBits !== 0) {
-                        LastWorker = (LastWorker ^ consumedBits) | 0;
-                        a_store(workerBits, 0, LastWorker);
-                        consumedBits = 0 | 0;
-                    }
-                    last = idx;
-                }
-                if (consumedBits !== 0) {
-                    LastWorker = (LastWorker ^ consumedBits) | 0;
-                    a_store(workerBits, 0, LastWorker);
-                }
-                lastResolved = last;
-                return modified;
-            };
-        }
-        const onResolvedTask = onResolved;
         return () => {
             let diff = (a_load(hostBits, 0) ^ LastWorker) | 0;
             if (diff === 0)
@@ -742,9 +622,10 @@ export const lock2 = ({ headers, headerSlotStrideU32, LockBoundSector, payload,
                 const task = getTask(idx);
                 decodeTask(task, idx);
                 consumedBits = (consumedBits ^ selectedBit) | 0;
-                if (shouldSettleTask(task)) {
+                if (shouldSettleTask === undefined || shouldSettleTask(task)) {
                     settleTask(task);
-                    onResolvedTask(task);
+                    if (hasOnResolved)
+                        onResolvedTask(task);
                 }
                 diff ^= selectedBit;
                 modified++;
@@ -765,9 +646,10 @@ export const lock2 = ({ headers, headerSlotStrideU32, LockBoundSector, payload,
                 const task = getTask(idx);
                 decodeTask(task, idx);
                 consumedBits = (consumedBits ^ selectedBit) | 0;
-                if (shouldSettleTask(task)) {
+                if (shouldSettleTask === undefined || shouldSettleTask(task)) {
                     settleTask(task);
-                    onResolvedTask(task);
+                    if (hasOnResolved)
+                        onResolvedTask(task);
                 }
                 diff ^= selectedBit;
                 modified++;
@@ -793,6 +675,7 @@ export const lock2 = ({ headers, headerSlotStrideU32, LockBoundSector, payload,
         if (recycled) {
             fillTaskFrom(recycled, headersBuffer, off);
             recycled.value = null;
+            recycled.finalize = undefined;
             recycled.resolve = def;
             recycled.reject = def;
             task = recycled;