From 40a0eb80e6cd25d099d3b679785dd88d090cd342 Mon Sep 17 00:00:00 2001
From: "yifeng.wang" <3038880699@qq.com>
Date: Fri, 14 Mar 2025 12:43:50 +0800
Subject: [PATCH] optimize stop
---
owl/webapp_zh.py | 69 +++++++++++++++---------------------------------
1 file changed, 22 insertions(+), 47 deletions(-)
diff --git a/owl/webapp_zh.py b/owl/webapp_zh.py
index 29998d0..659e177 100644
--- a/owl/webapp_zh.py
+++ b/owl/webapp_zh.py
@@ -248,56 +248,32 @@ def validate_input(question: str) -> bool:
# 终止运行进程的函数
def terminate_process():
- """终止当前运行的进程，适配不同操作系统平台"""
+ """停止当前运行的线程，而不是终止整个进程"""
global CURRENT_PROCESS, STOP_REQUESTED
if CURRENT_PROCESS is None:
- logging.info("没有正在运行的进程")
- return "没有正在运行的进程", " 已就绪"
+ logging.info("没有正在运行的线程")
+ return "没有正在运行的线程", " 已就绪"
try:
STOP_REQUESTED.set() # 设置停止标志
- logging.info("正在尝试终止进程...")
+ logging.info("已设置停止标志，正在等待线程响应...")
- # 获取当前操作系统
- current_os = platform.system()
-
- if current_os == "Windows":
- # Windows平台使用taskkill强制终止进程树
- if hasattr(CURRENT_PROCESS, 'pid'):
- subprocess.run(f"taskkill /F /T /PID {CURRENT_PROCESS.pid}", shell=True)
- logging.info(f"已发送Windows终止命令到进程 {CURRENT_PROCESS.pid}")
- else:
- # Unix-like系统 (Linux, macOS)
- if hasattr(CURRENT_PROCESS, 'pid'):
- # 发送SIGTERM信号
- os.killpg(os.getpgid(CURRENT_PROCESS.pid), signal.SIGTERM)
- logging.info(f"已发送SIGTERM信号到进程组 {CURRENT_PROCESS.pid}")
-
- # 给进程一些时间来清理
- time.sleep(0.5)
-
- # 如果进程仍在运行，发送SIGKILL
- try:
- if CURRENT_PROCESS.poll() is None:
- os.killpg(os.getpgid(CURRENT_PROCESS.pid), signal.SIGKILL)
- logging.info(f"已发送SIGKILL信号到进程组 {CURRENT_PROCESS.pid}")
- except (ProcessLookupError, OSError):
- pass # 进程可能已经终止
-
- # 如果是线程，尝试终止线程
+ # 如果是线程，只需要设置标志让它自行停止
if isinstance(CURRENT_PROCESS, threading.Thread) and CURRENT_PROCESS.is_alive():
- # 线程无法强制终止，但可以设置标志让线程自行退出
- logging.info("等待线程终止...")
- CURRENT_PROCESS.join(timeout=2)
-
- CURRENT_PROCESS = None
- logging.info("进程已终止")
- return "进程已终止", " 已就绪"
+ logging.info("等待线程处理停止请求...")
+ # 不强制终止线程，只设置标志位让线程自行退出
+ # 线程应该会定期检查STOP_REQUESTED标志
+ return "已请求停止生成", " 正在停止..."
+ else:
+ # 如果不是线程或线程已经不活跃，则重置状态
+ CURRENT_PROCESS = None
+ logging.info("线程已不活跃")
+ return "线程已停止", " 已就绪"
except Exception as e:
- logging.error(f"终止进程时出错: {str(e)}")
- return f"终止进程时出错: {str(e)}", f" 错误: {str(e)}"
+ logging.error(f"停止线程时出错: {str(e)}")
+ return f"停止线程时出错: {str(e)}", f" 错误: {str(e)}"
def run_owl(question: str, example_module: str) -> Tuple[str, List[List[str]], str, str]:
"""运行OWL系统并返回结果
@@ -577,6 +553,9 @@ def create_ui():
"""处理问题并实时更新日志和对话历史"""
global CURRENT_PROCESS, STOP_REQUESTED, CONVERSATION_UPDATE_QUEUE
+ # 重置停止标志
+ STOP_REQUESTED.clear()
+
# 创建一个后台线程来处理问题
result_queue = queue.Queue()
# 创建一个队列用于实时更新对话历史
@@ -696,7 +675,7 @@ def create_ui():
# 检查是否已请求停止
if STOP_REQUESTED.is_set():
logs = get_latest_logs(100)
- yield None, current_chat_history, None, " 正在终止...", logs
+ yield "操作已取消", current_chat_history, "0", " 正在停止...", logs
break
# 检查是否有新的对话更新（从日志解析）
@@ -754,9 +733,9 @@ def create_ui():
# 如果已请求停止但线程仍在运行
if STOP_REQUESTED.is_set() and bg_thread.is_alive():
- bg_thread.join(timeout=2) # 等待线程最多2秒
+ # 不再强制join线程，让它自然结束
logs = get_latest_logs(100)
- yield "操作已取消", current_chat_history, "0", " 已终止", logs
+ yield "生成已停止", current_chat_history, "0", " 已停止生成", logs
return
# 处理完成，获取结果
@@ -1179,8 +1158,6 @@ def main():
global STOP_LOG_THREAD, STOP_REQUESTED
STOP_LOG_THREAD.set()
STOP_REQUESTED.set()
- # 尝试终止当前进程
- terminate_process()
logging.info("应用程序关闭，停止日志线程")
app.launch(share=False,enable_queue=True,server_name="127.0.0.1",server_port=7860)
@@ -1194,8 +1171,6 @@ def main():
# 确保日志线程停止
STOP_LOG_THREAD.set()
STOP_REQUESTED.set()
- # 尝试终止当前进程
- terminate_process()
logging.info("应用程序关闭")
if __name__ == "__main__":