205 lines
7.6 KiB
Python
205 lines
7.6 KiB
Python
import eventlet
|
|
import logging
|
|
|
|
# 配置 eventlet 日志，抑制 IOClosed 等错误日志
|
|
logging.getLogger('eventlet.hubs').setLevel(logging.CRITICAL)
|
|
logging.getLogger('eventlet.greenio').setLevel(logging.CRITICAL)
|
|
|
|
# 进行猴子补丁操作
|
|
eventlet.monkey_patch()
|
|
|
|
# 配置 eventlet 以更优雅地处理错误，避免疯狂报错
|
|
try:
|
|
import eventlet
|
|
import traceback
|
|
|
|
# 1. 首先，禁用 eventlet 的调试日志，减少输出
|
|
import logging
|
|
logging.getLogger('eventlet').setLevel(logging.CRITICAL)
|
|
logging.getLogger('socketio').setLevel(logging.CRITICAL)
|
|
logging.getLogger('werkzeug').setLevel(logging.CRITICAL)
|
|
|
|
# 2. 配置 eventlet 只记录严重错误
|
|
eventlet.debug.hub_prevent_multiple_readers(False)
|
|
|
|
# 3. 避免重复打印相同错误的机制
|
|
seen_errors = set()
|
|
max_error_logs = 5 # 每个错误最多打印5次
|
|
error_counts = {}
|
|
|
|
def is_duplicate_error(e):
|
|
"""检查是否为重复错误"""
|
|
error_key = f"{type(e).__name__}: {e}"
|
|
if error_key in seen_errors:
|
|
error_counts[error_key] = error_counts.get(error_key, 0) + 1
|
|
if error_counts[error_key] > max_error_logs:
|
|
return True
|
|
else:
|
|
seen_errors.add(error_key)
|
|
error_counts[error_key] = 1
|
|
return False
|
|
|
|
# 4. 只修改关键的 greenio 方法，确保在致命错误时彻底关闭连接
|
|
from eventlet import greenio
|
|
|
|
# 保存并替换原始的 greenio 方法
|
|
if hasattr(greenio.base.GreenSocket, '_recv_loop'):
|
|
original_recv_loop = greenio.base.GreenSocket._recv_loop
|
|
|
|
def custom_recv_loop(self, recv_func, recv_args, *args, **kwargs):
|
|
try:
|
|
return original_recv_loop(self, recv_func, recv_args, *args, **kwargs)
|
|
except Exception as e:
|
|
if not is_duplicate_error(e):
|
|
print(f"[Eventlet IO Error] _recv_loop: {type(e).__name__}: {e}")
|
|
print("[Traceback]")
|
|
traceback.print_exc()
|
|
# 彻底关闭连接，不再继续处理
|
|
try:
|
|
self.close()
|
|
except Exception:
|
|
pass
|
|
return b''
|
|
|
|
greenio.base.GreenSocket._recv_loop = custom_recv_loop
|
|
|
|
if hasattr(greenio.base.GreenSocket, 'recv_into'):
|
|
original_recv_into = greenio.base.GreenSocket.recv_into
|
|
|
|
def custom_recv_into(self, buffer, nbytes=0, flags=0):
|
|
try:
|
|
return original_recv_into(self, buffer, nbytes, flags)
|
|
except Exception as e:
|
|
if not is_duplicate_error(e):
|
|
print(f"[Eventlet IO Error] recv_into: {type(e).__name__}: {e}")
|
|
print("[Traceback]")
|
|
traceback.print_exc()
|
|
# 彻底关闭连接，不再继续处理
|
|
try:
|
|
self.close()
|
|
except Exception:
|
|
pass
|
|
return 0
|
|
|
|
greenio.base.GreenSocket.recv_into = custom_recv_into
|
|
|
|
if hasattr(greenio.base.GreenSocket, '_send_loop'):
|
|
original_send_loop = greenio.base.GreenSocket._send_loop
|
|
|
|
def custom_send_loop(self, send_func, data, *args, **kwargs):
|
|
try:
|
|
return original_send_loop(self, send_func, data, *args, **kwargs)
|
|
except Exception as e:
|
|
if not is_duplicate_error(e):
|
|
print(f"[Eventlet IO Error] _send_loop: {type(e).__name__}: {e}")
|
|
print("[Traceback]")
|
|
traceback.print_exc()
|
|
# 彻底关闭连接，不再继续处理
|
|
try:
|
|
self.close()
|
|
except Exception:
|
|
pass
|
|
return 0
|
|
|
|
greenio.base.GreenSocket._send_loop = custom_send_loop
|
|
|
|
if hasattr(greenio.base.GreenSocket, 'send'):
|
|
original_send = greenio.base.GreenSocket.send
|
|
|
|
def custom_send(self, data, flags=0):
|
|
try:
|
|
return original_send(self, data, flags)
|
|
except Exception as e:
|
|
if not is_duplicate_error(e):
|
|
print(f"[Eventlet IO Error] send: {type(e).__name__}: {e}")
|
|
print("[Traceback]")
|
|
traceback.print_exc()
|
|
# 彻底关闭连接，不再继续处理
|
|
try:
|
|
self.close()
|
|
except Exception:
|
|
pass
|
|
return 0
|
|
|
|
greenio.base.GreenSocket.send = custom_send
|
|
|
|
# 5. 处理 WSGI 中的错误，确保彻底清理
|
|
try:
|
|
import eventlet.wsgi
|
|
|
|
if hasattr(eventlet.wsgi.HttpProtocol, 'handle_one_request'):
|
|
original_wsgi_handle_one_request = eventlet.wsgi.HttpProtocol.handle_one_request
|
|
|
|
def custom_wsgi_handle_one_request(self):
|
|
try:
|
|
return original_wsgi_handle_one_request(self)
|
|
except Exception as e:
|
|
if not is_duplicate_error(e):
|
|
print(f"[Eventlet WSGI Request Error] {type(e).__name__}: {e}")
|
|
print("[Traceback]")
|
|
traceback.print_exc()
|
|
# 彻底清理连接，不再继续
|
|
try:
|
|
self.rfile.close()
|
|
self.wfile.close()
|
|
self.finish()
|
|
except Exception:
|
|
pass
|
|
return
|
|
|
|
eventlet.wsgi.HttpProtocol.handle_one_request = custom_wsgi_handle_one_request
|
|
|
|
if hasattr(eventlet.wsgi.HttpProtocol, 'handle_one_response'):
|
|
original_wsgi_handle_one_response = eventlet.wsgi.HttpProtocol.handle_one_response
|
|
|
|
def custom_wsgi_handle_one_response(self):
|
|
try:
|
|
return original_wsgi_handle_one_response(self)
|
|
except Exception as e:
|
|
if not is_duplicate_error(e):
|
|
print(f"[Eventlet WSGI Response Error] {type(e).__name__}: {e}")
|
|
print("[Traceback]")
|
|
traceback.print_exc()
|
|
# 彻底清理连接，不再继续
|
|
try:
|
|
self.rfile.close()
|
|
self.wfile.close()
|
|
self.finish()
|
|
except Exception:
|
|
pass
|
|
return
|
|
|
|
eventlet.wsgi.HttpProtocol.handle_one_response = custom_wsgi_handle_one_response
|
|
except Exception as e:
|
|
# 如果修改失败，忽略错误
|
|
print(f"[WSGI Patch Error] {e}")
|
|
|
|
print("[Eventlet Error Handling] Configured to handle errors gracefully, avoiding duplicate logs")
|
|
except Exception as e:
|
|
# 如果修改失败，打印错误但继续运行
|
|
print(f"[Eventlet Patch Initialization Error] {e}")
|
|
traceback.print_exc()
|
|
|
|
from flask import Flask
|
|
from .views.routes import main_bp
|
|
from .views.file import file_bp
|
|
from .config import Config
|
|
from .extensions import init_extensions,register_namespaces
|
|
|
|
def create_app():
|
|
app = Flask(__name__, template_folder="../templates", static_folder="../static",static_url_path="/vsc-like/static")
|
|
app.config.from_object(Config)
|
|
@app.route('/')
|
|
def hello():
|
|
return 'Hello, World!'
|
|
|
|
# 注册蓝图
|
|
app.register_blueprint(main_bp)
|
|
app.register_blueprint(file_bp)
|
|
# 初始化SocketIO
|
|
init_extensions(app)
|
|
|
|
register_namespaces(app)
|
|
|
|
return app
|