Merge pull request #1330 from SaptakS/chat-socket-fix

Force shutdowns the socket server on closing chat server
This commit is contained in:
Micah Lee 2021-04-25 17:41:41 -04:00 committed by GitHub
commit 2e44dfbbda
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -360,13 +360,18 @@ class Web:
# Shutdown the flask service
try:
func = request.environ.get("werkzeug.server.shutdown")
if func is None:
if func is None and self.mode != "chat":
raise RuntimeError("Not running with the Werkzeug Server")
func()
except:
pass
self.running = False
# If chat, shutdown the socket server
if self.mode == "chat":
self.socketio.stop()
def start(self, port):
"""
Start the flask web server.