d07915eea0
- Removed UniversalMessageSender from group_generator.py and private_generator.py. - Updated PlatformIOManager to manage legacy send drivers and ensure send pipeline readiness. - Enhanced LegacyPlatformDriver to utilize prepared messages for sending. - Refactored send_service to unify message sending logic and integrate with Platform IO. - Added regression tests for Platform IO legacy driver and send service functionality.
132 lines
4.2 KiB
Python
132 lines
4.2 KiB
Python
import asyncio
|
||
|
||
from fastapi import FastAPI, APIRouter
|
||
from rich.traceback import install
|
||
from typing import Optional
|
||
from uvicorn import Config, Server as UvicornServer
|
||
|
||
from src.common.logger import get_logger
|
||
from src.common.utils.port_checker import assert_port_available, is_port_conflict_error, log_port_conflict
|
||
|
||
install(extra_lines=3)
|
||
|
||
logger = get_logger("message_server")
|
||
|
||
|
||
class Server:
|
||
def __init__(self, host: Optional[str] = None, port: Optional[int] = None, app_name: str = "MaiMCore"):
|
||
self.app = FastAPI(title=app_name)
|
||
self._host: str = "127.0.0.1"
|
||
self._port: int = 8080
|
||
self._server: Optional[UvicornServer] = None
|
||
self.set_address(host, port)
|
||
|
||
def register_router(self, router: APIRouter, prefix: str = ""):
|
||
"""注册路由
|
||
|
||
APIRouter 用于对相关的路由端点进行分组和模块化管理：
|
||
1. 可以将相关的端点组织在一起，便于管理
|
||
2. 支持添加统一的路由前缀
|
||
3. 可以为一组路由添加共同的依赖项、标签等
|
||
|
||
示例:
|
||
router = APIRouter()
|
||
|
||
@router.get("/users")
|
||
def get_users():
|
||
return {"users": [...]}
|
||
|
||
@router.post("/users")
|
||
def create_user():
|
||
return {"msg": "user created"}
|
||
|
||
# 注册路由，添加前缀 "/api/v1"
|
||
server.register_router(router, prefix="/api/v1")
|
||
"""
|
||
self.app.include_router(router, prefix=prefix)
|
||
|
||
def set_address(self, host: Optional[str] = None, port: Optional[int] = None):
|
||
"""设置服务器地址和端口"""
|
||
if host:
|
||
self._host = host
|
||
if port:
|
||
self._port = port
|
||
|
||
async def run(self):
|
||
"""启动服务器"""
|
||
assert_port_available(
|
||
host=self._host,
|
||
port=self._port,
|
||
service_name="消息服务器",
|
||
logger=logger,
|
||
config_hint="maim_message.ws_server_port (config/bot_config.toml)",
|
||
)
|
||
|
||
# 禁用 uvicorn 默认日志和访问日志
|
||
# 设置 ws_max_size 为 100MB，支持大消息（如包含多张图片的转发消息）
|
||
config = Config(
|
||
app=self.app,
|
||
host=self._host,
|
||
port=self._port,
|
||
log_config=None,
|
||
access_log=False,
|
||
ws_max_size=104_857_600, # 100MB
|
||
)
|
||
self._server = UvicornServer(config=config)
|
||
try:
|
||
await self._server.serve()
|
||
except KeyboardInterrupt:
|
||
await self.shutdown()
|
||
raise
|
||
except OSError as e:
|
||
if is_port_conflict_error(e):
|
||
log_port_conflict(
|
||
logger,
|
||
service_name="消息服务器",
|
||
host=self._host,
|
||
port=self._port,
|
||
config_hint="maim_message.ws_server_port (config/bot_config.toml)",
|
||
)
|
||
await self.shutdown()
|
||
raise RuntimeError(f"服务器运行错误: {str(e)}") from e
|
||
except Exception as e:
|
||
await self.shutdown()
|
||
raise RuntimeError(f"服务器运行错误: {str(e)}") from e
|
||
finally:
|
||
await self.shutdown()
|
||
|
||
async def shutdown(self):
|
||
"""安全关闭服务器"""
|
||
if self._server:
|
||
self._server.should_exit = True
|
||
try:
|
||
# 添加 3 秒超时，避免 shutdown 永久挂起
|
||
await asyncio.wait_for(self._server.shutdown(), timeout=3.0)
|
||
except asyncio.TimeoutError:
|
||
# 超时就强制标记为 None，让垃圾回收处理
|
||
pass
|
||
except Exception:
|
||
# 忽略其他异常
|
||
pass
|
||
finally:
|
||
self._server = None
|
||
|
||
def get_app(self) -> FastAPI:
|
||
"""获取 FastAPI 实例"""
|
||
return self.app
|
||
|
||
|
||
global_server = None
|
||
|
||
|
||
def get_global_server() -> Server:
|
||
"""获取全局服务器实例"""
|
||
from src.config.config import global_config
|
||
|
||
global global_server
|
||
if global_server is None:
|
||
global_server = Server(
|
||
host=global_config.maim_message.ws_server_host, port=global_config.maim_message.ws_server_port
|
||
)
|
||
return global_server
|