python - 如何将 aiohttp websocket 处理程序重写为 sanic？

Question

我的 aiohttp 项目中有以下 websocket 处理程序:

async def websocket_handler(request):
     ws = web.WebSocketResponse()
     await ws.prepare(request)
     request.app['websockets'].append(ws)

     async for msg in ws:
         if msg.type == aiohttp.WSMsgType.TEXT:
             if msg.data == 'close':
                 await ws.close()

         elif msg.type == aiohttp.WSMsgType.ERROR:
             logger.info('ws connection closed with exception %s' %
                            ws.exception())

     request.app['websockets'].remove(ws)
     return ws

但现在我想切换到 sanic 框架。如何重写这个方法？我不明白如何从这个 tutorial 做到这一点

Answer 1

@bp.websocket('/websocket_handler')
    async def websocket_handler(_, ws):
        self.app['web_socket'].append(ws)
        while True:
            try:
                await ws.recv()
            except ConnectionClosed:
                break

        self.app['web_socket'].remove(ws)