mirror/cherrypick
1
1
Fork 0
mirror of https://github.com/kokonect-link/cherrypick synced 2024-11-23 14:46:44 +09:00
Code

Graceful Shutdown (MisskeyIO#156)

Browse Source 
---------

Co-authored-by: まっちゃとーにゅ <17376330+u1-liquid@users.noreply.github.com>
This commit is contained in:
riku6460 2023-08-22 21:42:21 +09:00 committed by NoriDev
parent d5e6bc68a2
commit 1217790667
5 changed files with 38 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
Dockerfile
View File

@ -84,4 +84,4 @@ COPY --chown=cherrypick:cherrypick . ./
ENV NODE_ENV=production
HEALTHCHECK --interval=5s --retries=20 CMD ["/bin/bash", "/cherrypick/healthcheck.sh"]
ENTRYPOINT ["/usr/bin/tini", "--"]
CMD ["pnpm", "run", "migrateandstart"]
CMD ["pnpm", "run", "migrateandstart:docker"]

2
package.json
View File

@ -18,11 +18,13 @@
"build": "pnpm build-pre && pnpm -r build && pnpm gulp",
"build-storybook": "pnpm --filter frontend build-storybook",
"start": "pnpm check:connect && cd packages/backend && node ./built/boot/index.js",
"start:docker": "pnpm check:connect && cd packages/backend && exec node ./built/boot/index.js",
"start:test": "cd packages/backend && cross-env NODE_ENV=test node ./built/boot/index.js",
"init": "pnpm migrate",
"migrate": "cd packages/backend && pnpm migrate",
"check:connect": "cd packages/backend && pnpm check:connect",
"migrateandstart": "pnpm migrate && pnpm start",
"migrateandstart:docker": "pnpm migrate && exec pnpm start:docker",
"gulp": "pnpm exec gulp build",
"watch": "pnpm dev",
"dev": "node ./scripts/dev.mjs",

46
packages/backend/src/boot/index.ts
View File

@ -29,23 +29,39 @@ const ev = new Xev();
//#region Events
// Listen new workers
cluster.on('fork', worker => {
clusterLogger.debug(`Process forked: [${worker.id}]`);
});
if (cluster.isPrimary && !envOption.disableClustering) {
// Listen new workers
cluster.on('fork', worker => {
clusterLogger.debug(`Process forked: [${worker.id}]`);
});
// Listen online workers
cluster.on('online', worker => {
clusterLogger.debug(`Process is now online: [${worker.id}]`);
});
// Listen online workers
cluster.on('online', worker => {
clusterLogger.debug(`Process is now online: [${worker.id}]`);
});
// Listen for dying workers
cluster.on('exit', worker => {
// Replace the dead worker,
// we're not sentimental
clusterLogger.error(chalk.red(`[${worker.id}] died :(`));
cluster.fork();
});
// Listen for dying workers
cluster.on('exit', (worker, code, signal?) => {
// Replace the dead worker,
// we're not sentimental
if (signal) {
switch (signal) {
case 'SIGINT':
case 'SIGTERM':
console.log(chalk.green(`[${worker.id}] exited by signal: ${signal}`));
break;
default:
console.error(chalk.red(`[${worker.id}] killed by signal: ${signal}`));
cluster.fork();
break;
}
} else if (code !== 0) {
console.error(chalk.red(`[${worker.id}] exited with error code: ${code}`));
} else {
console.log(chalk.green(`[${worker.id}] exited normally`));
}
});
}
// Display detail of unhandled promise rejection
if (!envOption.quiet) {

2
packages/backend/src/server/ServerService.ts
View File

@ -250,7 +250,7 @@ export class ServerService implements OnApplicationShutdown {
@bindThis
public async dispose(): Promise<void> {
await this.streamingApiServerService.detach();
this.streamingApiServerService.detach();
await this.#fastify.close();
}

7
packages/backend/src/server/api/StreamingApiServerService.ts
View File

@ -168,13 +168,12 @@ export class StreamingApiServerService {
}
@bindThis
public detach(): Promise<void> {
public detach(): void {
if (this.#cleanConnectionsIntervalId) {
clearInterval(this.#cleanConnectionsIntervalId);
this.#cleanConnectionsIntervalId = null;
}
return new Promise((resolve) => {
this.#wss.close(() => resolve());
});
this.#wss.close();
this.#wss.clients.forEach(client => client.terminate());
}
}