acquireStateLock was allocating a fresh SharedArrayBuffer on every retry iteration via Atomics.wait(new Int32Array(new SharedArrayBuffer(4)), ...). The buffer is never mutated and never escapes, so hoisting it before the loop is a provably-equivalent transformation — Atomics.wait always sees value 0 whether the buffer is fresh or reused. Fixes #316 Co-authored-by: Claude Opus 4.7 (1M context) <noreply@anthropic.com>
This commit is contained in:
5
.changeset/316-state-lock-buffer-hoist.md
Normal file
5
.changeset/316-state-lock-buffer-hoist.md
Normal file
@@ -0,0 +1,5 @@
|
||||
---
|
||||
type: Fixed
|
||||
pr: 316
|
||||
---
|
||||
acquireStateLock no longer allocates a fresh SharedArrayBuffer on every retry iteration — the sleep buffer is now hoisted before the retry loop and reused, eliminating per-retry heap pressure. (#316)
|
||||
@@ -940,6 +940,7 @@ function acquireStateLock(statePath) {
|
||||
const staleThresholdMs = 10000;
|
||||
const maxWaitMs = 30000;
|
||||
const startedAt = Date.now();
|
||||
const sleepBuffer = new Int32Array(new SharedArrayBuffer(4)); // hoisted; value stays 0, pure Atomics.wait timeout target
|
||||
|
||||
// eslint-disable-next-line no-constant-condition
|
||||
while (true) {
|
||||
@@ -973,7 +974,7 @@ function acquireStateLock(statePath) {
|
||||
);
|
||||
}
|
||||
const jitter = Math.floor(Math.random() * 50);
|
||||
Atomics.wait(new Int32Array(new SharedArrayBuffer(4)), 0, 0, retryDelay + jitter);
|
||||
Atomics.wait(sleepBuffer, 0, 0, retryDelay + jitter);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
204
tests/perf-316-state-lock-buffer-alloc.test.cjs
Normal file
204
tests/perf-316-state-lock-buffer-alloc.test.cjs
Normal file
@@ -0,0 +1,204 @@
|
||||
/**
|
||||
* Regression test for perf #316 — acquireStateLock allocates a fresh
|
||||
* SharedArrayBuffer on every retry iteration.
|
||||
*
|
||||
* The fix: hoist the sleep buffer allocation to once before the retry loop.
|
||||
* The buffer is never mutated and never escapes — Atomics.wait(buf,0,0,delay)
|
||||
* always sees 0 whether the buffer is fresh or reused, so the behavior is
|
||||
* identical.
|
||||
*
|
||||
* Observable invariant (POST-FIX): exactly ONE SharedArrayBuffer is allocated
|
||||
* per acquireStateLock call, regardless of retry count.
|
||||
*
|
||||
* RED (pre-fix): sabCount >= 2 when >= 1 retry occurs.
|
||||
* GREEN (post-fix): sabCount === 1.
|
||||
*
|
||||
* Strategy: two Worker threads run in parallel.
|
||||
* Worker A (lock holder): writes the lock file with the current process pid,
|
||||
* sleeps 400ms via Atomics.wait, then removes the lock.
|
||||
* Worker B (writer): installs a counting SharedArrayBuffer stub, then calls
|
||||
* writeStateMd — which calls acquireStateLock and retries until A releases.
|
||||
* Reports sabCount via postMessage.
|
||||
*
|
||||
* Using Worker threads (not child processes) avoids the node --test subprocess-
|
||||
* detection hang that occurs with spawn() inside a test runner worker context.
|
||||
*
|
||||
* Total test wall-time: ~400-600ms.
|
||||
*/
|
||||
|
||||
const { test, describe, beforeEach, afterEach } = require('node:test');
|
||||
const assert = require('node:assert/strict');
|
||||
const fs = require('fs');
|
||||
const path = require('path');
|
||||
const os = require('os');
|
||||
const { Worker } = require('worker_threads');
|
||||
|
||||
// ─────────────────────────────────────────────────────────────────────────────
|
||||
// Constants
|
||||
// ─────────────────────────────────────────────────────────────────────────────
|
||||
|
||||
const STATE_CJS_PATH = path.join(
|
||||
__dirname, '..', 'get-shit-done', 'bin', 'lib', 'state.cjs'
|
||||
);
|
||||
|
||||
const MINIMAL_STATE_MD = [
|
||||
'# Project State',
|
||||
'',
|
||||
'**Status:** Planning',
|
||||
'**Current Phase:** 01',
|
||||
].join('\n') + '\n';
|
||||
|
||||
// Worker A: holds the lock file for holdMs, then removes it.
|
||||
// workerData: { lockPath, holdMs }
|
||||
const HOLDER_WORKER_CODE = `
|
||||
const { parentPort, workerData } = require('worker_threads');
|
||||
const fs = require('fs');
|
||||
// Write pid to lock file so acquireStateLock sees a live pid and retries.
|
||||
fs.writeFileSync(workerData.lockPath, String(process.pid));
|
||||
parentPort.postMessage({ pid: process.pid });
|
||||
// Synchronous sleep — blocks this worker thread for holdMs ms.
|
||||
const buf = new Int32Array(new SharedArrayBuffer(4));
|
||||
Atomics.wait(buf, 0, 0, workerData.holdMs);
|
||||
// Release the lock.
|
||||
try { fs.unlinkSync(workerData.lockPath); } catch { /* already gone */ }
|
||||
parentPort.postMessage({ done: true });
|
||||
`;
|
||||
|
||||
// Worker B: stubs global.SharedArrayBuffer with a counting call-through wrapper,
|
||||
// then calls writeStateMd (triggering acquireStateLock), and reports sabCount.
|
||||
// workerData: { stateCjsPath, statePath, content, tmpDir }
|
||||
const WRITER_WORKER_CODE = `
|
||||
const { parentPort, workerData } = require('worker_threads');
|
||||
const RealSAB = global.SharedArrayBuffer;
|
||||
let sabCount = 0;
|
||||
// Stub: increments sabCount, calls through so Atomics.wait gets a real SAB-backed buffer.
|
||||
function StubSAB(...args) {
|
||||
sabCount++;
|
||||
return new RealSAB(...args);
|
||||
}
|
||||
StubSAB.prototype = RealSAB.prototype;
|
||||
global.SharedArrayBuffer = StubSAB;
|
||||
|
||||
// Delete cache entry to ensure a fresh require picks up the stubbed constructor.
|
||||
// (The inline "new SharedArrayBuffer(4)" in acquireStateLock reads the global at
|
||||
// call time, so even a cached require would use our stub — but deleting avoids
|
||||
// any module-level SAB allocations from a prior require contaminating sabCount.)
|
||||
delete require.cache[workerData.stateCjsPath];
|
||||
const { writeStateMd } = require(workerData.stateCjsPath);
|
||||
|
||||
let callErr = null;
|
||||
try {
|
||||
writeStateMd(workerData.statePath, workerData.content, workerData.tmpDir);
|
||||
} catch (e) {
|
||||
callErr = (e && e.message) ? e.message : String(e);
|
||||
}
|
||||
parentPort.postMessage({ sabCount, callErr });
|
||||
`;
|
||||
|
||||
// ─────────────────────────────────────────────────────────────────────────────
|
||||
// Helpers
|
||||
// ─────────────────────────────────────────────────────────────────────────────
|
||||
|
||||
function makeTempDir() {
|
||||
const dir = fs.mkdtempSync(path.join(os.tmpdir(), 'gsd-316-'));
|
||||
fs.mkdirSync(path.join(dir, '.planning'), { recursive: true });
|
||||
return dir;
|
||||
}
|
||||
|
||||
function removeTempDir(dir) {
|
||||
try { fs.rmSync(dir, { recursive: true, force: true }); } catch { /* ignore */ }
|
||||
}
|
||||
|
||||
// ─────────────────────────────────────────────────────────────────────────────
|
||||
// Test
|
||||
// ─────────────────────────────────────────────────────────────────────────────
|
||||
|
||||
describe('perf #316: acquireStateLock hoists sleep buffer — exactly one SAB per call', () => {
|
||||
let tmpDir;
|
||||
let statePath;
|
||||
let lockPath;
|
||||
|
||||
beforeEach(() => {
|
||||
tmpDir = makeTempDir();
|
||||
statePath = path.join(tmpDir, '.planning', 'STATE.md');
|
||||
lockPath = statePath + '.lock';
|
||||
fs.writeFileSync(statePath, MINIMAL_STATE_MD, 'utf-8');
|
||||
});
|
||||
|
||||
afterEach(() => {
|
||||
try { fs.unlinkSync(lockPath); } catch { /* already gone */ }
|
||||
removeTempDir(tmpDir);
|
||||
});
|
||||
|
||||
test(
|
||||
'sabCount === 1 after a call that undergoes >= 1 retry (post-fix assertion)',
|
||||
{ timeout: 8000 },
|
||||
async () => {
|
||||
// ── Worker A: hold the lock for 400ms ──────────────────────────────────
|
||||
// retryDelay = 200ms + up to 50ms jitter, so >= 1 retry fires before
|
||||
// Worker A releases at 400ms.
|
||||
const holdMs = 400;
|
||||
let holderWorker;
|
||||
const holderDone = new Promise((resolve, reject) => {
|
||||
holderWorker = new Worker(HOLDER_WORKER_CODE, {
|
||||
eval: true,
|
||||
workerData: { lockPath, holdMs },
|
||||
});
|
||||
holderWorker.on('message', (msg) => { if (msg.done) resolve(); });
|
||||
holderWorker.on('error', reject);
|
||||
holderWorker.on('exit', (code) => {
|
||||
if (code !== 0) reject(new Error('Holder worker exit code: ' + code));
|
||||
});
|
||||
});
|
||||
|
||||
// Give Worker A 80ms to write the lock file before starting Worker B.
|
||||
await new Promise(resolve => setTimeout(resolve, 80));
|
||||
assert.ok(fs.existsSync(lockPath), 'Worker A must have written the lock file');
|
||||
|
||||
// ── Worker B: call writeStateMd, measure SAB allocations ───────────────
|
||||
const writeResult = await new Promise((resolve, reject) => {
|
||||
const writer = new Worker(WRITER_WORKER_CODE, {
|
||||
eval: true,
|
||||
workerData: {
|
||||
stateCjsPath: STATE_CJS_PATH,
|
||||
statePath,
|
||||
content: MINIMAL_STATE_MD,
|
||||
tmpDir,
|
||||
},
|
||||
});
|
||||
writer.on('message', resolve);
|
||||
writer.on('error', reject);
|
||||
writer.on('exit', (code) => {
|
||||
if (code !== 0) reject(new Error('Writer worker exit code: ' + code));
|
||||
});
|
||||
});
|
||||
|
||||
// Wait for Worker A to finish releasing
|
||||
await holderDone;
|
||||
|
||||
// ── Assertions ─────────────────────────────────────────────────────────
|
||||
assert.ok(
|
||||
writeResult.callErr === null,
|
||||
'writeStateMd must succeed once the lock is released — error: ' + writeResult.callErr
|
||||
);
|
||||
|
||||
assert.ok(
|
||||
writeResult.sabCount >= 1,
|
||||
'at least one SharedArrayBuffer must be allocated (the sleep buffer must exist)'
|
||||
);
|
||||
|
||||
// THE KEY INVARIANT:
|
||||
// POST-FIX: sabCount === 1 (buffer allocated once, before the retry loop)
|
||||
// PRE-FIX: sabCount >= 2 (new buffer on EVERY retry iteration)
|
||||
//
|
||||
// With a 400ms hold and 200ms retryDelay, the pre-fix code observes sabCount=2.
|
||||
// (Confirmed pre-fix RED: sabCount=2 with holdMs=400.)
|
||||
assert.strictEqual(
|
||||
writeResult.sabCount,
|
||||
1,
|
||||
'post-fix: exactly one SharedArrayBuffer must be allocated per acquireStateLock call ' +
|
||||
'(buffer hoisted before retry loop). Got: ' + writeResult.sabCount
|
||||
);
|
||||
}
|
||||
);
|
||||
});
|
||||
Reference in New Issue
Block a user