42 lines
1013 B
JavaScript
42 lines
1013 B
JavaScript
const { closeDatabase } = require('../database');
|
|
|
|
function registerShutdown(server) {
|
|
let isShuttingDown = false;
|
|
|
|
async function shutdown(reason) {
|
|
if (isShuttingDown) return;
|
|
isShuttingDown = true;
|
|
|
|
if (reason) {
|
|
console.log(`\n收到 ${reason}，正在关闭服务器...`);
|
|
}
|
|
|
|
await new Promise((resolve) => {
|
|
server.close(() => {
|
|
console.log('HTTP 服务器已关闭');
|
|
resolve();
|
|
});
|
|
});
|
|
|
|
await closeDatabase();
|
|
process.exit(reason === 'uncaughtException' || reason === 'unhandledRejection' ? 1 : 0);
|
|
}
|
|
|
|
process.on('SIGINT', () => shutdown('SIGINT'));
|
|
process.on('SIGTERM', () => shutdown('SIGTERM'));
|
|
|
|
process.on('uncaughtException', (err) => {
|
|
console.error('未捕获的异常:', err);
|
|
shutdown('uncaughtException');
|
|
});
|
|
|
|
process.on('unhandledRejection', (reason) => {
|
|
console.error('未处理的 Promise 拒绝:', reason);
|
|
shutdown('unhandledRejection');
|
|
});
|
|
}
|
|
|
|
module.exports = {
|
|
registerShutdown
|
|
};
|