""" L1报文解析服务 — UDP协议 假设报文格式（固定帧结构，收到实际协议文档后对应调整）： ┌─────────────────────────────────────────────────────┐ │ Offset Size 说明 │ │ 0 2B 魔数 0xAA 0xBB │ │ 2 4B 报文类型 ASCII，如 "PC01" │ │ 6 2B 序列号 uint16 大端 │ │ 8 2B Body长度 uint16 大端 │ │ 10 2B 校验和 所有Body字节累加低16位 │ │ 12 N B Body GBK编码固定列宽文本 │ └─────────────────────────────────────────────────────┘ UDP最大包：65507字节，单帧不分片。 回执帧：Header相同结构，Body = "ACK" + 原序列号(2B) """ import asyncio import struct import time import uuid import json from datetime import datetime from typing import Optional, Dict, Any, Tuple from loguru import logger from app.config import settings # ─────────────────────────── 报文类型注册表 ─────────────────────────── MSG_TYPES: Dict[str, str] = { "PC01": "卷材入口报文", "PC02": "卷材出口报文", "PC03": "过程数据报文", "PC04": "质量数据报文", "PC05": "设备状态报文", "PC10": "计划下发报文", "PC11": "计划确认报文", "PC20": "心跳报文", } HEADER_SIZE = 12 # 报文头固定长度 MAGIC = b'\xAA\xBB' # ─────────────────────────── 校验和 ─────────────────────────── def _checksum(body: bytes) -> int: return sum(body) & 0xFFFF # ─────────────────────────── 报文头解析 ─────────────────────────── def parse_header(raw: bytes) -> Optional[Dict[str, Any]]: if len(raw) < HEADER_SIZE: logger.warning(f"报文过短: {len(raw)}B，丢弃") return None magic = raw[0:2] if magic != MAGIC: logger.warning(f"魔数错误: {magic.hex()}，丢弃") return None msg_type = raw[2:6].decode("ascii", errors="replace").strip() seq = struct.unpack(">H", raw[6:8])[0] body_len = struct.unpack(">H", raw[8:10])[0] checksum = struct.unpack(">H", raw[10:12])[0] body = raw[HEADER_SIZE: HEADER_SIZE + body_len] if len(body) < body_len: logger.warning(f"Body不完整: 期望{body_len}B 实际{len(body)}B") return None if _checksum(body) != checksum: logger.warning(f"校验和错误 [{msg_type}] seq={seq}，丢弃") return None return {"msg_type": msg_type, "seq": seq, "body": body, "body_str": body.decode("gbk", errors="replace")} # ─────────────────────────── 构建回执帧 ─────────────────────────── def build_ack(seq: int) -> bytes: body = b"ACK" + struct.pack(">H", seq) hdr = MAGIC hdr += b"ACK ".ljust(4)[:4] hdr += struct.pack(">H", seq) hdr += struct.pack(">H", len(body)) hdr += struct.pack(">H", _checksum(body)) return hdr + body # ─────────────────────────── 构建发送帧 ─────────────────────────── def build_frame(msg_type: str, body: bytes, seq: int = 0) -> bytes: hdr = MAGIC hdr += msg_type.encode("ascii").ljust(4)[:4] hdr += struct.pack(">H", seq) hdr += struct.pack(">H", len(body)) hdr += struct.pack(">H", _checksum(body)) return hdr + body # ─────────────────────────── Body解析器 ─────────────────────────── class PC01Parser: """卷材入口报文 Body固定列宽(GBK): 卷号(20) 钢种(10) 厚度(6) 宽度(6) 重量(8) 班次(2) """ def parse(self, body: str) -> Dict[str, Any]: return { "coil_no": body[0:20].strip(), "steel_grade": body[20:30].strip(), "thickness": _safe_float(body[30:36]), "width": _safe_float(body[36:42]), "weight": _safe_float(body[42:50]), "shift": body[50:52].strip(), "event_time": datetime.now(), } class PC02Parser: """卷材出口报文 Body: 卷号(20) 实测厚度(6) 实测宽度(6) 处理长度(8) 平均速度(6) 质量等级(2) """ def parse(self, body: str) -> Dict[str, Any]: return { "coil_no": body[0:20].strip(), "actual_thickness": _safe_float(body[20:26]), "actual_width": _safe_float(body[26:32]), "process_length": _safe_float(body[32:40]), "avg_speed": _safe_float(body[40:46]), "quality_grade": body[46:48].strip(), "event_time": datetime.now(), } class PC03Parser: """过程数据报文（周期推送） Body: 卷号(20) 位置(10) 速度(6) 入口张力(8) 出口张力(8) 酸液温度(6) """ def parse(self, body: str) -> Dict[str, Any]: return { "coil_no": body[0:20].strip(), "position": body[20:30].strip(), "speed": _safe_float(body[30:36]), "tension_inlet": _safe_float(body[36:44]), "tension_outlet": _safe_float(body[44:52]), "acid_temp": _safe_float(body[52:58]), "event_time": datetime.now(), } class PC04Parser: """质量数据报文 Body: 卷号(20) 缺陷类型(10) 缺陷位置(8) 严重程度(2) """ def parse(self, body: str) -> Dict[str, Any]: return { "coil_no": body[0:20].strip(), "defect_type": body[20:30].strip(), "defect_pos": _safe_float(body[30:38]), "severity": body[38:40].strip(), "event_time": datetime.now(), } class PC05Parser: """设备状态报文 Body: 设备编号(10) 状态码(4) 故障码(6) 时间戳(14 yyyyMMddHHmmss) """ def parse(self, body: str) -> Dict[str, Any]: ts_str = body[20:34].strip() try: ts = datetime.strptime(ts_str, "%Y%m%d%H%M%S") except Exception: ts = datetime.now() return { "equipment_code": body[0:10].strip(), "status_code": body[10:14].strip(), "fault_code": body[14:20].strip(), "event_time": ts, } class PC20Parser: """心跳报文 Body: 时间戳(14)""" def parse(self, body: str) -> Dict[str, Any]: return {"event_time": datetime.now(), "raw_ts": body[0:14].strip()} def _safe_float(s: str) -> float: try: return float(s.strip()) except (ValueError, AttributeError): return 0.0 BODY_PARSERS: Dict[str, Any] = { "PC01": PC01Parser(), "PC02": PC02Parser(), "PC03": PC03Parser(), "PC04": PC04Parser(), "PC05": PC05Parser(), "PC20": PC20Parser(), } # ─────────────────────────── 分发器 ─────────────────────────── class MessageDispatcher: def __init__(self): self._handlers: Dict[str, list] = {} def register(self, msg_type: str): def decorator(func): self._handlers.setdefault(msg_type, []).append(func) return func return decorator async def dispatch(self, msg_type: str, data: Dict[str, Any]): for handler in self._handlers.get(msg_type, []): try: await handler(data) except Exception as e: logger.error(f"报文处理器异常 [{msg_type}]: {e}") dispatcher = MessageDispatcher() # ─────────────────────────── UDP 服务端 ─────────────────────────── class L1UdpProtocol(asyncio.DatagramProtocol): """asyncio UDP DatagramProtocol 实现""" def __init__(self, server: "L1UdpServer"): self._server = server self.transport: Optional[asyncio.DatagramTransport] = None def connection_made(self, transport: asyncio.DatagramTransport): self.transport = transport host, port = transport.get_extra_info("sockname") logger.info(f"UDP监听启动: {host}:{port}") def datagram_received(self, data: bytes, addr: Tuple[str, int]): asyncio.create_task(self._server.handle(data, addr, self.transport)) def error_received(self, exc: Exception): logger.error(f"UDP错误: {exc}") def connection_lost(self, exc): logger.warning(f"UDP连接丢失: {exc}") class L1UdpServer: """UDP报文接收服务""" def __init__(self): self._transport: Optional[asyncio.DatagramTransport] = None self._running = False # 统计 self.recv_count = 0 self.error_count = 0 async def start(self): self._running = True loop = asyncio.get_running_loop() self._transport, _ = await loop.create_datagram_endpoint( lambda: L1UdpProtocol(self), local_addr=(settings.L1_HOST, settings.L1_PORT), ) logger.info(f"L1 UDP服务已启动，监听 {settings.L1_HOST}:{settings.L1_PORT}") async def handle(self, raw: bytes, addr: Tuple[str, int], transport: asyncio.DatagramTransport): t0 = time.time() self.recv_count += 1 logger.debug(f"收到UDP包 from {addr[0]}:{addr[1]} {len(raw)}B") header = parse_header(raw) if not header: self.error_count += 1 await self._save_log(raw, addr, None, "error", "报文头解析失败", t0) return msg_type = header["msg_type"] seq = header["seq"] # 发送ACK回执 if msg_type != "ACK": ack = build_ack(seq) transport.sendto(ack, addr) # 心跳不做业务处理 if msg_type == "PC20": logger.debug(f"心跳 seq={seq}") return # Body解析 body_parser = BODY_PARSERS.get(msg_type) data: Dict[str, Any] = {} if body_parser: try: data = body_parser.parse(header["body_str"]) except Exception as e: logger.error(f"Body解析异常 [{msg_type}]: {e}") self.error_count += 1 await self._save_log(raw, addr, header, "error", str(e), t0) return else: logger.warning(f"未知报文类型: {msg_type}") elapsed_ms = (time.time() - t0) * 1000 logger.info(f"[{msg_type}] seq={seq} from {addr[0]} 耗时{elapsed_ms:.1f}ms") await self._save_log(raw, addr, header, "success", None, t0, data) await dispatcher.dispatch(msg_type, data) async def _save_log(self, raw: bytes, addr: Tuple[str, int], header: Optional[Dict], status: str, error_msg: Optional[str], t0: float, parsed_data: Optional[Dict] = None): try: from app.database import AsyncSessionLocal from app.models.message import MessageLog elapsed_ms = (time.time() - t0) * 1000 async with AsyncSessionLocal() as db: log = MessageLog( msg_id=str(uuid.uuid4())[:16], msg_type=header["msg_type"] if header else "UNKNOWN", direction="recv", source=f"{addr[0]}:{addr[1]}", raw_data=raw.hex(), parsed_data=json.dumps(parsed_data, default=str) if parsed_data else None, status=status, error_msg=error_msg, process_time=round(elapsed_ms, 2), received_at=datetime.now(), ) db.add(log) await db.commit() except Exception as e: logger.error(f"保存报文日志失败: {e}") def send(self, data: bytes, addr: Tuple[str, int]): """主动向L1发送报文""" if self._transport: self._transport.sendto(data, addr) else: raise RuntimeError("UDP服务未启动") def stop(self): self._running = False if self._transport: self._transport.close() # 全局单例 l1_server = L1UdpServer()