HoovesSocial/hippofish
2
0
Fork 0
Code Issues 3 Pull requests 1 Projects Releases Packages Wiki Activity Actions

refactor (backend): clean the booting process a bit

Browse source
This commit is contained in:
naskya 2024-04-20 02:58:39 +09:00
parent e82a07c03d
commit 037b7950a2
No known key found for this signature in database
GPG key ID: 712D413B3A9FED5C
1 changed files with 16 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

24
packages/backend/src/boot/master.ts
View file

@ -169,23 +169,31 @@ async function connectDb(): Promise<void> {
} }
async function spawnWorkers( async function spawnWorkers(
clusterLimits: Required<Config["clusterLimits"]>, clusterLimits: Config["clusterLimits"],
): Promise<void> { ): Promise<void> {
const modes = ["web", "queue"];
const cpus = os.cpus().length; const cpus = os.cpus().length;
for (const mode of modes.filter((mode) => clusterLimits[mode] > cpus)) {
if (clusterLimits.queue > cpus) {
bootLogger.warn( bootLogger.warn(
`configuration warning: cluster limit for ${mode} exceeds number of cores (${cpus})`, "config: queue cluster limit exceeds the number of cpu cores",
); );
} }
const total = modes.reduce((acc, mode) => acc + clusterLimits[mode], 0); if (clusterLimits.web > cpus) {
bootLogger.warn(
"config: web cluster limit exceeds the number of cpu cores",
);
}
const total = clusterLimits.queue + clusterLimits.web;
// workers = ["web", "web", ..., "web", "queue", "queue", ..., "queue"]
const workers = new Array(total); const workers = new Array(total);
workers.fill("web", 0, clusterLimits?.web); workers.fill("web", 0, clusterLimits.web);
workers.fill("queue", clusterLimits?.web); workers.fill("queue", clusterLimits.web);
bootLogger.info( bootLogger.info(
`Starting ${clusterLimits?.web} web workers and ${clusterLimits?.queue} queue workers (total ${total})...`, `Starting ${clusterLimits.web} web workers and ${clusterLimits.queue} queue workers (total ${total})...`,
); );
await Promise.all(workers.map((mode) => spawnWorker(mode))); await Promise.all(workers.map((mode) => spawnWorker(mode)));
bootLogger.succ("All workers started"); bootLogger.succ("All workers started");