CodeCow/stack
1
0
Fork 0
mirror of https://github.com/stack-auth/stack.git synced 2026-06-04 21:04:37 +08:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

added withPreviewPoolFillMutex and wrapped the refill sequence in fillPreviewPool. The mutex releases in finally, including on errors.

Browse Source
This commit is contained in:
Madison 2026-05-28 12:18:08 -05:00
parent e0d97388c0
commit ae8deed04c
1 changed files with 30 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

42
apps/backend/src/lib/preview-pool.ts
View File

@ -17,6 +17,22 @@ const DEFAULT_READY_POOL_SIZE = 3;
const DEFAULT_LEASE_DURATION_MS = 30 * 60 * 1000;
const DEFAULT_CLEANUP_MAX_DELETE_PER_RUN = 1;
let previewPoolFillMutexTail: Promise<void> = Promise.resolve();
async function withPreviewPoolFillMutex<T>(callback: () => Promise<T>): Promise<T> {
const previousTail = previewPoolFillMutexTail;
let releaseMutex!: () => void;
previewPoolFillMutexTail = new Promise<void>((resolve) => {
releaseMutex = resolve;
});
await previousTail;
try {
return await callback();
} finally {
releaseMutex();
}
}
type PreviewPoolState = "ready" | "leased";
type PreviewPoolMetadata = {
@ -546,20 +562,22 @@ export async function fillPreviewPool(options?: { maxCreate?: number }): Promise
}> {
assertPreviewModeEnabled();
const targetReadyCount = getReadyPoolSize();
const readyCountBefore = await countReadyPreviewPoolProjects();
const requestedMaxCreate = options?.maxCreate ?? 1;
const createCount = Math.max(0, Math.min(requestedMaxCreate, targetReadyCount - readyCountBefore));
return await withPreviewPoolFillMutex(async () => {
const targetReadyCount = getReadyPoolSize();
const readyCountBefore = await countReadyPreviewPoolProjects();
const requestedMaxCreate = options?.maxCreate ?? 1;
const createCount = Math.max(0, Math.min(requestedMaxCreate, targetReadyCount - readyCountBefore));
for (let i = 0; i < createCount; i++) {
await createPreviewPoolProject("ready");
}
for (let i = 0; i < createCount; i++) {
await createPreviewPoolProject("ready");
}
return {
readyCountBefore,
createdCount: createCount,
targetReadyCount,
};
return {
readyCountBefore,
createdCount: createCount,
targetReadyCount,
};
});
}
export async function claimPreviewPoolLease(options: { apiUrl: string }): Promise<PreviewPoolLease> {