From 6ceebb2cb2f8fa8d373cf93dedd668d2477caaf6 Mon Sep 17 00:00:00 2001
From: wangqing <wangqing@uniontech.com>
Date: Fri, 10 Oct 2025 17:43:05 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20=E4=BC=98=E9=9B=85=E5=85=B3=E9=97=AD?=
 =?UTF-8?q?=E6=9C=8D=E5=8A=A1=E5=99=A8=E7=9A=84=E5=8A=9F=E8=83=BD?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Signed-off-by: wangqing <wangqing@uniontech.com>
---
 oedp-mcp/mcp-oedp/mcp-oedp.py | 14 +++++++++++++-
 1 file changed, 13 insertions(+), 1 deletion(-)

diff --git a/oedp-mcp/mcp-oedp/mcp-oedp.py b/oedp-mcp/mcp-oedp/mcp-oedp.py
index 9d4a723..6e1a528 100644
--- a/oedp-mcp/mcp-oedp/mcp-oedp.py
+++ b/oedp-mcp/mcp-oedp/mcp-oedp.py
@@ -17,6 +17,8 @@ import json
 import os
 import subprocess
 import yaml
+import signal
+import sys
 from typing import Optional, Dict, Any
 from openai import OpenAI
 from mcp.server.fastmcp import FastMCP
@@ -544,8 +546,18 @@ SYSTEM_CONTENT = """你现在是一名资深的软件工程师,你熟悉多种
 你擅长解决技术问题,并具有优秀的逻辑思维能力.请在这个角色下为我解答以下问题.openEuler是我默认的Linux开发环境."""
 
 
+def signal_handler(sig, frame):
+    """处理信号，优雅关闭服务器"""
+    print("Shutting down server gracefully...")
+    sys.exit(0)
+
+
 def main():
     """主函数"""
+    # 注册信号处理器以实现优雅关闭
+    signal.signal(signal.SIGINT, signal_handler)
+    signal.signal(signal.SIGTERM, signal_handler)
+    
     # 解析命令行参数
     parser = argparse.ArgumentParser(description='oeDeploy MCP Server')
     parser.add_argument('--model_url', required=True, help='Model url')
@@ -582,4 +594,4 @@ def main():
 
 
 if __name__ == "__main__":
-    main()
+    main()
\ No newline at end of file
-- 
Gitee