包装了 eventlet.greenthread.main 函数,捕获所有绿色线程中的异常

This commit is contained in:
CakeCN
2026-01-08 00:10:33 +08:00
parent a07eee1ffc
commit 6ecd0c6735

View File

@@ -38,7 +38,28 @@ try:
error_counts[error_key] = 1 error_counts[error_key] = 1
return False return False
# 4. 只修改关键的 greenio 方法,确保在致命错误时彻底关闭连接 # 4. 增强错误处理,确保所有绿色线程错误都被捕获
# 保存原始的 greenlet 主函数,包装它以捕获所有异常
import eventlet.greenthread
original_main = eventlet.greenthread.main
def custom_greenthread_main(*args, **kwargs):
"""包装 greenlet 主函数,捕获所有异常,防止服务器宕机"""
try:
return original_main(*args, **kwargs)
except Exception as e:
# 捕获所有绿色线程中的异常
if not is_duplicate_error(e):
print(f"[Eventlet GreenThread Error] {type(e).__name__}: {e}")
print("[Traceback]")
traceback.print_exc()
# 返回 None 继续运行,防止服务器宕机
return None
# 替换原始的 greenlet 主函数
eventlet.greenthread.main = custom_greenthread_main
# 5. 只修改关键的 greenio 方法,确保在致命错误时彻底关闭连接
from eventlet import greenio from eventlet import greenio
# 保存并替换原始的 greenio 方法 # 保存并替换原始的 greenio 方法