Function acquireProcessLock

src/utils/process-lock.ts:17–60

(params: {
  lockDirPath: string;
  owner: ProcessLockOwner;
  timeoutMs?: number;
  pollMs?: number;
  ownerGraceMs?: number;
  description?: string;
})

};

export async function acquireProcessLock(params: {
  lockDirPath: string;
  owner: ProcessLockOwner;
  timeoutMs?: number;
  pollMs?: number;
  ownerGraceMs?: number;
  description?: string;
}): Promise<() => Promise<void>> {
  const { lockDirPath, owner } = params;
  const ownerFilePath = path.join(lockDirPath, 'owner.json');
  const deadline = Date.now() + (params.timeoutMs ?? DEFAULT_LOCK_TIMEOUT_MS);
  const pollMs = params.pollMs ?? DEFAULT_LOCK_POLL_MS;
  const ownerGraceMs = params.ownerGraceMs ?? DEFAULT_LOCK_OWNER_GRACE_MS;
  const description = params.description ?? 'process lock';

  fs.mkdirSync(path.dirname(lockDirPath), { recursive: true });

  while (Date.now() < deadline) {
    try {
      fs.mkdirSync(lockDirPath);
      writeProcessLockOwner(ownerFilePath, owner);
      let released = false;
      return async () => {
        if (released) return;
        released = true;
        fs.rmSync(lockDirPath, { recursive: true, force: true });
      };
    } catch (error) {
      const err = error as NodeJS.ErrnoException;
      if (err.code !== 'EEXIST') {
        throw err;
      }
      if (clearStaleProcessLock(lockDirPath, ownerFilePath, ownerGraceMs)) {
        continue;
      }
      await sleep(pollMs);
    }
  }

  throw new AppError('COMMAND_FAILED', `Timed out waiting for ${description}`, {
    lockDirPath,
    ...readProcessLockDiagnostics(lockDirPath, ownerFilePath),
  });
}

function writeProcessLockOwner(ownerFilePath: string, owner: ProcessLockOwner): void {
  const tmpOwnerFilePath = `${ownerFilePath}.${process.pid}.${Date.now()}.tmp`;