|
|
@ -21,7 +21,8 @@ def with_instrumented_server(auth=False, **ikwargs): |
|
|
def decorator(f): |
|
|
def decorator(f): |
|
|
@wraps(f) |
|
|
@wraps(f) |
|
|
def wrapped(self, *args, **kwargs): |
|
|
def wrapped(self, *args, **kwargs): |
|
|
sio = socketio.AsyncServer(async_mode='asgi') |
|
|
sio = socketio.AsyncServer(async_mode='asgi', ping_interval=10, |
|
|
|
|
|
ping_timeout=10) |
|
|
|
|
|
|
|
|
@sio.event |
|
|
@sio.event |
|
|
async def enter_room(sid, data): |
|
|
async def enter_room(sid, data): |
|
|
@ -37,6 +38,7 @@ def with_instrumented_server(auth=False, **ikwargs): |
|
|
|
|
|
|
|
|
async def shutdown(): |
|
|
async def shutdown(): |
|
|
await self.isvr.shutdown() |
|
|
await self.isvr.shutdown() |
|
|
|
|
|
self.isvr.uninstrument() |
|
|
await sio.shutdown() |
|
|
await sio.shutdown() |
|
|
|
|
|
|
|
|
if 'server_stats_interval' not in ikwargs: |
|
|
if 'server_stats_interval' not in ikwargs: |
|
|
@ -57,7 +59,6 @@ def with_instrumented_server(auth=False, **ikwargs): |
|
|
ret = f(self, *args, **kwargs) |
|
|
ret = f(self, *args, **kwargs) |
|
|
finally: |
|
|
finally: |
|
|
server.stop() |
|
|
server.stop() |
|
|
self.isvr.uninstrument() |
|
|
|
|
|
self.isvr = None |
|
|
self.isvr = None |
|
|
|
|
|
|
|
|
EngineIOSocket.schedule_ping = original_schedule_ping |
|
|
EngineIOSocket.schedule_ping = original_schedule_ping |
|
|
|