fix module 'eventlet.hubs' has no attribute 'handle_error'

code-server-like/app/__init__.py

@@ -14,21 +14,11 @@ try:
     import eventlet
     import traceback
     
-    # 获取当前的 hub 实例
+    # 只修改存在的属性，避免引用错误
-    current_hub = hubs.get_hub()
+    if hasattr(hubs, 'trampoline'):
-    
+        # 保存原始的trampoline方法
-    # 保存原始的方法
-    original_handle_error = hubs.handle_error
         original_trampoline = hubs.trampoline
         
-    def custom_handle_error(context, type, value, tb):
-        # 捕获并打印所有 eventlet.hubs 相关错误，不再往外raise
-        print(f"[Eventlet Hub Error] Type: {type.__name__}, Value: {value}")
-        print(f"[Context] {context}")
-        print("[Traceback]")
-        traceback.print_tb(tb)
-        # 不再调用原始方法，直接返回，不往外传播错误
-    
         def custom_trampoline(fd, read=False, write=False, timeout=None, timeout_exc=None):
             try:
                 return original_trampoline(fd, read, write, timeout, timeout_exc)
@@ -42,57 +32,82 @@ try:
                 return None
         
         # 替换原始方法
-    hubs.handle_error = custom_handle_error
         hubs.trampoline = custom_trampoline
     
     # 同样处理 greenio 模块的所有相关方法，捕获所有异常
     from eventlet import greenio
     
-    # 保存原始的 greenio 方法
+    # 保存并替换原始的 greenio 方法（只修改存在的方法）
+    if hasattr(greenio.base.GreenSocket, '_recv_loop'):
         original_recv_loop = greenio.base.GreenSocket._recv_loop
-    original_recv_into = greenio.base.GreenSocket.recv_into
-    original_send_loop = greenio.base.GreenSocket._send_loop
-    original_send = greenio.base.GreenSocket.send
-    
-    def safe_io_operation(io_func, *args, **kwargs):
-        """安全执行 IO 操作，捕获所有异常"""
-        try:
-            return io_func(*args, **kwargs)
-        except Exception as e:
-            # 捕获所有 IO 操作异常
-            print(f"[Eventlet IO Error] {io_func.__name__}: {type(e).__name__}: {e}")
-            print("[Traceback]")
-            traceback.print_exc()
-            # 根据操作类型返回适当的默认值
-            if 'recv' in io_func.__name__ or 'read' in io_func.__name__:
-                return b'' if 'recv' in io_func.__name__ else ''
-            elif 'send' in io_func.__name__ or 'write' in io_func.__name__:
-                return 0
-            return None
         
         def custom_recv_loop(self, recv_func, recv_args, *args, **kwargs):
-        return safe_io_operation(original_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:
+                # 捕获所有 IO 操作异常
+                print(f"[Eventlet IO Error] _recv_loop: {type(e).__name__}: {e}")
+                print("[Traceback]")
+                traceback.print_exc()
+                # 返回空字节串继续运行
+                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):
-        return safe_io_operation(original_recv_into, self, buffer, nbytes, flags)
+            try:
+                return original_recv_into(self, buffer, nbytes, flags)
+            except Exception as e:
+                # 捕获所有 IO 操作异常
+                print(f"[Eventlet IO Error] recv_into: {type(e).__name__}: {e}")
+                print("[Traceback]")
+                traceback.print_exc()
+                # 返回 0 继续运行
+                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):
-        return safe_io_operation(original_send_loop, self, send_func, data, *args, **kwargs)
+            try:
+                return original_send_loop(self, send_func, data, *args, **kwargs)
+            except Exception as e:
+                # 捕获所有 IO 操作异常
+                print(f"[Eventlet IO Error] _send_loop: {type(e).__name__}: {e}")
+                print("[Traceback]")
+                traceback.print_exc()
+                # 返回 0 继续运行
+                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):
-        return safe_io_operation(original_send, self, data, flags)
+            try:
+                return original_send(self, data, flags)
+            except Exception as e:
+                # 捕获所有 IO 操作异常
+                print(f"[Eventlet IO Error] send: {type(e).__name__}: {e}")
+                print("[Traceback]")
+                traceback.print_exc()
+                # 返回 0 继续运行
+                return 0
         
-    # 替换 greenio 方法
-    greenio.base.GreenSocket._recv_loop = custom_recv_loop
-    greenio.base.GreenSocket.recv_into = custom_recv_into
-    greenio.base.GreenSocket._send_loop = custom_send_loop
         greenio.base.GreenSocket.send = custom_send
     
     # 处理 WSGI 中的所有错误
     try:
         import eventlet.wsgi
+        
+        if hasattr(eventlet.wsgi.HttpProtocol, 'handle_one_request'):
             original_wsgi_handle_one_request = eventlet.wsgi.HttpProtocol.handle_one_request
-        original_wsgi_handle_one_response = eventlet.wsgi.HttpProtocol.handle_one_response
             
             def custom_wsgi_handle_one_request(self):
                 try:
@@ -109,6 +124,11 @@ try:
                         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)
@@ -124,13 +144,12 @@ try:
                         pass
                     return
             
-        eventlet.wsgi.HttpProtocol.handle_one_request = custom_wsgi_handle_one_request
             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] All eventlet.hubs errors will be caught and printed, not raised")
+    print("[Eventlet Error Handling] Eventlet errors will be caught and printed, not raised")
 except Exception as e:
     # 如果修改失败，打印错误但继续运行
     print(f"[Eventlet Patch Initialization Error] {e}")