Remove legacy temi_mcpserver files and migrate to terminal_temi_mcp structure, including new MQTT handling and navigation server implementation. Updated project configuration and added necessary scripts for deployment.
This commit is contained in:
17
terminal_temi_mcp/mcp-server-temi.json
Normal file
17
terminal_temi_mcp/mcp-server-temi.json
Normal file
@@ -0,0 +1,17 @@
|
||||
{
|
||||
"mcpServers": {
|
||||
"terminal-temi-mcp": {
|
||||
"disabled": false,
|
||||
"type": "stdio",
|
||||
"timeout": 30,
|
||||
"command": "uvx",
|
||||
"args": [
|
||||
"terminal_temi_mcp"
|
||||
],
|
||||
"env": {
|
||||
"employeeId": "$employeeId$",
|
||||
"userId": "$employeeId$"
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
26
terminal_temi_mcp/pyproject.toml
Normal file
26
terminal_temi_mcp/pyproject.toml
Normal file
@@ -0,0 +1,26 @@
|
||||
[build-system]
|
||||
requires = ["hatchling"]
|
||||
build-backend = "hatchling.build"
|
||||
|
||||
[project]
|
||||
name = "terminal_temi_mcp"
|
||||
version = "0.1.15"
|
||||
description = "MQTT-based navigation server for robot"
|
||||
requires-python = ">=3.10"
|
||||
dependencies = [
|
||||
"fastapi>=0.95.0",
|
||||
"uvicorn>=0.21.0",
|
||||
"paho-mqtt>=2.0.0",
|
||||
"pydantic>=1.10.0",
|
||||
"python-dotenv>=0.21.0",
|
||||
"mcp[cli]>=1.6.0",
|
||||
"requests>=2.25.0"
|
||||
]
|
||||
|
||||
[project.scripts]
|
||||
terminal_temi_mcp = "terminal_temi_mcp.main:main"
|
||||
|
||||
|
||||
[tool.hatch.build.targets.wheel]
|
||||
packages = ["terminal_temi_mcp"]
|
||||
|
||||
0
terminal_temi_mcp/terminal_temi_mcp/__init__.py
Normal file
0
terminal_temi_mcp/terminal_temi_mcp/__init__.py
Normal file
489
terminal_temi_mcp/terminal_temi_mcp/main.py
Normal file
489
terminal_temi_mcp/terminal_temi_mcp/main.py
Normal file
@@ -0,0 +1,489 @@
|
||||
from typing import Optional, Sequence
|
||||
import logging
|
||||
from fastapi import FastAPI
|
||||
from pydantic import BaseModel
|
||||
from mcp.server import Server
|
||||
from mcp.server.stdio import stdio_server
|
||||
from mcp.types import Tool, TextContent
|
||||
import asyncio
|
||||
import json
|
||||
import time
|
||||
import uuid
|
||||
from .mcp_mqtt import get_mcpmqtt_handler
|
||||
|
||||
# 配置日志
|
||||
logging.basicConfig(
|
||||
level=logging.INFO,
|
||||
format='%(asctime)s - %(name)s - %(levelname)s - %(message)s'
|
||||
)
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
class TourStop(BaseModel):
|
||||
name: str
|
||||
text: str
|
||||
|
||||
|
||||
class NavServer:
|
||||
def __init__(self, mmhandler=None):
|
||||
self.mmhandler = mmhandler or get_mcpmqtt_handler()
|
||||
|
||||
async def publish_Cmd(self, device_id: str, cmd: str, params: dict):
|
||||
try:
|
||||
payload = {
|
||||
"device_id": device_id,
|
||||
"cmd": cmd,
|
||||
"data": params["data"]
|
||||
}
|
||||
self.mmhandler.client.publish("robot/cmd", json.dumps(payload), qos=2)
|
||||
return f"{cmd}工具执行成功"
|
||||
except Exception as e:
|
||||
logger.error(f"Failed to publish {cmd} command: {str(e)}", exc_info=True)
|
||||
raise
|
||||
|
||||
async def nav_to(self, location: str):
|
||||
"""
|
||||
机器人导航去某个地方。
|
||||
Args:
|
||||
- location (str): 目标地点名称
|
||||
Returns:
|
||||
str: 命令执行结果消息
|
||||
|
||||
Raises:
|
||||
ValueError: 如果参数无效或为空
|
||||
Exception: 如果MQTT发布失败
|
||||
"""
|
||||
if location:
|
||||
try:
|
||||
if location == "充电桩":
|
||||
location = "home base"
|
||||
params = {
|
||||
"data": {"location": location}
|
||||
}
|
||||
return await self.publish_Cmd("123456", "nav", params)
|
||||
except Exception as e:
|
||||
logger.error(f"Failed to call navigation mcp-tool: {str(e)} ", exc_info=True)
|
||||
raise
|
||||
|
||||
async def speak(self, speech: str):
|
||||
"""
|
||||
机器人说某些话。
|
||||
Args:
|
||||
- speech (str): 说话内容
|
||||
Returns:
|
||||
str: 命令执行结果消息
|
||||
|
||||
Raises:
|
||||
ValueError: 如果参数无效或为空
|
||||
Exception: 如果MQTT发布失败
|
||||
"""
|
||||
if speech:
|
||||
try:
|
||||
params = {
|
||||
"data": {"speech": speech}
|
||||
}
|
||||
return await self.publish_Cmd("123456", "speak", params)
|
||||
except Exception as e:
|
||||
logger.error(f"Failed to call speak mcp-tool: {str(e)} ", exc_info=True)
|
||||
raise
|
||||
|
||||
async def reception(self, location: str):
|
||||
"""
|
||||
机器人接待客人。
|
||||
Args:
|
||||
- location (str): 机器人去这个位置接待客人
|
||||
- task_id (Optional[str]): 任务id，不传则自动生成
|
||||
Returns:
|
||||
str: 命令执行结果消息
|
||||
|
||||
Raises:
|
||||
ValueError: 如果参数无效或为空
|
||||
Exception: 如果MQTT发布失败
|
||||
"""
|
||||
if location :
|
||||
try:
|
||||
params = {
|
||||
"data": {
|
||||
"location": location
|
||||
}
|
||||
}
|
||||
return await self.publish_Cmd("123456", "reception", params)
|
||||
except Exception as e:
|
||||
logger.error(f"Failed to call reception mcp-tool: {str(e)} ", exc_info=True)
|
||||
raise
|
||||
|
||||
async def notification(self, location: str, text: str):
|
||||
"""
|
||||
机器人通知事情。
|
||||
Args:
|
||||
- location (str): 机器人去这个位置通知事情
|
||||
- text (str): 通知事情的内容
|
||||
- task_id (Optional[str]): 任务id，不传则自动生成
|
||||
Returns:
|
||||
str: 命令执行结果消息
|
||||
|
||||
Raises:
|
||||
ValueError: 如果参数无效或为空
|
||||
Exception: 如果MQTT发布失败
|
||||
"""
|
||||
if location :
|
||||
try:
|
||||
params = {
|
||||
"data": {
|
||||
"location": location,
|
||||
"text": text
|
||||
}
|
||||
}
|
||||
return await self.publish_Cmd("123456", "notification", params)
|
||||
except Exception as e:
|
||||
logger.error(f"Failed to call notification mcp-tool: {str(e)} ", exc_info=True)
|
||||
raise
|
||||
async def repose(self):
|
||||
"""
|
||||
机器人重新定位。
|
||||
"""
|
||||
try:
|
||||
params = {
|
||||
"data": {
|
||||
"action": "repose"
|
||||
}
|
||||
}
|
||||
return await self.publish_Cmd("123456", "map", params)
|
||||
except Exception as e:
|
||||
logger.error(f"Failed to call repose mcp-tool: {str(e)} ", exc_info=True)
|
||||
raise
|
||||
|
||||
async def delivery(self, first_location: str, next_location: str):
|
||||
"""
|
||||
机器人运送物品。
|
||||
Args:
|
||||
- first_location (str): 机器人去这个位置取物品
|
||||
- next_location (str): 机器人将物品送到这个位置
|
||||
- task_id (Optional[str]): 任务id，不传则自动生成
|
||||
Returns:
|
||||
str: 命令执行结果消息
|
||||
|
||||
Raises:
|
||||
ValueError: 如果参数无效或为空
|
||||
Exception: 如果MQTT发布失败
|
||||
|
||||
"""
|
||||
if first_location and next_location:
|
||||
try:
|
||||
params = {
|
||||
"data": {
|
||||
"first_location": first_location,
|
||||
"next_location": next_location
|
||||
}
|
||||
}
|
||||
return await self.publish_Cmd("123456", "delivery", params)
|
||||
except Exception as e:
|
||||
logger.error(f"Failed to call delivery mcp-tool: {str(e)} ", exc_info=True)
|
||||
raise
|
||||
|
||||
async def startFaceRecognize(self, device_id: str):
|
||||
"""
|
||||
机器人进行人脸识别。
|
||||
Args:
|
||||
- device_id (str): 设备id
|
||||
Returns:
|
||||
str: 命令执行结果消息
|
||||
Raises:
|
||||
ValueError: 如果参数无效或为空
|
||||
Exception: 如果MQTT发布失败
|
||||
"""
|
||||
if device_id:
|
||||
try:
|
||||
params = {
|
||||
"data": {}
|
||||
}
|
||||
return await self.publish_Cmd("123456", "face", params)
|
||||
except Exception as e:
|
||||
logger.error(f"Failed to call face recognize mcp-tool: {str(e)} ", exc_info=True)
|
||||
raise
|
||||
|
||||
async def scanQRCode(self, device_id: str):
|
||||
"""
|
||||
机器人进行二维码扫描。
|
||||
Args:
|
||||
- device_id (str): 设备id
|
||||
Returns:
|
||||
str: 命令执行结果消息
|
||||
Raises:
|
||||
ValueError: 如果参数无效或为空
|
||||
Exception: 如果MQTT发布失败
|
||||
"""
|
||||
if device_id:
|
||||
try:
|
||||
params = {
|
||||
"data": {}
|
||||
}
|
||||
return await self.publish_Cmd("123456", "qrcode", params)
|
||||
except Exception as e:
|
||||
logger.error(f"Failed to call QR code scan mcp-tool: {str(e)} ", exc_info=True)
|
||||
raise
|
||||
|
||||
async def save_position(self, device_id: str, location_name: str):
|
||||
"""
|
||||
机器人保存具体地点。
|
||||
Args:
|
||||
- device_id (str): 设备id
|
||||
- location_name (str): 地点名称
|
||||
Returns:
|
||||
str: 命令执行结果消息
|
||||
Raises:
|
||||
ValueError: 如果参数无效或为空
|
||||
Exception: 如果MQTT发布失败
|
||||
|
||||
"""
|
||||
if device_id and location_name:
|
||||
try:
|
||||
params = {
|
||||
"data": {"location_name": location_name}
|
||||
}
|
||||
return await self.publish_Cmd("123456", "saveLocation", params)
|
||||
except Exception as e:
|
||||
logger.error(f"Failed to call save_position mcp-tool: {str(e)} ", exc_info=True)
|
||||
raise
|
||||
|
||||
|
||||
async def guide(self):
|
||||
"""
|
||||
机器人导览。
|
||||
Args:
|
||||
Returns:
|
||||
str: 命令执行结果消息
|
||||
|
||||
Raises:
|
||||
ValueError: 如果参数无效或为空
|
||||
Exception: 如果MQTT发布失败
|
||||
"""
|
||||
try:
|
||||
datas = [
|
||||
TourStop(name="前台", text="欢迎来到我们的场所，这是入口区域"),
|
||||
TourStop(name="灵泽万川展厅", text="这是我们的主大厅，提供接待服务"),
|
||||
TourStop(name="爱易拍展厅", text="这里展示我们的主要产品和成果")
|
||||
]
|
||||
params = {
|
||||
"data": [data.model_dump() for data in datas]
|
||||
}
|
||||
return await self.publish_Cmd("123456", "guide", params)
|
||||
except Exception as e:
|
||||
logger.error(f"Failed to call guide mcp-tool: {str(e)} ", exc_info=True)
|
||||
raise
|
||||
|
||||
async def serve() -> None:
|
||||
server = Server("terminal_temi_mcp")
|
||||
mmhandler = get_mcpmqtt_handler()
|
||||
nav_server = NavServer(mmhandler)
|
||||
|
||||
@server.list_tools()
|
||||
async def list_tools() -> list[Tool]:
|
||||
"""列出所有工具"""
|
||||
return [
|
||||
Tool(
|
||||
name="nav_to",
|
||||
description="数字员工、终端、具身机器人、前台机器人、助手、机器人导航带路到指定地点:带我去、去XX、请引路、导航去、我想去、能带我去吗、带路、送我去、在哪里、前往XX、去XX位置、带我过去、请带我到、我想前往、去那个地方、我要去、能不能去、带我导航、这个地方怎么走、我想找XX、导航带我去、请带路到、我要去那边、走一趟去、去下XX、到XX怎么走",
|
||||
inputSchema={
|
||||
"type": "object",
|
||||
"properties": {
|
||||
"location": {
|
||||
"type": "string",
|
||||
"description": "目标地点名称",
|
||||
"minLength": 1
|
||||
}
|
||||
},
|
||||
"required": ["location"]
|
||||
}
|
||||
),
|
||||
Tool(
|
||||
name="speak",
|
||||
description="数字员工、终端、具身机器人、前台机器人、助手、机器人进行语音播报:告诉、提醒、告知、提示",
|
||||
inputSchema={
|
||||
"type": "object",
|
||||
"properties": {
|
||||
"speech": {
|
||||
"type": "string",
|
||||
"description": "要播报的语音内容",
|
||||
"minLength": 1
|
||||
}
|
||||
},
|
||||
"required": ["speech"]
|
||||
}
|
||||
),
|
||||
Tool(
|
||||
name="reception",
|
||||
description="数字员工、终端、具身机器人、前台机器人、助手、机器人去接待客人:去接人、请迎接客人、去接待、迎接一下、带人过来、把客人带过来、去门口接人、去接一下、请接人、接客人、帮我迎接、去把人接过来、接待来访者、请去接一下客人、送客人过来、从XX接到YY、引导访客、带访客过来、请引领客人、去门口把人接进来、从入口接人、去接待处接人、带客人进来",
|
||||
inputSchema={
|
||||
"type": "object",
|
||||
"properties": {
|
||||
"location": {
|
||||
"type": "string",
|
||||
"description": "引导接待客人到这个位置",
|
||||
"minLength": 1
|
||||
},
|
||||
},
|
||||
"required": ["location"]
|
||||
}
|
||||
),
|
||||
Tool(
|
||||
name="notification",
|
||||
description="数字员工、终端、具身机器人、前台机器人、助手、机器人去通知事情:通知、告诉、提醒、告知、提示",
|
||||
inputSchema={
|
||||
"type": "object",
|
||||
"properties": {
|
||||
"location": {
|
||||
"type": "string",
|
||||
"description": "去这个位置通知事情",
|
||||
"minLength": 1
|
||||
},
|
||||
"text": {
|
||||
"type": "string",
|
||||
"description": "通知事情的内容",
|
||||
"minLength": 1
|
||||
}
|
||||
},
|
||||
"required": ["location","text"]
|
||||
}
|
||||
),
|
||||
Tool(
|
||||
name="repose",
|
||||
description="数字员工、终端、具身机器人、前台机器人、助手、机器人去重新定位",
|
||||
inputSchema={
|
||||
"type": "object",
|
||||
"properties": {},
|
||||
"required": []
|
||||
}
|
||||
),
|
||||
Tool(
|
||||
name="delivery",
|
||||
description="数字员工、终端、具身机器人、前台机器人、助手、机器人去运输物品:帮我送东西、去送一下、去送物品、帮我拿过去、送这个过去、去送快递、运送一下、帮我带东西、把东西送过去、物品配送、拿去给XX、去取个东西、取完送过去、帮我捎个东西、把这个拿去、请运过去、送过去给谁、请配送一下、帮我转交、帮我送到、从XX拿到YY、物品搬运",
|
||||
inputSchema={
|
||||
"type": "object",
|
||||
"properties": {
|
||||
"first_location": {
|
||||
"type": "string",
|
||||
"description": "去这个位置取物品",
|
||||
"minLength": 1
|
||||
},
|
||||
"next_location": {
|
||||
"type": "string",
|
||||
"description": "将物品运输、送到这个位置",
|
||||
"minLength": 1
|
||||
}
|
||||
},
|
||||
"required": ["first_location","next_location"]
|
||||
}
|
||||
),
|
||||
# Tool(
|
||||
# name="guide",
|
||||
# description="数字员工、终端、具身机器人、前台机器人、助手、机器人进行导览服务:导览、介绍、参观、带领参观、导游、讲解、展示、演示、介绍场所、带我看看",
|
||||
# inputSchema={
|
||||
# "type": "object",
|
||||
# "properties": {
|
||||
# },
|
||||
# "required": []
|
||||
# }
|
||||
# )
|
||||
# Tool(
|
||||
# name="face",
|
||||
# description="数字员工、终端、具身机器人、前台机器人、助手、机器人进行人脸识别:识别人脸、人脸识别、脸部识别、面部识别、认脸、看看是谁、识别一下、看看这个人、辨识人脸、扫脸、人脸检测、看脸识别、脸部检测、识别面孔、看人脸、人脸扫描、脸识别、面孔识别、认人、识别身份、看看这是谁、人脸认证、脸部扫描、面部扫描",
|
||||
# inputSchema={
|
||||
# "type": "object",
|
||||
# "properties": {
|
||||
# "device_id": {
|
||||
# "type": "string",
|
||||
# "description": "设备ID",
|
||||
# "minLength": 1
|
||||
# },
|
||||
# },
|
||||
# "required": ["device_id"]
|
||||
# }
|
||||
# ),
|
||||
# Tool(
|
||||
# name="qrcode",
|
||||
# description="数字员工、终端、具身机器人、前台机器人、助手、机器人进行二维码扫描:扫码、扫二维码、扫QR码、二维码识别、码扫描、扫一下码、读取二维码、识别二维码、二维码扫描、QR扫描、扫描条码、扫描编码、读码、解析二维码、扫描识别、码识别、条码扫描、扫描二维码、读取码、扫描条形码、二维码读取、码解析",
|
||||
# inputSchema={
|
||||
# "type": "object",
|
||||
# "properties": {
|
||||
# "device_id": {
|
||||
# "type": "string",
|
||||
# "description": "设备ID",
|
||||
# "minLength": 1
|
||||
# },
|
||||
# "task_id": {
|
||||
# "type": "string",
|
||||
# "description": "任务ID"
|
||||
# }
|
||||
# },
|
||||
# "required": ["device_id"]
|
||||
# }
|
||||
# )
|
||||
]
|
||||
|
||||
@server.call_tool()
|
||||
async def call_tool(name: str, arguments: dict) -> Sequence[TextContent]:
|
||||
"""处理工具调用"""
|
||||
try:
|
||||
result = ""
|
||||
if name == "nav_to":
|
||||
if "location" not in arguments:
|
||||
raise ValueError("缺少必要参数: location")
|
||||
result = await nav_server.nav_to(
|
||||
location=arguments["location"]
|
||||
)
|
||||
elif name == "speak":
|
||||
if "speech" not in arguments:
|
||||
raise ValueError("缺少必要参数: speech")
|
||||
result = await nav_server.speak(
|
||||
speech=arguments["speech"]
|
||||
)
|
||||
elif name == "reception":
|
||||
if "location" not in arguments:
|
||||
raise ValueError("缺少必要参数: location")
|
||||
result = await nav_server.reception(
|
||||
location=arguments["location"],
|
||||
)
|
||||
elif name == "notification":
|
||||
if "location" not in arguments or "text" not in arguments:
|
||||
raise ValueError("缺少必要参数: location或text")
|
||||
result = await nav_server.notification(
|
||||
location=arguments["location"],
|
||||
text=arguments["text"]
|
||||
)
|
||||
elif name == "repose":
|
||||
result = await nav_server.repose()
|
||||
elif name == "delivery":
|
||||
if "first_location" not in arguments or "next_location" not in arguments:
|
||||
raise ValueError("缺少必要参数: first_location或next_location")
|
||||
result = await nav_server.delivery(
|
||||
first_location=arguments["first_location"],
|
||||
next_location=arguments["next_location"]
|
||||
)
|
||||
# elif name == "face":
|
||||
# if "device_id" not in arguments:
|
||||
# raise ValueError("缺少必要参数: device_id")
|
||||
# result = await nav_server.startFaceRecognize(
|
||||
# device_id=arguments["device_id"],
|
||||
# )
|
||||
# elif name == "qrcode":
|
||||
# if "device_id" not in arguments:
|
||||
# raise ValueError("缺少必要参数: device_id")
|
||||
# result = await nav_server.scanQRCode(
|
||||
# device_id=arguments["device_id"],
|
||||
# )
|
||||
else:
|
||||
raise ValueError(f"未知工具: {name}")
|
||||
return [TextContent(type="text", text=result)]
|
||||
except Exception as e:
|
||||
logger.error(f"工具调用失败: {str(e)}")
|
||||
raise ValueError(f"执行失败: {str(e)}")
|
||||
|
||||
options = server.create_initialization_options()
|
||||
async with stdio_server() as (read_stream, write_stream):
|
||||
await server.run(read_stream, write_stream, options)
|
||||
|
||||
def main():
|
||||
asyncio.run(serve())
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
113
terminal_temi_mcp/terminal_temi_mcp/mcp_mqtt.py
Normal file
113
terminal_temi_mcp/terminal_temi_mcp/mcp_mqtt.py
Normal file
@@ -0,0 +1,113 @@
|
||||
import paho.mqtt.client as mqtt
|
||||
import json
|
||||
import logging
|
||||
import threading
|
||||
from typing import Optional
|
||||
import uuid
|
||||
import threading
|
||||
import requests
|
||||
from os import getenv
|
||||
logging.basicConfig(level=logging.INFO)
|
||||
logger = logging.getLogger(__name__)
|
||||
MQTT_CLIENT_ID = f"MCPMQTT-{uuid.uuid4().hex[:8]}"
|
||||
|
||||
def getConfig_url(array):
|
||||
url = "http://lzwcai-demp-corp-manager:8086/system/config/getConfig"
|
||||
data = [array]
|
||||
try:
|
||||
response = requests.post(url, json=data, timeout=5)
|
||||
response.raise_for_status()
|
||||
data = response.json()['data']
|
||||
return data[0]['configValue']
|
||||
except Exception as e:
|
||||
print(f"Error fetching config for {array}: {e}")
|
||||
return None
|
||||
|
||||
class MQTTHandler:
|
||||
def __init__(self):
|
||||
self.client = mqtt.Client()
|
||||
self.tasks = {} # {task_id: task_data}
|
||||
self._lock = threading.Lock()
|
||||
|
||||
# 获取MQTT配置，提供默认值
|
||||
self.mqtt_username = getenv('MQTT_USERNAME') or 'lzwc'
|
||||
self.mqtt_password = getenv('MQTT_PASSWORD') or 'Lzwc@4187.'
|
||||
mqtt_broker_raw = getenv('MQTT_BROKER') or 'tcp://dempmana.lzwcai.com'
|
||||
# 移除协议前缀，只保留主机名
|
||||
self.mqtt_broker = mqtt_broker_raw.replace('tcp://', '').replace('mqtt://', '')
|
||||
mqtt_port_str = getenv('MQTT_PORT')
|
||||
self.mqtt_port = int(mqtt_port_str) if mqtt_port_str else 1883
|
||||
|
||||
# 记录配置信息
|
||||
logger.info(f"MQTT配置 - Broker: {self.mqtt_broker}, Port: {self.mqtt_port}, Username: {self.mqtt_username}")
|
||||
if not getenv('MQTT_BROKER'):
|
||||
logger.warning("MQTT_BROKER环境变量未设置，使用默认值")
|
||||
if not mqtt_port_str:
|
||||
logger.warning("MQTT_PORT环境变量未设置，使用默认端口1883")
|
||||
|
||||
self.client.username_pw_set(self.mqtt_username, self.mqtt_password)
|
||||
self.client.on_connect = self._on_connect
|
||||
self.client.on_message = self._on_message
|
||||
self.client.on_disconnect = self._on_disconnect
|
||||
try:
|
||||
logger.info(f"正在连接 MQTT 代理: {self.mqtt_broker}:{self.mqtt_port}")
|
||||
self.client.connect(self.mqtt_broker, self.mqtt_port, 60)
|
||||
self.client.loop_start()
|
||||
except Exception as e:
|
||||
logger.error(f"连接 MQTT 失败: {e}")
|
||||
|
||||
|
||||
def _on_connect(self, client, userdata, flags, rc):
|
||||
"""MQTT 连接回调"""
|
||||
if rc == 0:
|
||||
logger.info("已成功连接到 MQTT 代理服务器")
|
||||
else:
|
||||
logger.error(f"连接 MQTT 代理服务器失败，返回码: {rc}")
|
||||
|
||||
def get_status(self, task_id: str = None):
|
||||
"""获取任务状态
|
||||
Args:
|
||||
task_id: 可选参数，指定任务ID。如果为None则返回所有任务状态
|
||||
Returns:
|
||||
如果指定task_id则返回该任务的状态，否则返回所有任务状态
|
||||
"""
|
||||
with self._lock:
|
||||
if task_id:
|
||||
return self.tasks.get(task_id, {}).copy()
|
||||
return {k: v.copy() for k, v in self.tasks.items()}
|
||||
|
||||
def _on_message(self, client, userdata, msg):
|
||||
try:
|
||||
payload = json.loads(msg.payload.decode())
|
||||
logger.info(f"[MQTT] topic={msg.topic} msg: {payload}")
|
||||
|
||||
# task_id = payload.get('task_id')
|
||||
# if task_id and task_id in self.tasks:
|
||||
# with self._lock:
|
||||
# self.tasks[task_id].update({
|
||||
# 'status': payload.get('status', 'UNKNOWN'),
|
||||
# 'description': payload.get('description', '')
|
||||
# })
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"[错误] 处理消息失败: {e}")
|
||||
|
||||
def _on_disconnect(self, client, userdata, rc):
|
||||
"""MQTT 断开连接回调"""
|
||||
if rc != 0:
|
||||
logger.warning("意外断开与 MQTT 代理服务器的连接。")
|
||||
try:
|
||||
logger.info("尝试重新连接 MQTT...")
|
||||
client.reconnect()
|
||||
except Exception as e:
|
||||
logger.error(f"重新连接 MQTT 失败: {e}")
|
||||
|
||||
# 单例模式
|
||||
_instance: Optional[MQTTHandler] = None
|
||||
|
||||
def get_mcpmqtt_handler() -> MQTTHandler:
|
||||
"""获取单例"""
|
||||
global _instance
|
||||
if _instance is None:
|
||||
_instance = MQTTHandler()
|
||||
return _instance
|
||||
Reference in New Issue
Block a user