Merge pull request #1035 from yunstanford/patch-N

Adopt new websockets interface
This commit is contained in:
Raphael Deem
2017-12-02 01:27:09 -08:00
committed by GitHub
4 changed files with 8 additions and 4 deletions

View File

@@ -636,7 +636,9 @@ def serve(host, port, request_handler, error_handler, before_start=None,
coros = []
for conn in connections:
if hasattr(conn, "websocket") and conn.websocket:
coros.append(conn.websocket.close_connection(force=True))
coros.append(
conn.websocket.close_connection(after_handshake=True)
)
else:
conn.close()

View File

@@ -115,7 +115,9 @@ class GunicornWorker(base.Worker):
coros = []
for conn in self.connections:
if hasattr(conn, "websocket") and conn.websocket:
coros.append(conn.websocket.close_connection(force=True))
coros.append(
conn.websocket.close_connection(after_handshake=False)
)
else:
conn.close()
_shutdown = asyncio.gather(*coros, loop=self.loop)