change rule

nonebot/drivers/__init__.py

@@ -4,6 +4,7 @@
 import abc
 from ipaddress import IPv4Address
 
+from nonebot.log import logger
 from nonebot.config import Env, Config
 from nonebot.typing import Bot, Dict, Type, Optional, Callable
 
@@ -20,6 +21,7 @@ class BaseDriver(abc.ABC):
     @classmethod
     def register_adapter(cls, name: str, adapter: Type[Bot]):
         cls._adapters[name] = adapter
+        logger.debug(f'Succeeded to load adapter "{name}"')
 
     @property
     @abc.abstractmethod

nonebot/drivers/fastapi.py

@@ -152,15 +152,16 @@ class Driver(BaseDriver):
         await websocket.accept()
         self._clients[x_self_id] = bot
 
-        while not websocket.closed:
-            data = await websocket.receive()
+        try:
+            while not websocket.closed:
+                data = await websocket.receive()
 
-            if not data:
-                continue
+                if not data:
+                    continue
 
-            await bot.handle_message(data)
-
-        del self._clients[x_self_id]
+                await bot.handle_message(data)
+        finally:
+            del self._clients[x_self_id]
 
 
 class WebSocket(BaseWebSocket):