mirror of
https://github.com/paricafe/misskey.git
synced 2024-12-02 03:36:44 -06:00
24 lines
672 B
TypeScript
24 lines
672 B
TypeScript
|
/*
|
||
|
|
* SPDX-FileCopyrightText: syuilo and other misskey contributors
|
||
|
|
* SPDX-License-Identifier: AGPL-3.0-only
|
||
|
|
*/
|
||
|
|
|
||
|
|
const promiseRefs: Set<WeakRef<Promise<unknown>>> = new Set();
|
||
|
|
|
||
|
|
/**
|
||
|
|
* This tracks promises that other modules decided not to wait for,
|
||
|
|
* and makes sure they are all settled before fully closing down the server.
|
||
|
|
*/
|
||
|
|
export function trackPromise(promise: Promise<unknown>) {
|
||
|
|
if (process.env.NODE_ENV !== 'test') {
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
const ref = new WeakRef(promise);
|
||
|
|
promiseRefs.add(ref);
|
||
|
|
promise.finally(() => promiseRefs.delete(ref));
|
||
|
|
}
|
||
|
|
|
||
|
|
export async function allSettled(): Promise<void> {
|
||
|
|
await Promise.allSettled([...promiseRefs].map(r => r.deref()));
|
||
|
|
}
|