feat(bot): 支持设置事件队列的最大缓冲区大小 (#158)

Author: st1020

alicebot/bot.py

@@ -202,7 +202,9 @@ async def _init(self) -> None:
         self._should_exit = anyio.Event()
         self._condition = anyio.Condition()
         self._event_send_stream, self._event_receive_stream = (
-            anyio.create_memory_object_stream()
+            anyio.create_memory_object_stream(
+                max_buffer_size=self.config.bot.event_queue_size
+            )
         )
 
         # 加载配置文件

alicebot/config.py

@@ -45,12 +45,14 @@ class BotConfig(ConfigModel):
     """Bot 配置。
 
     Attributes:
+        event_queue_size: 事件队列的最大缓冲区大小。
         plugins: 将被加载的插件列表，将被 `Bot` 类的 `load_plugins()` 方法加载。
         plugin_dirs: 将被加载的插件目录列表，将被 `Bot` 类的 `load_plugins_from_dirs()` 方法加载。
         adapters: 将被加载的适配器列表，将依次被 `Bot` 类的 `load_adapters()` 方法加载。
         log: AliceBot 日志相关设置。
     """
 
+    event_queue_size: int = Field(default=0, ge=0)
     plugins: set[str] = Field(default_factory=set)
     plugin_dirs: set[DirectoryPath] = Field(default_factory=set)
     adapters: set[str] = Field(default_factory=set)

tests/config_files/raw_config.json

@@ -1,5 +1,6 @@
 {
   "bot": {
+    "event_queue_size": 0,
     "plugins": [],
     "plugin_dirs": ["plugins"],
     "adapters": [],