初始化

MaiBot-Napcat-Adapter-dev/main.py

@@ -0,0 +1,90 @@
+import asyncio
+import sys
+import json
+import websockets as Server
+from src.logger import logger
+from src.recv_handler.message_handler import message_handler
+from src.recv_handler.meta_event_handler import meta_event_handler
+from src.recv_handler.notice_handler import notice_handler
+from src.recv_handler.message_sending import message_send_instance
+from src.send_handler import send_handler
+from src.config import global_config
+from src.mmc_com_layer import mmc_start_com, mmc_stop_com, router
+from src.response_pool import put_response, check_timeout_response
+
+message_queue = asyncio.Queue()
+
+
+async def message_recv(server_connection: Server.ServerConnection):
+    await message_handler.set_server_connection(server_connection)
+    asyncio.create_task(notice_handler.set_server_connection(server_connection))
+    await send_handler.set_server_connection(server_connection)
+    async for raw_message in server_connection:
+        logger.debug(f"{raw_message[:1500]}..." if (len(raw_message) > 1500) else raw_message)
+        decoded_raw_message: dict = json.loads(raw_message)
+        post_type = decoded_raw_message.get("post_type")
+        if post_type in ["meta_event", "message", "notice"]:
+            await message_queue.put(decoded_raw_message)
+        elif post_type is None:
+            await put_response(decoded_raw_message)
+
+
+async def message_process():
+    while True:
+        message = await message_queue.get()
+        post_type = message.get("post_type")
+        if post_type == "message":
+            await message_handler.handle_raw_message(message)
+        elif post_type == "meta_event":
+            await meta_event_handler.handle_meta_event(message)
+        elif post_type == "notice":
+            await notice_handler.handle_notice(message)
+        else:
+            logger.warning(f"未知的post_type: {post_type}")
+        message_queue.task_done()
+        await asyncio.sleep(0.05)
+
+
+async def main():
+    message_send_instance.maibot_router = router
+    _ = await asyncio.gather(napcat_server(), mmc_start_com(), message_process(), check_timeout_response())
+
+
+async def napcat_server():
+    logger.info("正在启动adapter...")
+    async with Server.serve(message_recv, global_config.napcat_server.host, global_config.napcat_server.port, max_size=2**26) as server:
+        logger.info(
+            f"Adapter已启动，监听地址: ws://{global_config.napcat_server.host}:{global_config.napcat_server.port}"
+        )
+        await server.serve_forever()
+
+
+async def graceful_shutdown():
+    try:
+        logger.info("正在关闭adapter...")
+        tasks = [t for t in asyncio.all_tasks() if t is not asyncio.current_task()]
+        for task in tasks:
+            if not task.done():
+                task.cancel()
+        await asyncio.wait_for(asyncio.gather(*tasks, return_exceptions=True), 15)
+        await mmc_stop_com()  # 后置避免神秘exception
+        logger.info("Adapter已成功关闭")
+    except Exception as e:
+        logger.error(f"Adapter关闭中出现错误: {e}")
+
+
+if __name__ == "__main__":
+    loop = asyncio.new_event_loop()
+    asyncio.set_event_loop(loop)
+    try:
+        loop.run_until_complete(main())
+    except KeyboardInterrupt:
+        logger.warning("收到中断信号，正在优雅关闭...")
+        loop.run_until_complete(graceful_shutdown())
+    except Exception as e:
+        logger.exception(f"主程序异常: {str(e)}")
+        sys.exit(1)
+    finally:
+        if loop and not loop.is_closed():
+            loop.close()
+        sys.exit(0)