fix: shutdown
This commit is contained in:
parent
e5ce8c20b5
commit
71ab7523da
|
|
@ -471,17 +471,63 @@ async function mcpServeAction(host: string, options: MCPOptions) {
|
|||
await server.connect(transport);
|
||||
console.error('TTRPG 卡牌生成 MCP 服务器已启动(stdio)');
|
||||
|
||||
// 监听进程退出事件,关闭服务器
|
||||
process.on('SIGINT', () => {
|
||||
// 清理函数
|
||||
let isShuttingDown = false;
|
||||
async function shutdown() {
|
||||
if (isShuttingDown) return;
|
||||
isShuttingDown = true;
|
||||
|
||||
console.error('\n正在关闭服务器...');
|
||||
|
||||
try {
|
||||
// 先关闭服务器,这会触发 transport 的关闭
|
||||
await server.close();
|
||||
console.error('MCP 服务器已关闭');
|
||||
} catch (e) {
|
||||
console.error('关闭 MCP 服务器时出错:', e);
|
||||
}
|
||||
|
||||
try {
|
||||
// 确保 transport 被关闭
|
||||
await transport.close();
|
||||
console.error('MCP Transport 已关闭');
|
||||
} catch (e) {
|
||||
console.error('关闭 MCP Transport 时出错:', e);
|
||||
}
|
||||
|
||||
try {
|
||||
// 关闭内容服务器
|
||||
contentServer.close();
|
||||
console.error('内容服务器已关闭');
|
||||
} catch (e) {
|
||||
console.error('关闭内容服务器时出错:', e);
|
||||
}
|
||||
|
||||
// 给异步清理一些时间
|
||||
setTimeout(() => {
|
||||
process.exit(0);
|
||||
}, 100);
|
||||
}
|
||||
|
||||
// 监听进程退出事件
|
||||
process.on('SIGINT', () => shutdown());
|
||||
process.on('SIGTERM', () => shutdown());
|
||||
|
||||
// Windows 上 Ctrl+C 会触发 'SIGINT',但也监听 'exit' 事件作为后备
|
||||
process.on('exit', () => {
|
||||
contentServer.close();
|
||||
process.exit(0);
|
||||
});
|
||||
|
||||
process.on('SIGTERM', () => {
|
||||
console.error('\n正在关闭服务器...');
|
||||
contentServer.close();
|
||||
process.exit(0);
|
||||
// 监听 stdin 结束事件(当客户端关闭连接时)
|
||||
process.stdin.on('end', () => {
|
||||
console.error('客户端已断开连接,正在关闭...');
|
||||
shutdown();
|
||||
});
|
||||
|
||||
// 监听 stdin 关闭事件
|
||||
process.stdin.on('close', () => {
|
||||
console.error('stdin 已关闭,正在关闭服务器...');
|
||||
shutdown();
|
||||
});
|
||||
} else {
|
||||
// TODO: 支持 HTTP 传输
|
||||
|
|
|
|||
Loading…
Reference in New Issue