fix: 修复处理消息时占用连接的情况
This commit is contained in:
@@ -109,6 +109,7 @@ class MainSystem:
|
|||||||
self.remove_recalled_message_task(),
|
self.remove_recalled_message_task(),
|
||||||
emoji_manager.start_periodic_check(),
|
emoji_manager.start_periodic_check(),
|
||||||
self.app.run(),
|
self.app.run(),
|
||||||
|
self.app.message_process(),
|
||||||
]
|
]
|
||||||
await asyncio.gather(*tasks)
|
await asyncio.gather(*tasks)
|
||||||
|
|
||||||
|
|||||||
@@ -12,6 +12,7 @@ class BaseMessageAPI:
|
|||||||
self.host = host
|
self.host = host
|
||||||
self.port = port
|
self.port = port
|
||||||
self.message_handlers: List[Callable] = []
|
self.message_handlers: List[Callable] = []
|
||||||
|
self.cache = []
|
||||||
self._setup_routes()
|
self._setup_routes()
|
||||||
self._running = False
|
self._running = False
|
||||||
|
|
||||||
@@ -20,12 +21,11 @@ class BaseMessageAPI:
|
|||||||
|
|
||||||
@self.app.post("/api/message")
|
@self.app.post("/api/message")
|
||||||
async def handle_message(message: Dict[str, Any]):
|
async def handle_message(message: Dict[str, Any]):
|
||||||
# try:
|
try:
|
||||||
for handler in self.message_handlers:
|
self.cache.append(message)
|
||||||
await handler(message)
|
return {"status": "success"}
|
||||||
return {"status": "success"}
|
except Exception as e:
|
||||||
# except Exception as e:
|
raise HTTPException(status_code=500, detail=str(e)) from e
|
||||||
# raise HTTPException(status_code=500, detail=str(e)) from e
|
|
||||||
|
|
||||||
def register_message_handler(self, handler: Callable):
|
def register_message_handler(self, handler: Callable):
|
||||||
"""注册消息处理函数"""
|
"""注册消息处理函数"""
|
||||||
@@ -41,6 +41,20 @@ class BaseMessageAPI:
|
|||||||
# logger.error(f"发送消息失败: {str(e)}")
|
# logger.error(f"发送消息失败: {str(e)}")
|
||||||
pass
|
pass
|
||||||
|
|
||||||
|
async def message_process(
|
||||||
|
self,
|
||||||
|
):
|
||||||
|
"""启动消息处理"""
|
||||||
|
while True:
|
||||||
|
if len(self.cache) > 0:
|
||||||
|
for handler in self.message_handlers:
|
||||||
|
await handler(self.cache[0])
|
||||||
|
self.cache.pop(0)
|
||||||
|
if len(self.cache) > 0:
|
||||||
|
await asyncio.sleep(0.1 / len(self.cache))
|
||||||
|
else:
|
||||||
|
await asyncio.sleep(0.2)
|
||||||
|
|
||||||
def run_sync(self):
|
def run_sync(self):
|
||||||
"""同步方式运行服务器"""
|
"""同步方式运行服务器"""
|
||||||
uvicorn.run(self.app, host=self.host, port=self.port)
|
uvicorn.run(self.app, host=self.host, port=self.port)
|
||||||
|
|||||||
Reference in New Issue
Block a user