Better handling of Gunicorn threaded worker

miguelgrinberg · Aug 21, 2023 · 29e4492 · 29e4492
1 parent 87f6003
commit 29e4492
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/src/engineio/async_drivers/threading.py b/src/engineio/async_drivers/threading.py
@@ -20,10 +20,14 @@ def __init__(self, handler, server):
 
     def __call__(self, environ, start_response):
         self.ws = Server(environ)
-        return self.app(self)
+        ret = self.app(self)
+        if self.ws.mode == 'gunicorn':
+            raise StopIteration()
+        return ret
 
     def close(self):
-        return self.ws.close()
+        if self.ws.connected:
+            self.ws.close()
 
     def send(self, message):
         try: