在Aiohttp/Python中记录websocket消息的最佳方式是什么？

import aiohttp
import aiohttp.web
import logging

# 设置日志记录器
logging.basicConfig(level=logging.INFO)
logger = logging.getLogger(__name__)

async def websocket_handler(request):
    ws = aiohttp.web.WebSocketResponse()
    await ws.prepare(request)

    async for msg in ws:
        if msg.type == aiohttp.WSMsgType.TEXT:
            if msg.data == 'close':
                await ws.close()
            else:
                # 记录接收到的消息
                logger.info(f"Received message: {msg.data}")
                await ws.send_str(msg.data)  # 将消息回传给客户端
        elif msg.type == aiohttp.WSMsgType.ERROR:
            logger.error(f'WebSocket connection closed with exception {ws.exception()}')

    logger.info('WebSocket connection closed')
    return ws

app = aiohttp.web.Application()
app.add_routes([aiohttp.web.get('/', websocket_handler)])

if __name__ == '__main__':
    aiohttp.web.run_app(app)