Compare commits
	
		
			13 Commits
		
	
	
		
			7bc7a95379
			...
			main
		
	
	| Author | SHA1 | Date | |
|---|---|---|---|
| e3167a0144 | |||
| 452aaaeaba | |||
| c0a1925150 | |||
| 6d50e0a810 | |||
| b69805063d | |||
| c025530c25 | |||
| 902c90cf19 | |||
| 961b0c170b | |||
| c3d4531870 | |||
| 5f81314540 | |||
| d5de3a7a2b | |||
| 97a9c778e8 | |||
| dfb50f5c74 | 
							
								
								
									
										216
									
								
								README.md
									
									
									
									
									
								
							
							
						
						
									
										216
									
								
								README.md
									
									
									
									
									
								
							@@ -1,194 +1,44 @@
 | 
				
			|||||||
# 猪舍主控
 | 
					# pig-house-controller
 | 
				
			||||||
 | 
					
 | 
				
			||||||
## 简介
 | 
					## LoRa通信协议约定
 | 
				
			||||||
 | 
					
 | 
				
			||||||
猪舍主控系统根据上位机的指令控制当前猪舍内所有设备(传感器,阀门,电机等),并汇聚当前猪舍传感器数据统一上报。本系统作为猪场智能化管理的重要组成部分,实现了猪舍环境的自动化监控与调节。
 | 
					本项目中的LoRa通信采用自定义的帧格式,以支持精确寻址和大数据包的自动分片与重组。所有数据包均由主控节点主动发起。
 | 
				
			||||||
 | 
					
 | 
				
			||||||
## 功能概述
 | 
					### 1. 物理帧结构
 | 
				
			||||||
 | 
					
 | 
				
			||||||
### 与上位机交互
 | 
					每个通过LoRa UART模块发送的物理数据包都遵循以下结构:
 | 
				
			||||||
1. 根据上位机指令定期采集栏内所有传感器的数据,并统一上报
 | 
					 | 
				
			||||||
2. 根据上位机指令启动或关闭栏内设备,上位机发送的启动指令分两种:
 | 
					 | 
				
			||||||
    - 常开指令:收到后启动设备,直到收到关闭指令后关闭设备
 | 
					 | 
				
			||||||
    - 短暂开启指令:收到后启动设备,但需要上位机每过两秒发送一次指令,超过五秒没收到下一个开启指令或受到关闭指令将会关闭设备
 | 
					 | 
				
			||||||
3. 定期检查栏内设备状态,发现异常立即上报上位机
 | 
					 | 
				
			||||||
4. 定期向上位机发送心跳包
 | 
					 | 
				
			||||||
5. 接收上位机发送的总线上各机器的位置和类型
 | 
					 | 
				
			||||||
6. 根据上位机指令调整设备功率大小
 | 
					 | 
				
			||||||
7. 接收上位机批量控制指令并执行
 | 
					 | 
				
			||||||
8. 接收上位机发送的配置信息
 | 
					 | 
				
			||||||
 | 
					
 | 
				
			||||||
### 与设备交互
 | 
					| 字段                       | 长度 (字节) | 值 (Hex)          | 描述                            |
 | 
				
			||||||
1. 控制栏内设备启停
 | 
					|:-------------------------| :------------ | :---------------- |:------------------------------|
 | 
				
			||||||
2. 调整风机等功率可调设备的功率
 | 
					| **帧头 (Header)**          | 1             | `0xED`            | 固定值,表示一个数据包的开始。               |
 | 
				
			||||||
3. 定时检查栏内设备状态
 | 
					| **数据长度 (Length)**        | 1             | `0x00`-`0xFF`     | 从`总包数`字段到`数据块`末尾的总字节数,不包含源地址。 |
 | 
				
			||||||
4. 定时采集栏内数据
 | 
					| **目标地址 (DestAddr)**      | 2             | `0x0000`-`0xFFFF` | 接收该数据包的设备地址。                  |
 | 
				
			||||||
 | 
					| **总包数 (TotalChunks)**    | 1             | `0x01`-`0xFF`     | 表示当前消息被分成了几个包。`0x01`代表这是唯一的包。 |
 | 
				
			||||||
 | 
					| **当前包序号 (CurrentChunk)** | 1             | `0x00`-`0xFE`     | 当前是第几个数据包(从0开始计数)。            |
 | 
				
			||||||
 | 
					| **数据块 (ChunkData)**      | N             | -                 | 实际传输的数据片段。                    |
 | 
				
			||||||
 | 
					| **源地址 (SourceAddr)**      | 2             | `0x0000`-`0xFFFF` | 发送该数据包的设备地址,由硬件自动拼接。          |
 | 
				
			||||||
 | 
					
 | 
				
			||||||
### 数据管理
 | 
					**示例:**
 | 
				
			||||||
1. 保存总线上各机器的位置和类型
 | 
					 | 
				
			||||||
2. 临时保存上位机发送的指令
 | 
					 | 
				
			||||||
3. 保存上位机发送的配置信息
 | 
					 | 
				
			||||||
4. 汇总栏内所有传感器数据
 | 
					 | 
				
			||||||
5. 临时保存栏内设备故障信息,直到上报成功后清除
 | 
					 | 
				
			||||||
6. 根据批量指令控制对应设备工作
 | 
					 | 
				
			||||||
 | 
					
 | 
				
			||||||
# 猪舍控制器
 | 
					发送一个数据为 `[0x01, 0x02, 0x03]` 的单包消息到地址 `0x1234`,发送方地址为 `0x5678`:
 | 
				
			||||||
 | 
					`ED 05 12 34 01 00 01 02 03 56 78`
 | 
				
			||||||
 | 
					- `ED`: 帧头
 | 
				
			||||||
 | 
					- `05`: 后续长度 (1+1+3 = 5)
 | 
				
			||||||
 | 
					- `12 34`: 目标地址
 | 
				
			||||||
 | 
					- `01`: 总包数 (共1包)
 | 
				
			||||||
 | 
					- `00`: 当前包序号 (第0包)
 | 
				
			||||||
 | 
					- `01 02 03`: 数据块
 | 
				
			||||||
 | 
					- `56 78`: 源地址
 | 
				
			||||||
 | 
					
 | 
				
			||||||
猪舍控制器是一个用于监控和控制猪舍环境的系统。它可以通过LoRa与上位机通信,并通过RS485总线控制传感器和执行器设备。
 | 
					### 2. 数据分片 (Fragmentation)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
## 协议栈和技术选型
 | 
					- LoRa模块的物理层限制单次发送的数据部分**最大为240字节**。
 | 
				
			||||||
 | 
					- 根据项目约定,为自定义协议头(总包数、当前包序号)预留2字节,地址由模块处理。
 | 
				
			||||||
 | 
					- 因此,每个物理包中 **`数据块 (ChunkData)` 的最大长度为 `238` 字节**。
 | 
				
			||||||
 | 
					- `send_packet` 方法会自动处理分片逻辑。
 | 
				
			||||||
 | 
					
 | 
				
			||||||
本系统采用以下物联网标准协议栈:
 | 
					### 3. 数据重组 (Reassembly)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
### 物理层
 | 
					- `receive_packet` 方法会缓存收到的分片。
 | 
				
			||||||
- **LoRa**:低功耗广域网物理层技术,提供远距离无线传输能力
 | 
					- 当一个设备的所有分片都接收完毕后,`receive_packet` 会将它们自动重组成一个完整的消息,并向上层返回。
 | 
				
			||||||
 | 
					- 由于通信是单向的(仅主控发送),接收端无需管理多个源地址的重组缓冲区。
 | 
				
			||||||
### 数据链路层和网络层
 | 
					 | 
				
			||||||
- **LoRaWAN**:基于LoRa物理层的广域网协议,提供设备认证、加密和网络管理
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
### 传输层
 | 
					 | 
				
			||||||
- **CoAP**:受限应用协议,轻量级的RESTful协议,适用于资源受限设备
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
### 应用层
 | 
					 | 
				
			||||||
- **LwM2M**:轻量级机器到机器协议,提供设备管理、固件更新等功能
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
### 数据格式
 | 
					 | 
				
			||||||
- **SenML**:传感器标记语言,标准化的传感器数据表示格式
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
这种协议栈选择具有以下优势:
 | 
					 | 
				
			||||||
1. **低功耗**:适合电池供电或节能要求高的场景
 | 
					 | 
				
			||||||
2. **远距离传输**:LoRa技术可实现数公里覆盖
 | 
					 | 
				
			||||||
3. **标准化**:采用业界标准协议,便于系统集成和扩展
 | 
					 | 
				
			||||||
4. **安全性**:LoRaWAN和CoAP均提供安全机制
 | 
					 | 
				
			||||||
5. **互操作性**:基于标准协议,便于与不同厂商设备集成
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
## 系统架构
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
```
 | 
					 | 
				
			||||||
猪舍控制器
 | 
					 | 
				
			||||||
├── 通信层 (LoRa)
 | 
					 | 
				
			||||||
├── 控制层 (核心逻辑)
 | 
					 | 
				
			||||||
├── 设备层 (传感器和执行器)
 | 
					 | 
				
			||||||
└── 存储层 (数据存储)
 | 
					 | 
				
			||||||
```
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
## 抽象接口设计
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
为了提高系统的可扩展性和可维护性,我们定义了以下抽象接口:
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
### 通信接口 (BaseComm)
 | 
					 | 
				
			||||||
定义了通信模块的基本操作,包括连接、断开连接、发送和接收数据等方法。
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
### 设备接口 (BaseDevice)
 | 
					 | 
				
			||||||
定义了设备的基本操作,包括连接、断开连接、读取数据、写入数据和获取状态等方法。
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
### 存储接口 (BaseStorage)
 | 
					 | 
				
			||||||
定义了存储模块的基本操作,包括保存、加载、删除数据等方法。
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
### 命令处理器接口 (BaseHandler)
 | 
					 | 
				
			||||||
定义了命令处理的基本操作,包括处理命令、注册和注销命令处理函数等方法。
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
## 设计理念
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
按照功能区分将传感器和执行器分别连接到不同的RS485总线上,可以带来以下优势:
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
1. **减少总线负载**:传感器通常需要频繁读取数据,而执行器可能需要较大的电流,分离可以避免相互干扰。
 | 
					 | 
				
			||||||
2. **提高响应速度**:控制命令可以直接发送到执行器总线,无需等待传感器数据采集完成。
 | 
					 | 
				
			||||||
3. **增强系统稳定性**:一条总线故障不会影响另一条总线上的设备。
 | 
					 | 
				
			||||||
4. **便于维护**:可以根据需要单独重启或维护某一总线。
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
## 配置说明
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
系统配置文件为 `config.json`,如果不存在,系统会根据默认配置创建。配置项包括:
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
### LoRa通信配置
 | 
					 | 
				
			||||||
- `lora.address`: 本机LoRa地址
 | 
					 | 
				
			||||||
- `lora.frequency`: 工作频率(MHz)
 | 
					 | 
				
			||||||
- `lora.bandwidth`: 带宽(kHz)
 | 
					 | 
				
			||||||
- `lora.spreading_factor`: 扩频因子
 | 
					 | 
				
			||||||
- `lora.coding_rate`: 编码率
 | 
					 | 
				
			||||||
- `lora.encryption_key`: 加密密钥
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
### 上位机配置
 | 
					 | 
				
			||||||
- `master.lora_address`: 上位机LoRa地址
 | 
					 | 
				
			||||||
- `master.protocol`: 与上位机通信协议
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
### 总线配置
 | 
					 | 
				
			||||||
- `bus.sensor.port`: 传感器总线串口
 | 
					 | 
				
			||||||
- `bus.sensor.baudrate`: 传感器总线波特率
 | 
					 | 
				
			||||||
- `bus.actuator.port`: 执行器总线串口
 | 
					 | 
				
			||||||
- `bus.actuator.baudrate`: 执行器总线波特率
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
### 日志配置
 | 
					 | 
				
			||||||
- `log.level`: 日志级别 (DEBUG, INFO, WARNING, ERROR)
 | 
					 | 
				
			||||||
- `log.file_path`: 日志文件路径
 | 
					 | 
				
			||||||
- `log.max_size`: 日志文件最大大小
 | 
					 | 
				
			||||||
- `log.backup_count`: 保留的日志文件数量
 | 
					 | 
				
			||||||
- `log.report_errors`: 是否上报错误信息
 | 
					 | 
				
			||||||
- `log.terminate_on_report_failure`: 错误上报失败时是否终止程序
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
### 系统参数
 | 
					 | 
				
			||||||
- `system.heartbeat_interval`: 心跳包发送间隔(秒)
 | 
					 | 
				
			||||||
- `system.data_collection_interval`: 数据采集间隔(秒)
 | 
					 | 
				
			||||||
- `system.command_timeout`: 命令超时时间(秒)
 | 
					 | 
				
			||||||
- `system.retry_count`: 命令重试次数
 | 
					 | 
				
			||||||
- `system.error_handling`: 错误处理策略
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
### 设备配置
 | 
					 | 
				
			||||||
- `devices`: 设备列表(包括传感器和执行器)
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
每个设备包含以下属性:
 | 
					 | 
				
			||||||
- `id`: 设备唯一标识
 | 
					 | 
				
			||||||
- `type`: 设备类型
 | 
					 | 
				
			||||||
- `address`: 设备地址
 | 
					 | 
				
			||||||
- `bus`: 所在总线(sensor/actuator)
 | 
					 | 
				
			||||||
- `location`: 设备位置(可选)
 | 
					 | 
				
			||||||
- `unit`: 单位(仅传感器需要,如温度单位、湿度单位等)
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
参考示例配置文件 `config.json.example` 创建您的配置文件。
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
## 枚举类型定义
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
为了提高代码的可读性和维护性,系统定义了以下枚举类型:
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
1. `LogLevel`: 日志等级枚举 (DEBUG, INFO, WARNING, ERROR, CRITICAL)
 | 
					 | 
				
			||||||
2. `DeviceType`: 设备类型枚举 (包括温度、湿度等传感器类型和喂料口、阀门等执行器类型)
 | 
					 | 
				
			||||||
3. `BusType`: 总线类型枚举 (SENSOR, ACTUATOR)
 | 
					 | 
				
			||||||
4. `ErrorHandlingStrategy`: 错误处理策略枚举 (RETRY, SKIP, ALERT)
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
## 日志和错误处理机制
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
考虑到树莓派等嵌入式设备的存储空间限制,系统采用以下策略:
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
1. **限制日志文件大小**:默认将日志文件大小限制为1MB,仅保留一个备份文件
 | 
					 | 
				
			||||||
2. **错误上报机制**:当发生错误时,系统会尝试通过LoRa将错误信息上报给上位机
 | 
					 | 
				
			||||||
3. **上报成功处理**:错误信息上报成功后,系统会删除本地日志中的该条目
 | 
					 | 
				
			||||||
4. **上报失败处理**:如果错误信息上报失败,说明与上位机之间的通信不稳定,系统将根据配置决定是否终止程序运行
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
这种机制既节省了本地存储空间,又能确保关键错误信息能够及时传递给上位机。
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
## 开发顺序建议
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
当然,从对其他模块依赖最小的模块开始开发是一个明智的策略,以便逐步构建项目的基础。以下是建议的开发顺序:
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
1. **配置模块(config.py)**:首先实现配置模块,以定义应用程序所需的基本配置。这将为其他模块提供必要的设置。
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
2. **实用程序模块(utils/)**:开发实用程序函数,这些函数可以在整个项目中被重复使用。这样可以为其他模块提供基本的辅助功能。
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
3. **数据存储模块(storage/)**:实现数据存储逻辑,包括存储传感器数据、设备状态和配置信息。这一模块可以在独立于其他模块的情况下开发。
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
4. **设备交互模块(devices/)**:实现与设备交互的模块,包括传感器和执行器。这将为之后的通信和核心逻辑提供基础。
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
5. **通信模块(comms/)**:开发通信模块,以处理与上位机和设备的通信协议。此模块可能需要依赖设备交互模块。
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
6. **核心逻辑模块(core/)**:实现核心逻辑,包括处理命令、控制设备和管理传感器数据。这一模块将利用之前开发的模块。
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
7. **测试模块(tests/)**:在开发过程中,逐步添加测试用例以验证每个模块的功能。
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
8. **主程序(main.py)**:最后实现主程序,作为应用程序的入口点,将所有模块整合在一起。
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
这种顺序将帮助你逐步构建项目,并确保每个模块在开发过程中得到充分的测试和验证。
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
## 许可证
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
禁止未经授权使用本项目代码,否则后果自负。
 | 
					 | 
				
			||||||
@@ -7,9 +7,7 @@ RS485 总线管理器实现
 | 
				
			|||||||
此模块实现了 IBusManager 接口,用于管理 RS485 总线通信。
 | 
					此模块实现了 IBusManager 接口,用于管理 RS485 总线通信。
 | 
				
			||||||
"""
 | 
					"""
 | 
				
			||||||
 | 
					
 | 
				
			||||||
from .bus_interface import IBusManager
 | 
					from ..logs.logger import log
 | 
				
			||||||
from typing import Dict, Any
 | 
					 | 
				
			||||||
from main.logs.logger import log
 | 
					 | 
				
			||||||
 | 
					
 | 
				
			||||||
# 导入 MicroPython 的 UART 和 Pin 库
 | 
					# 导入 MicroPython 的 UART 和 Pin 库
 | 
				
			||||||
from machine import UART, Pin
 | 
					from machine import UART, Pin
 | 
				
			||||||
@@ -18,26 +16,26 @@ import _thread  # 用于线程同步
 | 
				
			|||||||
import struct  # 用于浮点数转换
 | 
					import struct  # 用于浮点数转换
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
class RS485Manager(IBusManager):
 | 
					class RS485Manager:
 | 
				
			||||||
    """
 | 
					    """
 | 
				
			||||||
    RS485 总线管理器。
 | 
					    RS485 总线管理器。
 | 
				
			||||||
    负责 RS485 设备的指令发送、响应接收和数据解析。
 | 
					    负责 RS485 设备的指令发送、响应接收和数据解析。
 | 
				
			||||||
    """
 | 
					    """
 | 
				
			||||||
 | 
					
 | 
				
			||||||
    def __init__(self, bus_config: Dict[int, Dict[str, Any]], default_timeouts: Dict[str, int]):
 | 
					    def __init__(self, bus_config, default_timeouts):
 | 
				
			||||||
        """
 | 
					        """
 | 
				
			||||||
        构造函数,注入配置。
 | 
					        构造函数,注入配置。
 | 
				
			||||||
        根据传入的配置初始化 RS485 总线对应的 UART 管理器。
 | 
					        根据传入的配置初始化 RS485 总线对应的 UART 管理器。
 | 
				
			||||||
 | 
					
 | 
				
			||||||
        Args:
 | 
					        Args:
 | 
				
			||||||
            bus_config (Dict[int, Dict[str, Any]]): 包含所有总线配置的字典。
 | 
					            bus_config: 包含所有总线配置的字典。
 | 
				
			||||||
                                                    键是总线ID,值是该总线的详细配置。
 | 
					                                                    键是总线ID,值是该总线的详细配置。
 | 
				
			||||||
            default_timeouts (Dict[str, int]): 包含各种默认超时设置的字典。
 | 
					            default_timeouts: 包含各种默认超时设置的字典。
 | 
				
			||||||
        """
 | 
					        """
 | 
				
			||||||
        self.bus_config = bus_config
 | 
					        self.bus_config = bus_config
 | 
				
			||||||
        self.default_timeouts = default_timeouts
 | 
					        self.default_timeouts = default_timeouts
 | 
				
			||||||
        # 存储以总线号为key的UART管理器实例、RTS引脚和锁
 | 
					        # 存储以总线号为key的UART管理器实例、RTS引脚和锁
 | 
				
			||||||
        self.bus_ports: Dict[int, Dict[str, Any]] = {}
 | 
					        self.bus_ports = {}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
        log("RS485Manager 已使用配置初始化。")
 | 
					        log("RS485Manager 已使用配置初始化。")
 | 
				
			||||||
        log(f"总线配置: {self.bus_config}")
 | 
					        log(f"总线配置: {self.bus_config}")
 | 
				
			||||||
@@ -77,7 +75,8 @@ class RS485Manager(IBusManager):
 | 
				
			|||||||
            else:
 | 
					            else:
 | 
				
			||||||
                log(f"总线 {bus_id} 的协议不是 RS485,跳过初始化。")
 | 
					                log(f"总线 {bus_id} 的协议不是 RS485,跳过初始化。")
 | 
				
			||||||
 | 
					
 | 
				
			||||||
    def _calculate_crc16_modbus(self, data: bytes) -> int:
 | 
					    @staticmethod
 | 
				
			||||||
 | 
					    def _calculate_crc16_modbus(data):
 | 
				
			||||||
        """
 | 
					        """
 | 
				
			||||||
        计算 Modbus RTU 的 CRC16 校验码。
 | 
					        计算 Modbus RTU 的 CRC16 校验码。
 | 
				
			||||||
        """
 | 
					        """
 | 
				
			||||||
@@ -85,69 +84,14 @@ class RS485Manager(IBusManager):
 | 
				
			|||||||
        for byte in data:
 | 
					        for byte in data:
 | 
				
			||||||
            crc ^= byte
 | 
					            crc ^= byte
 | 
				
			||||||
            for _ in range(8):
 | 
					            for _ in range(8):
 | 
				
			||||||
                if (crc & 0x0001):
 | 
					                if crc & 0x0001:
 | 
				
			||||||
                    crc >>= 1
 | 
					                    crc >>= 1
 | 
				
			||||||
                    crc ^= 0xA001
 | 
					                    crc ^= 0xA001
 | 
				
			||||||
                else:
 | 
					                else:
 | 
				
			||||||
                    crc >>= 1
 | 
					                    crc >>= 1
 | 
				
			||||||
        return crc
 | 
					        return crc
 | 
				
			||||||
 | 
					
 | 
				
			||||||
    def _parse_modbus_rtu_float_default(self, response_bytes: bytes) -> float | None:
 | 
					    def execute_raw_command(self, bus_id, command):
 | 
				
			||||||
        """
 | 
					 | 
				
			||||||
        默认解析 Modbus RTU 响应中的 32 位 IEEE 754 单精度浮点数。
 | 
					 | 
				
			||||||
        假定为大端 (ABCD) 字节序。
 | 
					 | 
				
			||||||
        """
 | 
					 | 
				
			||||||
        # 最小预期长度: 从站ID(1) + 功能码(1) + 字节计数(1) + 4字节数据 + CRC(2) = 9字节
 | 
					 | 
				
			||||||
        MIN_RESPONSE_LEN = 9
 | 
					 | 
				
			||||||
        EXPECTED_DATA_BYTE_COUNT = 4  # 32位浮点数占用4字节
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
        if not response_bytes or len(response_bytes) < MIN_RESPONSE_LEN:
 | 
					 | 
				
			||||||
            log(f"警告: 响应字节过短或为空,无法解析为浮点数。响应: {response_bytes.hex() if response_bytes else 'None'}")
 | 
					 | 
				
			||||||
            return None
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
        # 提取响应组件
 | 
					 | 
				
			||||||
        # 注意: Modbus RTU CRC是LSB在前,所以这里需要调整
 | 
					 | 
				
			||||||
        # response_bytes[:-2] 是用于CRC计算的数据部分
 | 
					 | 
				
			||||||
        # response_bytes[-2:] 是CRC本身
 | 
					 | 
				
			||||||
        data_for_crc = response_bytes[:-2]
 | 
					 | 
				
			||||||
        received_crc = (response_bytes[-1] << 8) | response_bytes[-2] # CRC的低字节在前,高字节在后
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
        # 1. CRC 校验
 | 
					 | 
				
			||||||
        calculated_crc = self._calculate_crc16_modbus(data_for_crc)
 | 
					 | 
				
			||||||
        if calculated_crc != received_crc:
 | 
					 | 
				
			||||||
            log(f"错误: CRC校验失败。接收CRC: {received_crc:04X}, 计算CRC: {calculated_crc:04X}. 响应: {response_bytes.hex()}")
 | 
					 | 
				
			||||||
            return None
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
        slave_id = response_bytes[0]
 | 
					 | 
				
			||||||
        function_code = response_bytes[1]
 | 
					 | 
				
			||||||
        byte_count = response_bytes[2]
 | 
					 | 
				
			||||||
        data_bytes = response_bytes[3:3 + EXPECTED_DATA_BYTE_COUNT]
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
        # 2. 功能码检查 (假设读取保持寄存器0x03或输入寄存器0x04)
 | 
					 | 
				
			||||||
        if function_code not in [0x03, 0x04]:
 | 
					 | 
				
			||||||
            log(f"警告: 响应功能码 {function_code:02X} 不符合预期 (期望0x03或0x04)。响应: {response_bytes.hex()}")
 | 
					 | 
				
			||||||
            return None
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
        # 3. 字节计数检查
 | 
					 | 
				
			||||||
        if byte_count != EXPECTED_DATA_BYTE_COUNT:
 | 
					 | 
				
			||||||
            log(f"警告: 响应字节计数 {byte_count} 不符合预期 (期望{EXPECTED_DATA_BYTE_COUNT})。响应: {response_bytes.hex()}")
 | 
					 | 
				
			||||||
            return None
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
        # 4. 提取的数据字节长度检查 (与字节计数检查有重叠,但更安全)
 | 
					 | 
				
			||||||
        if len(data_bytes) != EXPECTED_DATA_BYTE_COUNT:
 | 
					 | 
				
			||||||
            log(f"错误: 提取的数据字节长度不正确。期望{EXPECTED_DATA_BYTE_COUNT}, 实际{len(data_bytes)}. 响应: {response_bytes.hex()}")
 | 
					 | 
				
			||||||
            return None
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
        # 5. 转换为浮点数 (大端, ABCD)
 | 
					 | 
				
			||||||
        try:
 | 
					 | 
				
			||||||
            parsed_float = struct.unpack('>f', data_bytes)[0]
 | 
					 | 
				
			||||||
            log(f"成功解析浮点数: {parsed_float}")
 | 
					 | 
				
			||||||
            return parsed_float
 | 
					 | 
				
			||||||
        except Exception as e:
 | 
					 | 
				
			||||||
            log(f"错误: 浮点数转换失败: {e}. 数据字节: {data_bytes.hex()}. 响应: {response_bytes.hex()}")
 | 
					 | 
				
			||||||
            return None
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
    def execute_raw_command(self, bus_id: int, command: bytes) -> None:
 | 
					 | 
				
			||||||
        """
 | 
					        """
 | 
				
			||||||
        【契约】执行一个“发后不理”的原始指令。
 | 
					        【契约】执行一个“发后不理”的原始指令。
 | 
				
			||||||
 | 
					
 | 
				
			||||||
@@ -176,7 +120,7 @@ class RS485Manager(IBusManager):
 | 
				
			|||||||
            except Exception as e:
 | 
					            except Exception as e:
 | 
				
			||||||
                log(f"错误: 在总线 {bus_id} 上执行原始命令失败: {e}")
 | 
					                log(f"错误: 在总线 {bus_id} 上执行原始命令失败: {e}")
 | 
				
			||||||
 | 
					
 | 
				
			||||||
    def execute_collect_task(self, task: dict) -> float | None:
 | 
					    def execute_collect_task(self, task):
 | 
				
			||||||
        """
 | 
					        """
 | 
				
			||||||
        【契约】执行一个完整的采集任务,并直接返回最终的数值。
 | 
					        【契约】执行一个完整的采集任务,并直接返回最终的数值。
 | 
				
			||||||
 | 
					
 | 
				
			||||||
@@ -188,11 +132,11 @@ class RS485Manager(IBusManager):
 | 
				
			|||||||
        - 返回最终的float数值,或在任何失败情况下返回None。
 | 
					        - 返回最终的float数值,或在任何失败情况下返回None。
 | 
				
			||||||
 | 
					
 | 
				
			||||||
        Args:
 | 
					        Args:
 | 
				
			||||||
            task (dict): 从Protobuf解码出的单个CollectTask消息字典。
 | 
					            task: 从Protobuf解码出的单个CollectTask消息字典。
 | 
				
			||||||
                         期望结构: {"command": {"bus_number": int, "command_bytes": bytes}}
 | 
					                         期望结构: {"command": {"bus_number": int, "command_bytes": bytes}}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
        Returns:
 | 
					        Returns:
 | 
				
			||||||
            float | None: 成功解析则返回数值,否则返回None。
 | 
					            成功解析则返回数值,否则返回None。
 | 
				
			||||||
        """
 | 
					        """
 | 
				
			||||||
        # I. 任务参数解析与初步验证
 | 
					        # I. 任务参数解析与初步验证
 | 
				
			||||||
        try:
 | 
					        try:
 | 
				
			||||||
@@ -204,8 +148,9 @@ class RS485Manager(IBusManager):
 | 
				
			|||||||
            bus_id = command_info.get("bus_number")
 | 
					            bus_id = command_info.get("bus_number")
 | 
				
			||||||
            command_bytes = command_info.get("command_bytes")
 | 
					            command_bytes = command_info.get("command_bytes")
 | 
				
			||||||
 | 
					
 | 
				
			||||||
            if bus_id is None or command_bytes is None:
 | 
					            # 增加对命令有效性的检查
 | 
				
			||||||
                log("错误: Raw485Command 缺少 'bus_number' 或 'command_bytes' 字段。")
 | 
					            if bus_id is None or not command_bytes or len(command_bytes) < 2:
 | 
				
			||||||
 | 
					                log(f"错误: CollectTask 的 'command' 字段无效。bus_id: {bus_id}, command_bytes: {command_bytes}")
 | 
				
			||||||
                return None
 | 
					                return None
 | 
				
			||||||
 | 
					
 | 
				
			||||||
        except Exception as e:
 | 
					        except Exception as e:
 | 
				
			||||||
@@ -221,34 +166,135 @@ class RS485Manager(IBusManager):
 | 
				
			|||||||
        rts_pin = port_info['rts_pin']
 | 
					        rts_pin = port_info['rts_pin']
 | 
				
			||||||
        lock = port_info['lock']
 | 
					        lock = port_info['lock']
 | 
				
			||||||
 | 
					
 | 
				
			||||||
        response_bytes = None
 | 
					        response_bytes = None  # 在锁外部初始化,确保其作用域
 | 
				
			||||||
 | 
					        response_buffer = bytearray()
 | 
				
			||||||
        with lock:
 | 
					        with lock:
 | 
				
			||||||
            try:
 | 
					            try:
 | 
				
			||||||
                # II. 线程安全与指令发送
 | 
					                # II. 线程安全与指令发送
 | 
				
			||||||
                rts_pin.value(1)  # 设置为发送模式 (DE/RE = HIGH)
 | 
					                rts_pin.value(1)
 | 
				
			||||||
                time.sleep_us(100)  # 短暂延时,确保方向切换完成
 | 
					                time.sleep_us(100)
 | 
				
			||||||
                uart.write(command_bytes)
 | 
					                uart.write(command_bytes)
 | 
				
			||||||
                uart.flush()
 | 
					                uart.flush()
 | 
				
			||||||
                time.sleep_us(100)  # 短暂延时,确保数据完全发出
 | 
					                time.sleep_us(100)
 | 
				
			||||||
                rts_pin.value(0)  # 切换回接收模式 (DE/RE = LOW)
 | 
					                rts_pin.value(0)
 | 
				
			||||||
                log(f"总线 {bus_id} 原始命令发送成功: {command_bytes.hex()}")
 | 
					                log(f"总线 {bus_id} 原始命令发送成功: {command_bytes.hex()}")
 | 
				
			||||||
 | 
					
 | 
				
			||||||
                # III. 接收响应
 | 
					                # III. 接收响应
 | 
				
			||||||
                response_bytes = uart.read()
 | 
					                start_time = time.ticks_ms()
 | 
				
			||||||
                if response_bytes:
 | 
					                response_timeout = self.default_timeouts.get('rs485_response', 500)
 | 
				
			||||||
                    log(f"总线 {bus_id} 收到响应: {response_bytes.hex()}")
 | 
					                while time.ticks_diff(time.ticks_ms(), start_time) < response_timeout:
 | 
				
			||||||
 | 
					                    if uart.any():
 | 
				
			||||||
 | 
					                        chunk = uart.read(32)
 | 
				
			||||||
 | 
					                        if chunk:
 | 
				
			||||||
 | 
					                            response_buffer.extend(chunk)
 | 
				
			||||||
 | 
					                            start_time = time.ticks_ms()  # 收到数据就重置超时
 | 
				
			||||||
 | 
					                    time.sleep_ms(5)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					                if response_buffer:
 | 
				
			||||||
 | 
					                    # 动态地从请求命令中获取预期的从站ID和功能码
 | 
				
			||||||
 | 
					                    expected_slave_id = command_bytes[0]
 | 
				
			||||||
 | 
					                    expected_func_code = command_bytes[1]
 | 
				
			||||||
 | 
					                    
 | 
				
			||||||
 | 
					                    found_frame = self._find_modbus_frame(response_buffer, expected_slave_id, expected_func_code)
 | 
				
			||||||
 | 
					                    if found_frame:
 | 
				
			||||||
 | 
					                        log(f"总线 {bus_id} 收到有效响应: {found_frame.hex()}")
 | 
				
			||||||
 | 
					                        response_bytes = found_frame  # 将找到的帧赋值给外部变量
 | 
				
			||||||
                    else:
 | 
					                    else:
 | 
				
			||||||
                    log(f"警告: 总线 {bus_id} 未收到响应或响应超时。")
 | 
					                        log(f"警告: 总线 {bus_id} 响应中无有效帧。收到响应: {response_buffer.hex()}")
 | 
				
			||||||
                    return None
 | 
					                else:
 | 
				
			||||||
 | 
					                    log(f"警告: 总线 {bus_id} 未收到响应。")
 | 
				
			||||||
 | 
					
 | 
				
			||||||
            except Exception as e:
 | 
					            except Exception as e:
 | 
				
			||||||
                log(f"错误: 在总线 {bus_id} 上执行采集命令失败: {e}")
 | 
					                log(f"错误: 在总线 {bus_id} 上执行采集命令失败: {e}")
 | 
				
			||||||
 | 
					        
 | 
				
			||||||
 | 
					        # IV. 统一处理和解析
 | 
				
			||||||
 | 
					        # 无论是因为超时、未找到有效帧还是发生异常,只要 response_bytes 仍为 None,就任务失败
 | 
				
			||||||
 | 
					        if response_bytes is None:
 | 
				
			||||||
            return None
 | 
					            return None
 | 
				
			||||||
 | 
					
 | 
				
			||||||
        # IV. 响应解析与数据提取 (默认 Modbus RTU 浮点数)
 | 
					        # 使用找到的有效帧进行解析
 | 
				
			||||||
        # TODO: 根据CollectTask的Protobuf定义,此处需要根据parser_type来选择具体的解析逻辑和类型。
 | 
					        parsed_value = RS485Manager._parse_modbus_rtu_default(response_bytes)
 | 
				
			||||||
        #       目前默认使用Modbus RTU大端浮点数解析。
 | 
					 | 
				
			||||||
        parsed_value = self._parse_modbus_rtu_float_default(response_bytes)
 | 
					 | 
				
			||||||
 | 
					
 | 
				
			||||||
        # V. 返回结果
 | 
					 | 
				
			||||||
        return parsed_value
 | 
					        return parsed_value
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    def _find_modbus_frame(self, buffer: bytearray, expected_slave: int, func_code: int) -> bytes | None:
 | 
				
			||||||
 | 
					        """
 | 
				
			||||||
 | 
					        修复版:加调试;优先头检查;CRC 字节序标准 Modbus (低字节在前)。
 | 
				
			||||||
 | 
					        """
 | 
				
			||||||
 | 
					        log(f"搜索帧: buffer 长度 {len(buffer)}, hex {buffer.hex()}")
 | 
				
			||||||
 | 
					        i = 0
 | 
				
			||||||
 | 
					        while i < len(buffer) - 6:  # 最小 7 字节,-6 安全
 | 
				
			||||||
 | 
					            if buffer[i] == expected_slave and buffer[i + 1] == func_code:
 | 
				
			||||||
 | 
					                byte_count = buffer[i + 2]
 | 
				
			||||||
 | 
					                frame_len = 3 + byte_count + 2
 | 
				
			||||||
 | 
					                if len(buffer) - i >= frame_len:
 | 
				
			||||||
 | 
					                    frame = bytes(buffer[i:i + frame_len])
 | 
				
			||||||
 | 
					                    # CRC 预校验(标准 Modbus:CRC 低字节在前)
 | 
				
			||||||
 | 
					                    core = frame[:-2]
 | 
				
			||||||
 | 
					                    calc_crc = self._calculate_crc16_modbus(core)
 | 
				
			||||||
 | 
					                    low_crc = frame[-2]
 | 
				
			||||||
 | 
					                    high_crc = frame[-1]
 | 
				
			||||||
 | 
					                    recv_crc = (high_crc << 8) | low_crc  # 高<<8 | 低
 | 
				
			||||||
 | 
					                    log(f"候选帧 at {i}: {frame.hex()}, calc CRC {calc_crc:04X}, recv {recv_crc:04X}")
 | 
				
			||||||
 | 
					                    if calc_crc == recv_crc:
 | 
				
			||||||
 | 
					                        log(f"找到有效帧: {frame.hex()}")
 | 
				
			||||||
 | 
					                        return frame
 | 
				
			||||||
 | 
					                    else:
 | 
				
			||||||
 | 
					                        log(f"CRC 不匹配,跳过 (calc {calc_crc:04X} != recv {recv_crc:04X})")
 | 
				
			||||||
 | 
					            i += 1
 | 
				
			||||||
 | 
					        log("无有效帧")
 | 
				
			||||||
 | 
					        return None
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    @staticmethod
 | 
				
			||||||
 | 
					    def _parse_modbus_rtu_default(response_bytes):  # 改名,支持整数/浮点
 | 
				
			||||||
 | 
					        """
 | 
				
			||||||
 | 
					        修复版:动态数据长;CRC 只用核心。
 | 
				
			||||||
 | 
					        """
 | 
				
			||||||
 | 
					        if not response_bytes or len(response_bytes) < 7:
 | 
				
			||||||
 | 
					            log(f"警告: 响应过短。响应: {response_bytes.hex() if response_bytes else 'None'}")
 | 
				
			||||||
 | 
					            return None
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					        # CRC 校验(只核心)
 | 
				
			||||||
 | 
					        data_for_crc = response_bytes[:-2]
 | 
				
			||||||
 | 
					        received_crc = (response_bytes[-1] << 8) | response_bytes[-2]
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					        calculated_crc = RS485Manager._calculate_crc16_modbus(data_for_crc)
 | 
				
			||||||
 | 
					        if calculated_crc != received_crc:
 | 
				
			||||||
 | 
					            log(f"错误: CRC失败。接收: {received_crc:04X}, 计算: {calculated_crc:04X}. 响应: {response_bytes.hex()}")
 | 
				
			||||||
 | 
					            return None
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					        function_code = response_bytes[1]
 | 
				
			||||||
 | 
					        byte_count = response_bytes[2]
 | 
				
			||||||
 | 
					        data_bytes = response_bytes[3:3 + byte_count]
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					        if function_code not in [0x03, 0x04]:
 | 
				
			||||||
 | 
					            log(f"警告: 功能码 {function_code:02X} 不符。")
 | 
				
			||||||
 | 
					            return None
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					        if len(data_bytes) != byte_count:
 | 
				
			||||||
 | 
					            log(f"错误: 数据长 {len(data_bytes)} != {byte_count}")
 | 
				
			||||||
 | 
					            return None
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					        # 动态解析
 | 
				
			||||||
 | 
					        if byte_count == 2:
 | 
				
			||||||
 | 
					            # 整数 (e.g., 温度)
 | 
				
			||||||
 | 
					            try:
 | 
				
			||||||
 | 
					                value = int.from_bytes(data_bytes, 'big')  # 或 signed '>h'
 | 
				
			||||||
 | 
					                parsed_value = value
 | 
				
			||||||
 | 
					                log(f"成功解析整数: {parsed_value}")
 | 
				
			||||||
 | 
					                return parsed_value
 | 
				
			||||||
 | 
					            except Exception as e:
 | 
				
			||||||
 | 
					                log(f"整数解析失败: {e}")
 | 
				
			||||||
 | 
					                return None
 | 
				
			||||||
 | 
					        elif byte_count == 4:
 | 
				
			||||||
 | 
					            # 浮点
 | 
				
			||||||
 | 
					            try:
 | 
				
			||||||
 | 
					                parsed_value = struct.unpack('>f', data_bytes)[0]
 | 
				
			||||||
 | 
					                log(f"成功解析浮点: {parsed_value}")
 | 
				
			||||||
 | 
					                return parsed_value
 | 
				
			||||||
 | 
					            except Exception as e:
 | 
				
			||||||
 | 
					                log(f"浮点失败: {e}")
 | 
				
			||||||
 | 
					                return None
 | 
				
			||||||
 | 
					        else:
 | 
				
			||||||
 | 
					            log(f"警告: 未知字节数 {byte_count}")
 | 
				
			||||||
 | 
					            return None
 | 
				
			||||||
@@ -11,8 +11,8 @@
 | 
				
			|||||||
# --- LoRa 模块配置 ---
 | 
					# --- LoRa 模块配置 ---
 | 
				
			||||||
# 假设LoRa模块使用独立的UART进行通信
 | 
					# 假设LoRa模块使用独立的UART进行通信
 | 
				
			||||||
LORA_CONFIG = {
 | 
					LORA_CONFIG = {
 | 
				
			||||||
    # # 平台LoRa地址
 | 
					    # 平台LoRa地址
 | 
				
			||||||
    # 'master_address': 0x01, // 目前的LoRa模块主从模式下 从节点发送广播包=点对点发送给主节点, 所有从节点都会忽略其他从节点发的广播
 | 
					    'master_address': 0x01,
 | 
				
			||||||
 | 
					
 | 
				
			||||||
    # LoRa模块连接的UART总线ID (0, 1, or 2 on ESP32)
 | 
					    # LoRa模块连接的UART总线ID (0, 1, or 2 on ESP32)
 | 
				
			||||||
    'uart_id': 2,
 | 
					    'uart_id': 2,
 | 
				
			||||||
@@ -22,8 +22,8 @@ LORA_CONFIG = {
 | 
				
			|||||||
 | 
					
 | 
				
			||||||
    # LoRa模块连接的GPIO引脚
 | 
					    # LoRa模块连接的GPIO引脚
 | 
				
			||||||
    'pins': {
 | 
					    'pins': {
 | 
				
			||||||
        'tx': 17,  # UART TX
 | 
					        'tx': 5,  # UART TX
 | 
				
			||||||
        'rx': 16,  # UART RX
 | 
					        'rx': 4,  # UART RX
 | 
				
			||||||
    },
 | 
					    },
 | 
				
			||||||
 | 
					
 | 
				
			||||||
    # LoRa Mesh 模块发送模式(EC: 透传; ED: 完整数据包)
 | 
					    # LoRa Mesh 模块发送模式(EC: 透传; ED: 完整数据包)
 | 
				
			||||||
@@ -32,11 +32,14 @@ LORA_CONFIG = {
 | 
				
			|||||||
    #       e.g. 发送: EC 05 02 01 48 65 6c 6c 6f
 | 
					    #       e.g. 发送: EC 05 02 01 48 65 6c 6c 6f
 | 
				
			||||||
    #            (EC + 05(消息长度) + 0201(地址) + "Hello"(消息本体))
 | 
					    #            (EC + 05(消息长度) + 0201(地址) + "Hello"(消息本体))
 | 
				
			||||||
    #            接收: 48 65 6c 6c 6f ("Hello")
 | 
					    #            接收: 48 65 6c 6c 6f ("Hello")
 | 
				
			||||||
    #   ED: 接收端会接收完整数据包,包含请求头
 | 
					    #   ED: 接收端会接收完整数据包,包含自定义协议头和地址信息。
 | 
				
			||||||
    #       e.g. 发送: ED 05 02 01 48 65 6c 6c 6f
 | 
					    #       e.g. 发送: ED 05 12 34 01 00 01 02 03
 | 
				
			||||||
    #            (ED + 05(消息长度) + 0201(地址) + "Hello"(消息本体))
 | 
					    #            (ED(帧头) + 05(Length, 即 1(总包数)+1(当前包序号)+3(数据块)) + 12 34(目标地址) + 01(总包数) + 00(当前包序号) + 01 02 03(数据块))
 | 
				
			||||||
    #            接收: ED 05 02 01 48 65 6c 6c 6f
 | 
					    #            接收: ED 05 12 34 01 00 01 02 03 56 78(56 78 是发送方地址,会自动拼接到消息末尾)
 | 
				
			||||||
    'lora_mesh_mode': 'EC',
 | 
					    'lora_mesh_mode': 'ED',
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    # 单包最大用户数据数据长度, 模块限制240, 去掉两位自定义包头, 还剩238
 | 
				
			||||||
 | 
					    'max_chunk_size': 238
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
# --- 总线配置 ---
 | 
					# --- 总线配置 ---
 | 
				
			||||||
@@ -57,9 +60,9 @@ BUS_CONFIG = {
 | 
				
			|||||||
 | 
					
 | 
				
			||||||
        # 该总线使用的GPIO引脚
 | 
					        # 该总线使用的GPIO引脚
 | 
				
			||||||
        'pins': {
 | 
					        'pins': {
 | 
				
			||||||
            'tx': 4,  # RS485 TX
 | 
					            'tx': 16,  # RS485 TX
 | 
				
			||||||
            'rx': 5,  # RS485 RX
 | 
					            'rx': 17,  # RS485 RX
 | 
				
			||||||
            'rts': 2,  # RS485 DE/RE 方向控制引脚
 | 
					            'rts': 15,  # RS485 DE/RE 方向控制引脚
 | 
				
			||||||
        }
 | 
					        }
 | 
				
			||||||
    },
 | 
					    },
 | 
				
			||||||
 | 
					
 | 
				
			||||||
@@ -4,13 +4,17 @@
 | 
				
			|||||||
"""
 | 
					"""
 | 
				
			||||||
一个简单的、可配置的日志记录器模块。
 | 
					一个简单的、可配置的日志记录器模块。
 | 
				
			||||||
"""
 | 
					"""
 | 
				
			||||||
 | 
					import _thread
 | 
				
			||||||
 | 
					from app.config.config import *
 | 
				
			||||||
 | 
					
 | 
				
			||||||
from main.config.config import *
 | 
					# 创建一个锁,用于在多线程环境中同步对print的调用
 | 
				
			||||||
 | 
					log_lock = _thread.allocate_lock()
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
def log(message: str):
 | 
					def log(message: str):
 | 
				
			||||||
    """
 | 
					    """
 | 
				
			||||||
    打印一条日志消息,是否实际输出取决于配置文件。
 | 
					    打印一条日志消息,是否实际输出取决于配置文件。
 | 
				
			||||||
 | 
					    使用锁来确保多线程环境下的输出不会混乱。
 | 
				
			||||||
 | 
					
 | 
				
			||||||
    Args:
 | 
					    Args:
 | 
				
			||||||
        message (str): 要打印的日志消息。
 | 
					        message (str): 要打印的日志消息。
 | 
				
			||||||
@@ -18,6 +22,7 @@ def log(message: str):
 | 
				
			|||||||
    # 从配置文件中获取调试开关的状态
 | 
					    # 从配置文件中获取调试开关的状态
 | 
				
			||||||
    # .get()方法可以安全地获取值,如果键不存在,则返回默认值False
 | 
					    # .get()方法可以安全地获取值,如果键不存在,则返回默认值False
 | 
				
			||||||
    if SYSTEM_PARAMS.get('debug_enabled', False):
 | 
					    if SYSTEM_PARAMS.get('debug_enabled', False):
 | 
				
			||||||
 | 
					        with log_lock:
 | 
				
			||||||
            print(message)
 | 
					            print(message)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
    # 如果开关为False,此函数会立即返回,不执行任何操作。
 | 
					    # 如果开关为False,此函数会立即返回,不执行任何操作。
 | 
				
			||||||
							
								
								
									
										203
									
								
								app/lora/lora_mesh_uart_passthrough_manager.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										203
									
								
								app/lora/lora_mesh_uart_passthrough_manager.py
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,203 @@
 | 
				
			|||||||
 | 
					#!/usr/bin/env python
 | 
				
			||||||
 | 
					# -*- coding: utf-8 -*-
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					"""
 | 
				
			||||||
 | 
					LoRa模块的具体实现 (UART Passthrough for LoRa Mesh)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					负责与LoRa模块进行底层通信,并向上层提供标准化的数据包收发接口。
 | 
				
			||||||
 | 
					这个实现针对的是通过UART进行透传的LoRa Mesh模块。
 | 
				
			||||||
 | 
					"""
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					from ..logs.logger import log
 | 
				
			||||||
 | 
					from machine import UART
 | 
				
			||||||
 | 
					import time
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					class LoRaMeshUartPassthroughManager:
 | 
				
			||||||
 | 
					    """
 | 
				
			||||||
 | 
					    通过UART与LoRa Mesh模块通信的处理器实现 (ED模式)。
 | 
				
			||||||
 | 
					    实现了自动分片与重组逻辑。
 | 
				
			||||||
 | 
					    """
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    def __init__(self, lora_config: dict):
 | 
				
			||||||
 | 
					        """
 | 
				
			||||||
 | 
					        初始化LoRa处理器。
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					        Args:
 | 
				
			||||||
 | 
					            lora_config (dict): 来自全局配置文件的LoRa配置字典。
 | 
				
			||||||
 | 
					        """
 | 
				
			||||||
 | 
					        log("LoRaMeshUartPassthroughManager: 初始化...")
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					        # --- 配置注入 ---
 | 
				
			||||||
 | 
					        self.master_address = lora_config.get('master_address')
 | 
				
			||||||
 | 
					        self.uart_id = lora_config.get('uart_id')
 | 
				
			||||||
 | 
					        self.baudrate = lora_config.get('baudrate')
 | 
				
			||||||
 | 
					        self.pins = lora_config.get('pins')
 | 
				
			||||||
 | 
					        self.max_chunk_size = lora_config.get('max_chunk_size')
 | 
				
			||||||
 | 
					        self.lora_mesh_mode = b'\xed'
 | 
				
			||||||
 | 
					        # TODO 目前这个配置没用, 完全按ED处理的
 | 
				
			||||||
 | 
					        if lora_config.get('lora_mesh_mode') == 'EC':
 | 
				
			||||||
 | 
					            self.lora_mesh_mode = b'\xec'
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					        # --- 硬件初始化 ---
 | 
				
			||||||
 | 
					        self.uart = UART(self.uart_id, self.baudrate, tx=self.pins['tx'], rx=self.pins['rx'])
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					        # --- 内部状态变量 ---
 | 
				
			||||||
 | 
					        self._rx_buffer = bytearray()  # UART接收缓冲区
 | 
				
			||||||
 | 
					        self._reassembly_cache = {}  # 分片重组缓冲区 { chunk_index: chunk_data }
 | 
				
			||||||
 | 
					        self._expected_chunks = 0  # 当前会话期望的总分片数
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					        log(f"LoRaMeshUartPassthroughManager: 配置加载完成. UART ID: {self.uart_id}, Baudrate: {self.baudrate}, 针脚: {self.pins}")
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    def send_packet(self, payload: bytes) -> bool:
 | 
				
			||||||
 | 
					        """
 | 
				
			||||||
 | 
					        【实现】发送一个数据包,自动处理分片。
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					        Args:
 | 
				
			||||||
 | 
					            payload (bytes): 需要发送的完整业务数据。
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					        Returns:
 | 
				
			||||||
 | 
					            bool: True表示所有分片都已成功提交发送,False表示失败。
 | 
				
			||||||
 | 
					        """
 | 
				
			||||||
 | 
					        max_chunk_size = self.max_chunk_size
 | 
				
			||||||
 | 
					        if not payload:
 | 
				
			||||||
 | 
					            total_chunks = 1
 | 
				
			||||||
 | 
					        else:
 | 
				
			||||||
 | 
					            total_chunks = (len(payload) + max_chunk_size - 1) // max_chunk_size
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					        try:
 | 
				
			||||||
 | 
					            for i in range(total_chunks):
 | 
				
			||||||
 | 
					                chunk_index = i
 | 
				
			||||||
 | 
					                start = i * max_chunk_size
 | 
				
			||||||
 | 
					                end = start + max_chunk_size
 | 
				
			||||||
 | 
					                chunk_data = payload[start:end]
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					                # --- 组装物理包 ---
 | 
				
			||||||
 | 
					                header = b'\xed'
 | 
				
			||||||
 | 
					                dest_addr_bytes = self.master_address.to_bytes(2, 'big')
 | 
				
			||||||
 | 
					                total_chunks_bytes = total_chunks.to_bytes(1, 'big')
 | 
				
			||||||
 | 
					                current_chunk_bytes = chunk_index.to_bytes(1, 'big')
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					                # 计算后续长度(总包数和当前包序号是自定义包头, 各占一位, 标准包头算在长度内)
 | 
				
			||||||
 | 
					                length_val = 2 + len(chunk_data)
 | 
				
			||||||
 | 
					                length_bytes = length_val.to_bytes(1, 'big')
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					                # 拼接成最终的数据包
 | 
				
			||||||
 | 
					                packet_to_send = header + length_bytes + dest_addr_bytes + total_chunks_bytes + current_chunk_bytes + chunk_data
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					                self.uart.write(packet_to_send)
 | 
				
			||||||
 | 
					                log(f"LoRa: 发送分片 {chunk_index + 1}/{total_chunks} 到地址 {self.master_address}")
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					                # 让出CPU, 模块将缓存区的数据发出去本身也需要时间
 | 
				
			||||||
 | 
					                time.sleep_ms(10)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					            return True
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					        except Exception as e:
 | 
				
			||||||
 | 
					            log(f"LoRa: 发送数据包失败: {e}")
 | 
				
			||||||
 | 
					            return False
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    def receive_packet(self) -> bytes | None:
 | 
				
			||||||
 | 
					        """
 | 
				
			||||||
 | 
					        【实现】非阻塞地检查、解析并重组一个完整的数据包。
 | 
				
			||||||
 | 
					        """
 | 
				
			||||||
 | 
					        # 1. 从硬件读取数据到缓冲区
 | 
				
			||||||
 | 
					        if self.uart.any():
 | 
				
			||||||
 | 
					            new_data = self.uart.read()
 | 
				
			||||||
 | 
					            if new_data:
 | 
				
			||||||
 | 
					                log(f"LoRa: UART收到原始数据 (长度 {len(new_data)}): {new_data.hex()}")
 | 
				
			||||||
 | 
					                self._rx_buffer.extend(new_data)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					        # 如果缓冲区为空,没有必要继续处理
 | 
				
			||||||
 | 
					        if not self._rx_buffer:
 | 
				
			||||||
 | 
					            return None
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					        # 2. 只要缓冲区有数据就持续尝试从缓冲区解析包
 | 
				
			||||||
 | 
					        while len(self._rx_buffer) > 0:
 | 
				
			||||||
 | 
					            log(f"LoRa: --- 开始新一轮解析, 缓冲区 (长度 {len(self._rx_buffer)}): {self._rx_buffer.hex()} ---")
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					            # 2.1 检查头部和长度字段是否存在
 | 
				
			||||||
 | 
					            if len(self._rx_buffer) < 2:
 | 
				
			||||||
 | 
					                log("LoRa: 缓冲区数据不足 (小于2字节),无法读取包头。等待更多数据...")
 | 
				
			||||||
 | 
					                return None  # 数据不足,无法读取长度
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					            # 2.2 检查帧头是否正确
 | 
				
			||||||
 | 
					            if self._rx_buffer[0] != 0xED:
 | 
				
			||||||
 | 
					                log(f"LoRa: 接收到错误帧头: {hex(self._rx_buffer[0])},正在寻找下一个ED...")
 | 
				
			||||||
 | 
					                next_ed = self._rx_buffer.find(b'\xed', 1)
 | 
				
			||||||
 | 
					                if next_ed == -1:
 | 
				
			||||||
 | 
					                    log("LoRa: 缓冲区无有效帧头,已清空。")
 | 
				
			||||||
 | 
					                    self._rx_buffer[:] = b''
 | 
				
			||||||
 | 
					                    return None  # 清空后没有数据了, 直接返回
 | 
				
			||||||
 | 
					                else:
 | 
				
			||||||
 | 
					                    log(f"LoRa: 在位置 {next_ed} 找到下一个有效帧头,丢弃之前的数据。")
 | 
				
			||||||
 | 
					                    self._rx_buffer = self._rx_buffer[next_ed:]
 | 
				
			||||||
 | 
					                    continue  # 继续循环,用新的缓冲区数据重新开始解析
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					            # 2.3 检查包是否完整
 | 
				
			||||||
 | 
					            payload_len = self._rx_buffer[1]
 | 
				
			||||||
 | 
					            # 物理层在末尾又加了2字节的源地址,所以完整包长需要+2。
 | 
				
			||||||
 | 
					            total_packet_len = 1 + 1 + payload_len + 2
 | 
				
			||||||
 | 
					            log(f"LoRa: 帧头正确(ED)。声明的后续包长(payload_len): {payload_len}。计算出的总包长: {total_packet_len}。")
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					            if len(self._rx_buffer) < total_packet_len:
 | 
				
			||||||
 | 
					                log(f"LoRa: '半包'情况,需要 {total_packet_len} 字节,但缓冲区只有 {len(self._rx_buffer)} 字节。等待更多数据...")
 | 
				
			||||||
 | 
					                return None  # "半包"情况,等待更多数据
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					            # 3. 提取和解析一个完整的物理包
 | 
				
			||||||
 | 
					            log(f"LoRa: 发现完整物理包 (长度 {total_packet_len}),正在提取...")
 | 
				
			||||||
 | 
					            packet = self._rx_buffer[:total_packet_len]
 | 
				
			||||||
 | 
					            self._rx_buffer = self._rx_buffer[total_packet_len:]
 | 
				
			||||||
 | 
					            log(f"LoRa: 提取的包: {packet.hex()}。剩余缓冲区 (长度 {len(self._rx_buffer)}): {self._rx_buffer.hex()}")
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					            # --- 包结构解析 ---
 | 
				
			||||||
 | 
					            if len(packet) < 8:
 | 
				
			||||||
 | 
					                log(f"LoRa: 包长度 {len(packet)} 小于协议最小长度8, 判定为坏包,已丢弃。")
 | 
				
			||||||
 | 
					                continue
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					            addr = int.from_bytes(packet[2:4], 'big')
 | 
				
			||||||
 | 
					            total_chunks = packet[4]
 | 
				
			||||||
 | 
					            current_chunk = packet[5]
 | 
				
			||||||
 | 
					            # 提取数据块
 | 
				
			||||||
 | 
					            chunk_data = packet[6:]
 | 
				
			||||||
 | 
					            source_addr = int.from_bytes(packet[-2:], 'big')
 | 
				
			||||||
 | 
					            log(f"LoRa: 解析包: 源地址={source_addr}, 目标地址={addr}, 总分片={total_chunks}, 当前分片={current_chunk}, 数据块长度={len(chunk_data)}")
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					            # 4. 重组逻辑
 | 
				
			||||||
 | 
					            if total_chunks == 1:
 | 
				
			||||||
 | 
					                log(f"LoRa: 收到单包消息,来自地址 {source_addr},长度 {len(chunk_data)}")
 | 
				
			||||||
 | 
					                self._reassembly_cache.clear()
 | 
				
			||||||
 | 
					                self._expected_chunks = 0
 | 
				
			||||||
 | 
					                return chunk_data
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					            # 对于多包消息,只有当收到第一个分片时才清空缓存并设置期望分片数
 | 
				
			||||||
 | 
					            if current_chunk == 0:
 | 
				
			||||||
 | 
					                log(f"LoRa: 开始接收新的多包会话 ({total_chunks}个分片) from {source_addr}...")
 | 
				
			||||||
 | 
					                self._reassembly_cache.clear()
 | 
				
			||||||
 | 
					                self._expected_chunks = total_chunks
 | 
				
			||||||
 | 
					            elif not self._reassembly_cache and self._expected_chunks == 0:
 | 
				
			||||||
 | 
					                # 如果不是第一个分片,但缓存是空的,说明错过了第一个分片,丢弃当前分片
 | 
				
			||||||
 | 
					                log(f"LoRa: 收到非首个分片 {current_chunk} from {source_addr},但未检测到会话开始,已丢弃。")
 | 
				
			||||||
 | 
					                continue
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					            self._reassembly_cache[current_chunk] = chunk_data
 | 
				
			||||||
 | 
					            log(f"LoRa: 收到分片 {current_chunk + 1}/{self._expected_chunks} from {source_addr},已缓存 {len(self._reassembly_cache)} 个")
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					            if len(self._reassembly_cache) == self._expected_chunks:
 | 
				
			||||||
 | 
					                log(f"LoRa: 所有分片已集齐 (from {source_addr}),正在重组...")
 | 
				
			||||||
 | 
					                full_payload = bytearray()
 | 
				
			||||||
 | 
					                for i in range(self._expected_chunks):
 | 
				
			||||||
 | 
					                    if i not in self._reassembly_cache:
 | 
				
			||||||
 | 
					                        log(f"LoRa: 重组失败!缺少分片 {i}。")
 | 
				
			||||||
 | 
					                        self._reassembly_cache.clear()
 | 
				
			||||||
 | 
					                        self._expected_chunks = 0
 | 
				
			||||||
 | 
					                        return None
 | 
				
			||||||
 | 
					                    full_payload.extend(self._reassembly_cache[i])
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					                log(f"LoRa: 重组完成,总长度 {len(full_payload)}")
 | 
				
			||||||
 | 
					                self._reassembly_cache.clear()
 | 
				
			||||||
 | 
					                self._expected_chunks = 0
 | 
				
			||||||
 | 
					                return bytes(full_payload)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					        # while 循环结束,意味着缓冲区被处理完毕但没有返回一个完整的包
 | 
				
			||||||
 | 
					        return None
 | 
				
			||||||
@@ -8,15 +8,12 @@
 | 
				
			|||||||
- 编排业务流程:解码指令,并将业务任务分发给相应的管理器。
 | 
					- 编排业务流程:解码指令,并将业务任务分发给相应的管理器。
 | 
				
			||||||
- 完全不关心总线通信和数据解析的技术实现细节。
 | 
					- 完全不关心总线通信和数据解析的技术实现细节。
 | 
				
			||||||
"""
 | 
					"""
 | 
				
			||||||
 | 
					from app.lora.lora_mesh_uart_passthrough_manager import LoRaMeshUartPassthroughManager
 | 
				
			||||||
# 导入我们定义的“契约”(接口)
 | 
					from app.bus.rs485_manager import RS485Manager
 | 
				
			||||||
from lora.lora_interface import ILoraManager
 | 
					 | 
				
			||||||
from bus.bus_interface import IBusManager
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
# 导入Protobuf解析代码
 | 
					# 导入Protobuf解析代码
 | 
				
			||||||
from proto import client_pb
 | 
					from app.proto import client_pb
 | 
				
			||||||
 | 
					
 | 
				
			||||||
from logs.logger import log
 | 
					from app.logs.logger import log
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
class Processor:
 | 
					class Processor:
 | 
				
			||||||
@@ -25,7 +22,7 @@ class Processor:
 | 
				
			|||||||
    它依赖于抽象的、面向业务的接口。
 | 
					    它依赖于抽象的、面向业务的接口。
 | 
				
			||||||
    """
 | 
					    """
 | 
				
			||||||
 | 
					
 | 
				
			||||||
    def __init__(self, lora_handler: ILoraManager, bus_manager: IBusManager):
 | 
					    def __init__(self, lora_handler: LoRaMeshUartPassthroughManager, bus_manager: RS485Manager):
 | 
				
			||||||
        """
 | 
					        """
 | 
				
			||||||
        构造函数 (依赖注入)。
 | 
					        构造函数 (依赖注入)。
 | 
				
			||||||
        
 | 
					        
 | 
				
			||||||
@@ -45,16 +42,25 @@ class Processor:
 | 
				
			|||||||
 | 
					
 | 
				
			||||||
        try:
 | 
					        try:
 | 
				
			||||||
            instruction = client_pb.decode_instruction(packet_bytes)
 | 
					            instruction = client_pb.decode_instruction(packet_bytes)
 | 
				
			||||||
 | 
					            log(f"解析指令成功: {instruction}")
 | 
				
			||||||
        except Exception as e:
 | 
					        except Exception as e:
 | 
				
			||||||
            log(f"错误:解码指令失败: {e}")
 | 
					            log(f"错误:解码指令失败: {e}")
 | 
				
			||||||
            return
 | 
					            return
 | 
				
			||||||
 | 
					
 | 
				
			||||||
        # 根据指令类型,分发到不同的业务处理方法
 | 
					        # 根据指令类型,分发到不同的业务处理方法
 | 
				
			||||||
        if 'raw_485_command' in instruction:
 | 
					        if 'raw_485_command' in instruction:
 | 
				
			||||||
            self._process_exec_command(instruction['raw_485_command'])
 | 
					            cmd = instruction['raw_485_command']
 | 
				
			||||||
 | 
					            if cmd:
 | 
				
			||||||
 | 
					                self._process_exec_command(cmd)
 | 
				
			||||||
 | 
					            else:
 | 
				
			||||||
 | 
					                log("警告:'raw_485_command' 指令内容为空。")
 | 
				
			||||||
 | 
					
 | 
				
			||||||
        elif 'batch_collect_command' in instruction:
 | 
					        elif 'batch_collect_command' in instruction:
 | 
				
			||||||
            self._process_collect_command(instruction['batch_collect_command'])
 | 
					            cmd = instruction['batch_collect_command']
 | 
				
			||||||
 | 
					            if cmd:
 | 
				
			||||||
 | 
					                self._process_collect_command(cmd)
 | 
				
			||||||
 | 
					            else:
 | 
				
			||||||
 | 
					                log("警告:'batch_collect_command' 指令内容为空。")
 | 
				
			||||||
 | 
					
 | 
				
			||||||
        else:
 | 
					        else:
 | 
				
			||||||
            log(f"警告:收到未知或不适用于此设备的指令类型: {instruction}")
 | 
					            log(f"警告:收到未知或不适用于此设备的指令类型: {instruction}")
 | 
				
			||||||
							
								
								
									
										305
									
								
								app/proto/client_pb.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										305
									
								
								app/proto/client_pb.py
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,305 @@
 | 
				
			|||||||
 | 
					#!/usr/bin/env python
 | 
				
			||||||
 | 
					# -*- coding: utf-8 -*-
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					"""
 | 
				
			||||||
 | 
					根据client.proto生成的解析代码 (V2 - 已修复解码逻辑)
 | 
				
			||||||
 | 
					适用于ESP32 MicroPython环境
 | 
				
			||||||
 | 
					"""
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					import struct
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					# --- Protobuf基础类型辅助函数 ---
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					def encode_varint(value):
 | 
				
			||||||
 | 
					    buf = bytearray()
 | 
				
			||||||
 | 
					    while value >= 0x80:
 | 
				
			||||||
 | 
					        buf.append((value & 0x7F) | 0x80)
 | 
				
			||||||
 | 
					        value >>= 7
 | 
				
			||||||
 | 
					    buf.append(value & 0x7F)
 | 
				
			||||||
 | 
					    return buf
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					def decode_varint(buf, pos=0):
 | 
				
			||||||
 | 
					    result = 0
 | 
				
			||||||
 | 
					    shift = 0
 | 
				
			||||||
 | 
					    while pos < len(buf):
 | 
				
			||||||
 | 
					        byte = buf[pos]
 | 
				
			||||||
 | 
					        pos += 1
 | 
				
			||||||
 | 
					        result |= (byte & 0x7F) << shift
 | 
				
			||||||
 | 
					        if not (byte & 0x80):
 | 
				
			||||||
 | 
					            break
 | 
				
			||||||
 | 
					        shift += 7
 | 
				
			||||||
 | 
					    return result, pos
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					def encode_string(value):
 | 
				
			||||||
 | 
					    value_bytes = value.encode('utf-8')
 | 
				
			||||||
 | 
					    length = encode_varint(len(value_bytes))
 | 
				
			||||||
 | 
					    return length + value_bytes
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					def decode_string(buf, pos=0):
 | 
				
			||||||
 | 
					    """解码字符串 (已修复)"""
 | 
				
			||||||
 | 
					    length, pos_after_len = decode_varint(buf, pos)
 | 
				
			||||||
 | 
					    end_pos = pos_after_len + length
 | 
				
			||||||
 | 
					    value = buf[pos_after_len:end_pos].decode('utf-8')
 | 
				
			||||||
 | 
					    return value, end_pos
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					# --- 消息编码/解码函数 ---
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					def encode_raw_485_command(bus_number, command_bytes):
 | 
				
			||||||
 | 
					    result = bytearray()
 | 
				
			||||||
 | 
					    result.extend(encode_varint((1 << 3) | 0))
 | 
				
			||||||
 | 
					    result.extend(encode_varint(bus_number))
 | 
				
			||||||
 | 
					    result.extend(encode_varint((2 << 3) | 2))
 | 
				
			||||||
 | 
					    result.extend(encode_varint(len(command_bytes)))
 | 
				
			||||||
 | 
					    result.extend(command_bytes)
 | 
				
			||||||
 | 
					    return result
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					def decode_raw_485_command(buf):
 | 
				
			||||||
 | 
					    """解码Raw485Command消息 (已修复)"""
 | 
				
			||||||
 | 
					    result = {}
 | 
				
			||||||
 | 
					    pos = 0
 | 
				
			||||||
 | 
					    while pos < len(buf):
 | 
				
			||||||
 | 
					        tag, pos = decode_varint(buf, pos)
 | 
				
			||||||
 | 
					        field_number = tag >> 3
 | 
				
			||||||
 | 
					        wire_type = tag & 0x07
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					        if field_number == 1:  # bus_number
 | 
				
			||||||
 | 
					            value, pos = decode_varint(buf, pos)
 | 
				
			||||||
 | 
					            result['bus_number'] = value
 | 
				
			||||||
 | 
					        elif field_number == 2:  # command_bytes
 | 
				
			||||||
 | 
					            length, pos_after_len = decode_varint(buf, pos)
 | 
				
			||||||
 | 
					            end_pos = pos_after_len + length
 | 
				
			||||||
 | 
					            result['command_bytes'] = buf[pos_after_len:end_pos]
 | 
				
			||||||
 | 
					            pos = end_pos
 | 
				
			||||||
 | 
					        else:
 | 
				
			||||||
 | 
					            # 跳过未知字段
 | 
				
			||||||
 | 
					            if wire_type == 0:
 | 
				
			||||||
 | 
					                _, pos = decode_varint(buf, pos)
 | 
				
			||||||
 | 
					            elif wire_type == 2:
 | 
				
			||||||
 | 
					                length, pos_after_len = decode_varint(buf, pos)
 | 
				
			||||||
 | 
					                pos = pos_after_len + length
 | 
				
			||||||
 | 
					            elif wire_type == 5:
 | 
				
			||||||
 | 
					                pos += 4
 | 
				
			||||||
 | 
					            else:
 | 
				
			||||||
 | 
					                pos += 1
 | 
				
			||||||
 | 
					    return result
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					def encode_collect_task(command_msg):
 | 
				
			||||||
 | 
					    result = bytearray()
 | 
				
			||||||
 | 
					    encoded_command = encode_raw_485_command(command_msg['bus_number'], command_msg['command_bytes'])
 | 
				
			||||||
 | 
					    result.extend(encode_varint((1 << 3) | 2))
 | 
				
			||||||
 | 
					    result.extend(encode_varint(len(encoded_command)))
 | 
				
			||||||
 | 
					    result.extend(encoded_command)
 | 
				
			||||||
 | 
					    return result
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					def decode_collect_task(buf):
 | 
				
			||||||
 | 
					    """解码CollectTask消息 (已修复)"""
 | 
				
			||||||
 | 
					    result = {}
 | 
				
			||||||
 | 
					    pos = 0
 | 
				
			||||||
 | 
					    while pos < len(buf):
 | 
				
			||||||
 | 
					        tag, pos = decode_varint(buf, pos)
 | 
				
			||||||
 | 
					        field_number = tag >> 3
 | 
				
			||||||
 | 
					        wire_type = tag & 0x07
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					        if field_number == 1:  # command
 | 
				
			||||||
 | 
					            length, pos_after_len = decode_varint(buf, pos)
 | 
				
			||||||
 | 
					            end_pos = pos_after_len + length
 | 
				
			||||||
 | 
					            value_buf = buf[pos_after_len:end_pos]
 | 
				
			||||||
 | 
					            result['command'] = decode_raw_485_command(value_buf)
 | 
				
			||||||
 | 
					            pos = end_pos
 | 
				
			||||||
 | 
					        else:
 | 
				
			||||||
 | 
					            if wire_type == 0:
 | 
				
			||||||
 | 
					                _, pos = decode_varint(buf, pos)
 | 
				
			||||||
 | 
					            elif wire_type == 2:
 | 
				
			||||||
 | 
					                length, pos_after_len = decode_varint(buf, pos)
 | 
				
			||||||
 | 
					                pos = pos_after_len + length
 | 
				
			||||||
 | 
					            elif wire_type == 5:
 | 
				
			||||||
 | 
					                pos += 4
 | 
				
			||||||
 | 
					            else:
 | 
				
			||||||
 | 
					                pos += 1
 | 
				
			||||||
 | 
					    return result
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					def encode_batch_collect_command(correlation_id, tasks):
 | 
				
			||||||
 | 
					    result = bytearray()
 | 
				
			||||||
 | 
					    result.extend(encode_varint((1 << 3) | 2))
 | 
				
			||||||
 | 
					    result.extend(encode_string(correlation_id))
 | 
				
			||||||
 | 
					    for task in tasks:
 | 
				
			||||||
 | 
					        encoded_task = encode_collect_task(task['command'])
 | 
				
			||||||
 | 
					        result.extend(encode_varint((2 << 3) | 2))
 | 
				
			||||||
 | 
					        result.extend(encode_varint(len(encoded_task)))
 | 
				
			||||||
 | 
					        result.extend(encoded_task)
 | 
				
			||||||
 | 
					    return result
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					def decode_batch_collect_command(buf):
 | 
				
			||||||
 | 
					    """解码BatchCollectCommand消息 (已修复)"""
 | 
				
			||||||
 | 
					    result = {'tasks': []}
 | 
				
			||||||
 | 
					    pos = 0
 | 
				
			||||||
 | 
					    while pos < len(buf):
 | 
				
			||||||
 | 
					        tag, pos = decode_varint(buf, pos)
 | 
				
			||||||
 | 
					        field_number = tag >> 3
 | 
				
			||||||
 | 
					        wire_type = tag & 0x07
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					        if field_number == 1:  # correlation_id
 | 
				
			||||||
 | 
					            value, pos = decode_string(buf, pos)
 | 
				
			||||||
 | 
					            result['correlation_id'] = value
 | 
				
			||||||
 | 
					        elif field_number == 2:  # tasks (repeated)
 | 
				
			||||||
 | 
					            length, pos_after_len = decode_varint(buf, pos)
 | 
				
			||||||
 | 
					            end_pos = pos_after_len + length
 | 
				
			||||||
 | 
					            value_buf = buf[pos_after_len:end_pos]
 | 
				
			||||||
 | 
					            result['tasks'].append(decode_collect_task(value_buf))
 | 
				
			||||||
 | 
					            pos = end_pos
 | 
				
			||||||
 | 
					        else:
 | 
				
			||||||
 | 
					            if wire_type == 0:
 | 
				
			||||||
 | 
					                _, pos = decode_varint(buf, pos)
 | 
				
			||||||
 | 
					            elif wire_type == 2:
 | 
				
			||||||
 | 
					                length, pos_after_len = decode_varint(buf, pos)
 | 
				
			||||||
 | 
					                pos = pos_after_len + length
 | 
				
			||||||
 | 
					            elif wire_type == 5:
 | 
				
			||||||
 | 
					                pos += 4
 | 
				
			||||||
 | 
					            else:
 | 
				
			||||||
 | 
					                pos += 1
 | 
				
			||||||
 | 
					    return result
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					def encode_collect_result(correlation_id, values):
 | 
				
			||||||
 | 
					    result = bytearray()
 | 
				
			||||||
 | 
					    result.extend(encode_varint((1 << 3) | 2))
 | 
				
			||||||
 | 
					    result.extend(encode_string(correlation_id))
 | 
				
			||||||
 | 
					    for value in values:
 | 
				
			||||||
 | 
					        result.extend(encode_varint((2 << 3) | 5))
 | 
				
			||||||
 | 
					        result.extend(struct.pack('<f', value))
 | 
				
			||||||
 | 
					    return result
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					def decode_collect_result(buf):
 | 
				
			||||||
 | 
					    """解码CollectResult消息 (已修复)"""
 | 
				
			||||||
 | 
					    result = {'values': []}
 | 
				
			||||||
 | 
					    pos = 0
 | 
				
			||||||
 | 
					    while pos < len(buf):
 | 
				
			||||||
 | 
					        tag, pos = decode_varint(buf, pos)
 | 
				
			||||||
 | 
					        field_number = tag >> 3
 | 
				
			||||||
 | 
					        wire_type = tag & 0x07
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					        if field_number == 1:  # correlation_id
 | 
				
			||||||
 | 
					            value, pos = decode_string(buf, pos)
 | 
				
			||||||
 | 
					            result['correlation_id'] = value
 | 
				
			||||||
 | 
					        elif field_number == 2:  # values (repeated)
 | 
				
			||||||
 | 
					            if wire_type == 5:  # fixed32
 | 
				
			||||||
 | 
					                value = struct.unpack('<f', buf[pos:pos + 4])[0]
 | 
				
			||||||
 | 
					                pos += 4
 | 
				
			||||||
 | 
					                result['values'].append(value)
 | 
				
			||||||
 | 
					        else:
 | 
				
			||||||
 | 
					            if wire_type == 0:
 | 
				
			||||||
 | 
					                _, pos = decode_varint(buf, pos)
 | 
				
			||||||
 | 
					            elif wire_type == 2:
 | 
				
			||||||
 | 
					                length, pos_after_len = decode_varint(buf, pos)
 | 
				
			||||||
 | 
					                pos = pos_after_len + length
 | 
				
			||||||
 | 
					            elif wire_type == 5:
 | 
				
			||||||
 | 
					                pos += 4
 | 
				
			||||||
 | 
					            else:
 | 
				
			||||||
 | 
					                pos += 1
 | 
				
			||||||
 | 
					    return result
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					def encode_instruction(payload_type, payload_data):
 | 
				
			||||||
 | 
					    result = bytearray()
 | 
				
			||||||
 | 
					    encoded_payload = bytearray()
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    if payload_type == 'raw_485_command':
 | 
				
			||||||
 | 
					        encoded_payload = encode_raw_485_command(payload_data['bus_number'], payload_data['command_bytes'])
 | 
				
			||||||
 | 
					        result.extend(encode_varint((1 << 3) | 2))
 | 
				
			||||||
 | 
					    elif payload_type == 'batch_collect_command':
 | 
				
			||||||
 | 
					        encoded_payload = encode_batch_collect_command(payload_data['correlation_id'], payload_data['tasks'])
 | 
				
			||||||
 | 
					        result.extend(encode_varint((2 << 3) | 2))
 | 
				
			||||||
 | 
					    elif payload_type == 'collect_result':
 | 
				
			||||||
 | 
					        encoded_payload = encode_collect_result(payload_data['correlation_id'], payload_data['values'])
 | 
				
			||||||
 | 
					        result.extend(encode_varint((3 << 3) | 2))
 | 
				
			||||||
 | 
					    else:
 | 
				
			||||||
 | 
					        raise ValueError("未知的指令负载类型")
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    result.extend(encode_varint(len(encoded_payload)))
 | 
				
			||||||
 | 
					    result.extend(encoded_payload)
 | 
				
			||||||
 | 
					    return result
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					def decode_instruction(buf):
 | 
				
			||||||
 | 
					    """解码Instruction消息 (已修复)"""
 | 
				
			||||||
 | 
					    result = {}
 | 
				
			||||||
 | 
					    pos = 0
 | 
				
			||||||
 | 
					    while pos < len(buf):
 | 
				
			||||||
 | 
					        tag, pos = decode_varint(buf, pos)
 | 
				
			||||||
 | 
					        field_number = tag >> 3
 | 
				
			||||||
 | 
					        wire_type = tag & 0x07
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					        if wire_type == 2:
 | 
				
			||||||
 | 
					            length, pos_after_len = decode_varint(buf, pos)
 | 
				
			||||||
 | 
					            end_pos = pos_after_len + length
 | 
				
			||||||
 | 
					            value_buf = buf[pos_after_len:end_pos]
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					            if field_number == 1:
 | 
				
			||||||
 | 
					                result['raw_485_command'] = decode_raw_485_command(value_buf)
 | 
				
			||||||
 | 
					            elif field_number == 2:
 | 
				
			||||||
 | 
					                result['batch_collect_command'] = decode_batch_collect_command(value_buf)
 | 
				
			||||||
 | 
					            elif field_number == 3:
 | 
				
			||||||
 | 
					                result['collect_result'] = decode_collect_result(value_buf)
 | 
				
			||||||
 | 
					            
 | 
				
			||||||
 | 
					            pos = end_pos
 | 
				
			||||||
 | 
					        else:
 | 
				
			||||||
 | 
					            if wire_type == 0:
 | 
				
			||||||
 | 
					                _, pos = decode_varint(buf, pos)
 | 
				
			||||||
 | 
					            elif wire_type == 5:
 | 
				
			||||||
 | 
					                pos += 4
 | 
				
			||||||
 | 
					            else:
 | 
				
			||||||
 | 
					                pos += 1
 | 
				
			||||||
 | 
					    return result
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					# --- 单元测试与使用范例 ---
 | 
				
			||||||
 | 
					if __name__ == "__main__":
 | 
				
			||||||
 | 
					    print("--- 测试 Raw485Command ---")
 | 
				
			||||||
 | 
					    raw_cmd_data = {'bus_number': 1, 'command_bytes': b'\x01\x03\x00\x00\x00\x02\xc4\x0b'}
 | 
				
			||||||
 | 
					    encoded_raw_cmd = encode_raw_485_command(raw_cmd_data['bus_number'], raw_cmd_data['command_bytes'])
 | 
				
			||||||
 | 
					    decoded_raw_cmd = decode_raw_485_command(encoded_raw_cmd)
 | 
				
			||||||
 | 
					    assert decoded_raw_cmd == raw_cmd_data, f"Expected {raw_cmd_data}, got {decoded_raw_cmd}"
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    print("\n--- 测试 CollectTask ---")
 | 
				
			||||||
 | 
					    collect_task_data = {'command': raw_cmd_data}
 | 
				
			||||||
 | 
					    encoded_collect_task = encode_collect_task(collect_task_data['command'])
 | 
				
			||||||
 | 
					    decoded_collect_task = decode_collect_task(encoded_collect_task)
 | 
				
			||||||
 | 
					    assert decoded_collect_task == collect_task_data, f"Expected {collect_task_data}, got {decoded_collect_task}"
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    print("\n--- 测试 BatchCollectCommand ---")
 | 
				
			||||||
 | 
					    batch_collect_data = {
 | 
				
			||||||
 | 
					        'correlation_id': 'abc-123',
 | 
				
			||||||
 | 
					        'tasks': [
 | 
				
			||||||
 | 
					            {'command': {'bus_number': 1, 'command_bytes': b'\x01\x04\x00\x01\x00\x01\x60\x0a'}},
 | 
				
			||||||
 | 
					            {'command': {'bus_number': 2, 'command_bytes': b'\x02\x03\x00\x01\x00\x01\xd5\xfa'}}
 | 
				
			||||||
 | 
					        ]
 | 
				
			||||||
 | 
					    }
 | 
				
			||||||
 | 
					    encoded_batch_collect = encode_batch_collect_command(batch_collect_data['correlation_id'], batch_collect_data['tasks'])
 | 
				
			||||||
 | 
					    decoded_batch_collect = decode_batch_collect_command(encoded_batch_collect)
 | 
				
			||||||
 | 
					    assert decoded_batch_collect == batch_collect_data, f"Expected {batch_collect_data}, got {decoded_batch_collect}"
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    print("\n--- 测试 CollectResult ---")
 | 
				
			||||||
 | 
					    collect_result_data = {'correlation_id': 'res-456', 'values': [12.34, 56.78]}
 | 
				
			||||||
 | 
					    encoded_collect_result = encode_collect_result(collect_result_data['correlation_id'], collect_result_data['values'])
 | 
				
			||||||
 | 
					    decoded_collect_result = decode_collect_result(encoded_collect_result)
 | 
				
			||||||
 | 
					    assert decoded_collect_result['correlation_id'] == collect_result_data['correlation_id']
 | 
				
			||||||
 | 
					    for i in range(len(collect_result_data['values'])):
 | 
				
			||||||
 | 
					        assert abs(decoded_collect_result['values'][i] - collect_result_data['values'][i]) < 1e-5
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    print("\n--- 测试 Instruction (内含BatchCollectCommand) ---")
 | 
				
			||||||
 | 
					    instruction_batch_collect = encode_instruction('batch_collect_command', batch_collect_data)
 | 
				
			||||||
 | 
					    decoded_instruction_batch_collect = decode_instruction(instruction_batch_collect)
 | 
				
			||||||
 | 
					    assert decoded_instruction_batch_collect == {'batch_collect_command': batch_collect_data}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    print("\n所有测试均已通过!")
 | 
				
			||||||
@@ -10,12 +10,12 @@
 | 
				
			|||||||
- 从队列中取出任务,并交给Processor进行耗时处理。
 | 
					- 从队列中取出任务,并交给Processor进行耗时处理。
 | 
				
			||||||
"""
 | 
					"""
 | 
				
			||||||
 | 
					
 | 
				
			||||||
import uqueue
 | 
					from app.uqueue import Queue
 | 
				
			||||||
from processor import Processor
 | 
					from app.processor import Processor
 | 
				
			||||||
from logs.logger import log
 | 
					from app.logs.logger import log
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
def worker_task(task_queue: uqueue.Queue, processor: Processor):
 | 
					def worker_task(task_queue: Queue, processor: Processor):
 | 
				
			||||||
    """
 | 
					    """
 | 
				
			||||||
    工作线程的主函数。
 | 
					    工作线程的主函数。
 | 
				
			||||||
    
 | 
					    
 | 
				
			||||||
@@ -15,25 +15,23 @@
 | 
				
			|||||||
 | 
					
 | 
				
			||||||
import time
 | 
					import time
 | 
				
			||||||
import _thread
 | 
					import _thread
 | 
				
			||||||
from config import config
 | 
					from app.config import config
 | 
				
			||||||
import uqueue  # 导入我们自己创建的本地uqueue模块
 | 
					from app.uqueue import Queue  # 导入我们自己创建的本地uqueue模块
 | 
				
			||||||
 | 
					
 | 
				
			||||||
# 导入接口和实现
 | 
					# 导入接口和实现
 | 
				
			||||||
from lora.lora_interface import ILoraManager
 | 
					from app.lora.lora_mesh_uart_passthrough_manager import LoRaMeshUartPassthroughManager
 | 
				
			||||||
from bus.bus_interface import IBusManager
 | 
					from app.bus.rs485_manager import RS485Manager
 | 
				
			||||||
from lora.lora_mesh_uart_passthrough_manager import LoRaMeshUartPassthroughManager
 | 
					from app.processor import Processor
 | 
				
			||||||
from bus.rs485_manager import RS485Manager
 | 
					 | 
				
			||||||
from processor import Processor
 | 
					 | 
				
			||||||
 | 
					
 | 
				
			||||||
# 导入工作线程的执行函数
 | 
					# 导入工作线程的执行函数
 | 
				
			||||||
from worker import worker_task
 | 
					from app.worker import worker_task
 | 
				
			||||||
from logs.logger import log
 | 
					from app.logs.logger import log
 | 
				
			||||||
 | 
					
 | 
				
			||||||
# --- 模块级变量定义 (带有类型提示) ---
 | 
					# --- 模块级变量定义 (带有类型提示) ---
 | 
				
			||||||
lora_manager: ILoraManager | None = None
 | 
					lora_manager: LoRaMeshUartPassthroughManager | None = None
 | 
				
			||||||
bus_manager: IBusManager | None = None
 | 
					bus_manager: RS485Manager | None = None
 | 
				
			||||||
processor: Processor | None = None
 | 
					processor: Processor | None = None
 | 
				
			||||||
task_queue: uqueue.Queue | None = None
 | 
					task_queue: Queue | None = None
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
def setup():
 | 
					def setup():
 | 
				
			||||||
@@ -52,7 +50,7 @@ def setup():
 | 
				
			|||||||
 | 
					
 | 
				
			||||||
    # 2. 从配置文件读取队列长度,并创建线程安全的队列
 | 
					    # 2. 从配置文件读取队列长度,并创建线程安全的队列
 | 
				
			||||||
    queue_size = config.SYSTEM_PARAMS.get('task_queue_max_size', 10)
 | 
					    queue_size = config.SYSTEM_PARAMS.get('task_queue_max_size', 10)
 | 
				
			||||||
    task_queue = uqueue.Queue(maxsize=queue_size)
 | 
					    task_queue = Queue(maxsize=queue_size)
 | 
				
			||||||
    log(f"任务队列已创建,最大容量: {queue_size}")
 | 
					    log(f"任务队列已创建,最大容量: {queue_size}")
 | 
				
			||||||
 | 
					
 | 
				
			||||||
    # 3. 启动工作线程
 | 
					    # 3. 启动工作线程
 | 
				
			||||||
@@ -67,8 +65,8 @@ def loop():
 | 
				
			|||||||
    只负责监听LoRa,并将数据放入队列。
 | 
					    只负责监听LoRa,并将数据放入队列。
 | 
				
			||||||
    """
 | 
					    """
 | 
				
			||||||
    packet = lora_manager.receive_packet()
 | 
					    packet = lora_manager.receive_packet()
 | 
				
			||||||
 | 
					 | 
				
			||||||
    if packet:
 | 
					    if packet:
 | 
				
			||||||
 | 
					        log(f"主线程:收到新LoRa数据包: {packet.hex()}")
 | 
				
			||||||
        if task_queue.full():
 | 
					        if task_queue.full():
 | 
				
			||||||
            log("警告:任务队列已满,新的LoRa数据包被丢弃!")
 | 
					            log("警告:任务队列已满,新的LoRa数据包被丢弃!")
 | 
				
			||||||
            return
 | 
					            return
 | 
				
			||||||
@@ -1,64 +0,0 @@
 | 
				
			|||||||
#!/usr/bin/env python
 | 
					 | 
				
			||||||
# -*- coding: utf-8 -*-
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
"""
 | 
					 | 
				
			||||||
LoRa模块的具体实现 (UART Passthrough for LoRa Mesh)
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
负责与LoRa模块进行底层通信,并向上层提供标准化的数据包收发接口。
 | 
					 | 
				
			||||||
这个实现针对的是通过UART进行透传的LoRa Mesh模块。
 | 
					 | 
				
			||||||
"""
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
from .lora_interface import ILoraManager
 | 
					 | 
				
			||||||
from main.logs.logger import log
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
class LoRaMeshUartPassthroughManager(ILoraManager):
 | 
					 | 
				
			||||||
    """
 | 
					 | 
				
			||||||
    通过UART与LoRa Mesh模块通信的处理器实现 (透传模式)。
 | 
					 | 
				
			||||||
    """
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
    def __init__(self, lora_config: dict):
 | 
					 | 
				
			||||||
        """
 | 
					 | 
				
			||||||
        初始化LoRa处理器。
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
        Args:
 | 
					 | 
				
			||||||
            lora_config (dict): 来自全局配置文件的LoRa配置字典。
 | 
					 | 
				
			||||||
        """
 | 
					 | 
				
			||||||
        log("LoRaMeshUartPassthroughHandler: 初始化...")
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
        # --- 将配置注入到实例变量 ---
 | 
					 | 
				
			||||||
        self.master_address = lora_config.get('master_address')
 | 
					 | 
				
			||||||
        self.uart_id = lora_config.get('uart_id')
 | 
					 | 
				
			||||||
        self.baudrate = lora_config.get('baudrate')
 | 
					 | 
				
			||||||
        self.pins = lora_config.get('pins')
 | 
					 | 
				
			||||||
        self.lora_mesh_mode = lora_config.get('lora_mesh_mode')
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
        # 在这里可以添加真实的硬件初始化代码,例如初始化UART
 | 
					 | 
				
			||||||
        # self.uart = UART(self.uart_id, self.baudrate, tx=self.pins['tx'], rx=self.pins['rx'])
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
        log(f"LoRaMeshUartPassthroughHandler: 配置加载完成. UART ID: {self.uart_id}, Baudrate: {self.baudrate}")
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
    def receive_packet(self):
 | 
					 | 
				
			||||||
        """
 | 
					 | 
				
			||||||
        【实现】非阻塞地检查并接收一个数据包。
 | 
					 | 
				
			||||||
        (当前为存根实现)
 | 
					 | 
				
			||||||
        """
 | 
					 | 
				
			||||||
        # 具体的实现将在这里...
 | 
					 | 
				
			||||||
        # e.g. self.uart.read()
 | 
					 | 
				
			||||||
        pass
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
    def send_packet(self, data_bytes: bytes) -> bool:
 | 
					 | 
				
			||||||
        """
 | 
					 | 
				
			||||||
        【实现】发送一个数据包。
 | 
					 | 
				
			||||||
        (当前为存根实现)
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
        Args:
 | 
					 | 
				
			||||||
            data_bytes (bytes): 需要发送的字节数据。
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
        Returns:
 | 
					 | 
				
			||||||
            bool: True表示发送成功,False表示失败。
 | 
					 | 
				
			||||||
        """
 | 
					 | 
				
			||||||
        # 具体的实现将在这里...
 | 
					 | 
				
			||||||
        # e.g. self.uart.write(data_bytes)
 | 
					 | 
				
			||||||
        log(f"LoRaMeshUartPassthroughHandler: 模拟发送数据 -> {data_bytes}")
 | 
					 | 
				
			||||||
        return True
 | 
					 | 
				
			||||||
@@ -1,419 +0,0 @@
 | 
				
			|||||||
#!/usr/bin/env python
 | 
					 | 
				
			||||||
# -*- coding: utf-8 -*-
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
"""
 | 
					 | 
				
			||||||
根据client.proto生成的解析代码
 | 
					 | 
				
			||||||
适用于ESP32 MicroPython环境
 | 
					 | 
				
			||||||
"""
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
import struct
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
# --- Protobuf基础类型辅助函数 ---
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
def encode_varint(value):
 | 
					 | 
				
			||||||
    """编码varint整数"""
 | 
					 | 
				
			||||||
    buf = bytearray()
 | 
					 | 
				
			||||||
    while value >= 0x80:
 | 
					 | 
				
			||||||
        buf.append((value & 0x7F) | 0x80)
 | 
					 | 
				
			||||||
        value >>= 7
 | 
					 | 
				
			||||||
    buf.append(value & 0x7F)
 | 
					 | 
				
			||||||
    return buf
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
def decode_varint(buf, pos=0):
 | 
					 | 
				
			||||||
    """解码varint整数"""
 | 
					 | 
				
			||||||
    result = 0
 | 
					 | 
				
			||||||
    shift = 0
 | 
					 | 
				
			||||||
    while pos < len(buf):
 | 
					 | 
				
			||||||
        byte = buf[pos]
 | 
					 | 
				
			||||||
        pos += 1
 | 
					 | 
				
			||||||
        result |= (byte & 0x7F) << shift
 | 
					 | 
				
			||||||
        if not (byte & 0x80):
 | 
					 | 
				
			||||||
            break
 | 
					 | 
				
			||||||
        shift += 7
 | 
					 | 
				
			||||||
    return result, pos
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
def encode_string(value):
 | 
					 | 
				
			||||||
    """编码字符串"""
 | 
					 | 
				
			||||||
    value_bytes = value.encode('utf-8')
 | 
					 | 
				
			||||||
    length = encode_varint(len(value_bytes))
 | 
					 | 
				
			||||||
    return length + value_bytes
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
def decode_string(buf, pos=0):
 | 
					 | 
				
			||||||
    """解码字符串"""
 | 
					 | 
				
			||||||
    length, pos = decode_varint(buf, pos)
 | 
					 | 
				
			||||||
    value = buf[pos:pos + length].decode('utf-8')
 | 
					 | 
				
			||||||
    pos += length
 | 
					 | 
				
			||||||
    return value, pos
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
# --- 消息编码/解码函数 ---
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
def encode_raw_485_command(bus_number, command_bytes):
 | 
					 | 
				
			||||||
    """
 | 
					 | 
				
			||||||
    编码Raw485Command消息
 | 
					 | 
				
			||||||
    Args:
 | 
					 | 
				
			||||||
        bus_number (int): 总线号
 | 
					 | 
				
			||||||
        command_bytes (bytes): 原始485指令
 | 
					 | 
				
			||||||
    Returns:
 | 
					 | 
				
			||||||
        bytearray: 编码后的数据
 | 
					 | 
				
			||||||
    """
 | 
					 | 
				
			||||||
    result = bytearray()
 | 
					 | 
				
			||||||
    # bus_number (field 1, wire type 0)
 | 
					 | 
				
			||||||
    result.extend(encode_varint((1 << 3) | 0))
 | 
					 | 
				
			||||||
    result.extend(encode_varint(bus_number))
 | 
					 | 
				
			||||||
    # command_bytes (field 2, wire type 2)
 | 
					 | 
				
			||||||
    result.extend(encode_varint((2 << 3) | 2))
 | 
					 | 
				
			||||||
    result.extend(encode_varint(len(command_bytes)))
 | 
					 | 
				
			||||||
    result.extend(command_bytes)
 | 
					 | 
				
			||||||
    return result
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
def decode_raw_485_command(buf):
 | 
					 | 
				
			||||||
    """
 | 
					 | 
				
			||||||
    解码Raw485Command消息
 | 
					 | 
				
			||||||
    Args:
 | 
					 | 
				
			||||||
        buf (bytes): 编码后的数据
 | 
					 | 
				
			||||||
    Returns:
 | 
					 | 
				
			||||||
        dict: 解码后的消息
 | 
					 | 
				
			||||||
    """
 | 
					 | 
				
			||||||
    result = {}
 | 
					 | 
				
			||||||
    pos = 0
 | 
					 | 
				
			||||||
    while pos < len(buf):
 | 
					 | 
				
			||||||
        tag, pos = decode_varint(buf, pos)
 | 
					 | 
				
			||||||
        field_number = tag >> 3
 | 
					 | 
				
			||||||
        wire_type = tag & 0x07
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
        if field_number == 1:  # bus_number
 | 
					 | 
				
			||||||
            if wire_type == 0:
 | 
					 | 
				
			||||||
                value, pos = decode_varint(buf, pos)
 | 
					 | 
				
			||||||
                result['bus_number'] = value
 | 
					 | 
				
			||||||
        elif field_number == 2:  # command_bytes
 | 
					 | 
				
			||||||
            if wire_type == 2:
 | 
					 | 
				
			||||||
                length, pos = decode_varint(buf, pos)
 | 
					 | 
				
			||||||
                value = buf[pos:pos + length]
 | 
					 | 
				
			||||||
                pos += length
 | 
					 | 
				
			||||||
                result['command_bytes'] = value
 | 
					 | 
				
			||||||
        else:
 | 
					 | 
				
			||||||
            # 跳过未知字段
 | 
					 | 
				
			||||||
            if wire_type == 0:
 | 
					 | 
				
			||||||
                _, pos = decode_varint(buf, pos)
 | 
					 | 
				
			||||||
            elif wire_type == 2:
 | 
					 | 
				
			||||||
                length, pos = decode_varint(buf, pos);
 | 
					 | 
				
			||||||
                pos += length
 | 
					 | 
				
			||||||
            else:
 | 
					 | 
				
			||||||
                pos += 1
 | 
					 | 
				
			||||||
    return result
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
def encode_collect_task(command_msg):
 | 
					 | 
				
			||||||
    """
 | 
					 | 
				
			||||||
    编码CollectTask消息
 | 
					 | 
				
			||||||
    Args:
 | 
					 | 
				
			||||||
        command_msg (dict): Raw485Command消息字典
 | 
					 | 
				
			||||||
    Returns:
 | 
					 | 
				
			||||||
        bytearray: 编码后的数据
 | 
					 | 
				
			||||||
    """
 | 
					 | 
				
			||||||
    result = bytearray()
 | 
					 | 
				
			||||||
    # command (field 1, wire type 2)
 | 
					 | 
				
			||||||
    encoded_command = encode_raw_485_command(command_msg['bus_number'], command_msg['command_bytes'])
 | 
					 | 
				
			||||||
    result.extend(encode_varint((1 << 3) | 2))  # 字段编号已改为1
 | 
					 | 
				
			||||||
    result.extend(encode_varint(len(encoded_command)))
 | 
					 | 
				
			||||||
    result.extend(encoded_command)
 | 
					 | 
				
			||||||
    return result
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
def decode_collect_task(buf):
 | 
					 | 
				
			||||||
    """
 | 
					 | 
				
			||||||
    解码CollectTask消息
 | 
					 | 
				
			||||||
    Args:
 | 
					 | 
				
			||||||
        buf (bytes): 编码后的数据
 | 
					 | 
				
			||||||
    Returns:
 | 
					 | 
				
			||||||
        dict: 解码后的消息
 | 
					 | 
				
			||||||
    """
 | 
					 | 
				
			||||||
    result = {}
 | 
					 | 
				
			||||||
    pos = 0
 | 
					 | 
				
			||||||
    while pos < len(buf):
 | 
					 | 
				
			||||||
        tag, pos = decode_varint(buf, pos)
 | 
					 | 
				
			||||||
        field_number = tag >> 3
 | 
					 | 
				
			||||||
        wire_type = tag & 0x07
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
        if field_number == 1:  # command
 | 
					 | 
				
			||||||
            if wire_type == 2:
 | 
					 | 
				
			||||||
                length, pos = decode_varint(buf, pos)
 | 
					 | 
				
			||||||
                value_buf = buf[pos:pos + length]
 | 
					 | 
				
			||||||
                pos += length
 | 
					 | 
				
			||||||
                result['command'] = decode_raw_485_command(value_buf)
 | 
					 | 
				
			||||||
        else:
 | 
					 | 
				
			||||||
            if wire_type == 0:
 | 
					 | 
				
			||||||
                _, pos = decode_varint(buf, pos)
 | 
					 | 
				
			||||||
            elif wire_type == 2:
 | 
					 | 
				
			||||||
                length, pos = decode_varint(buf, pos);
 | 
					 | 
				
			||||||
                pos += length
 | 
					 | 
				
			||||||
            else:
 | 
					 | 
				
			||||||
                pos += 1
 | 
					 | 
				
			||||||
    return result
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
def encode_batch_collect_command(correlation_id, tasks):
 | 
					 | 
				
			||||||
    """
 | 
					 | 
				
			||||||
    编码BatchCollectCommand消息
 | 
					 | 
				
			||||||
    Args:
 | 
					 | 
				
			||||||
        correlation_id (str): 关联ID
 | 
					 | 
				
			||||||
        tasks (list): CollectTask消息字典列表
 | 
					 | 
				
			||||||
    Returns:
 | 
					 | 
				
			||||||
        bytearray: 编码后的数据
 | 
					 | 
				
			||||||
    """
 | 
					 | 
				
			||||||
    result = bytearray()
 | 
					 | 
				
			||||||
    # correlation_id (field 1, wire type 2)
 | 
					 | 
				
			||||||
    result.extend(encode_varint((1 << 3) | 2))
 | 
					 | 
				
			||||||
    result.extend(encode_string(correlation_id))
 | 
					 | 
				
			||||||
    # tasks (field 2, wire type 2) - repeated
 | 
					 | 
				
			||||||
    for task in tasks:
 | 
					 | 
				
			||||||
        encoded_task = encode_collect_task(task['command'])
 | 
					 | 
				
			||||||
        result.extend(encode_varint((2 << 3) | 2))
 | 
					 | 
				
			||||||
        result.extend(encode_varint(len(encoded_task)))
 | 
					 | 
				
			||||||
        result.extend(encoded_task)
 | 
					 | 
				
			||||||
    return result
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
def decode_batch_collect_command(buf):
 | 
					 | 
				
			||||||
    """
 | 
					 | 
				
			||||||
    解码BatchCollectCommand消息
 | 
					 | 
				
			||||||
    Args:
 | 
					 | 
				
			||||||
        buf (bytes): 编码后的数据
 | 
					 | 
				
			||||||
    Returns:
 | 
					 | 
				
			||||||
        dict: 解码后的消息
 | 
					 | 
				
			||||||
    """
 | 
					 | 
				
			||||||
    result = {'tasks': []}
 | 
					 | 
				
			||||||
    pos = 0
 | 
					 | 
				
			||||||
    while pos < len(buf):
 | 
					 | 
				
			||||||
        tag, pos = decode_varint(buf, pos)
 | 
					 | 
				
			||||||
        field_number = tag >> 3
 | 
					 | 
				
			||||||
        wire_type = tag & 0x07
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
        if field_number == 1:  # correlation_id
 | 
					 | 
				
			||||||
            if wire_type == 2:
 | 
					 | 
				
			||||||
                value, pos = decode_string(buf, pos)
 | 
					 | 
				
			||||||
                result['correlation_id'] = value
 | 
					 | 
				
			||||||
        elif field_number == 2:  # tasks (repeated)
 | 
					 | 
				
			||||||
            if wire_type == 2:
 | 
					 | 
				
			||||||
                length, pos = decode_varint(buf, pos)
 | 
					 | 
				
			||||||
                value_buf = buf[pos:pos + length]
 | 
					 | 
				
			||||||
                pos += length
 | 
					 | 
				
			||||||
                result['tasks'].append(decode_collect_task(value_buf))
 | 
					 | 
				
			||||||
        else:
 | 
					 | 
				
			||||||
            if wire_type == 0:
 | 
					 | 
				
			||||||
                _, pos = decode_varint(buf, pos)
 | 
					 | 
				
			||||||
            elif wire_type == 2:
 | 
					 | 
				
			||||||
                length, pos = decode_varint(buf, pos);
 | 
					 | 
				
			||||||
                pos += length
 | 
					 | 
				
			||||||
            else:
 | 
					 | 
				
			||||||
                pos += 1
 | 
					 | 
				
			||||||
    return result
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
def encode_collect_result(correlation_id, values):
 | 
					 | 
				
			||||||
    """
 | 
					 | 
				
			||||||
    编码CollectResult消息
 | 
					 | 
				
			||||||
    Args:
 | 
					 | 
				
			||||||
        correlation_id (str): 关联ID
 | 
					 | 
				
			||||||
        values (list): 采集值列表 (float)
 | 
					 | 
				
			||||||
    Returns:
 | 
					 | 
				
			||||||
        bytearray: 编码后的数据
 | 
					 | 
				
			||||||
    """
 | 
					 | 
				
			||||||
    result = bytearray()
 | 
					 | 
				
			||||||
    # correlation_id (field 1, wire type 2)
 | 
					 | 
				
			||||||
    result.extend(encode_varint((1 << 3) | 2))
 | 
					 | 
				
			||||||
    result.extend(encode_string(correlation_id))
 | 
					 | 
				
			||||||
    # values (field 2, wire type 5) - repeated fixed32
 | 
					 | 
				
			||||||
    for value in values:
 | 
					 | 
				
			||||||
        result.extend(encode_varint((2 << 3) | 5))  # Tag for fixed32
 | 
					 | 
				
			||||||
        result.extend(struct.pack('<f', value))  # 小端序浮点数
 | 
					 | 
				
			||||||
    return result
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
def decode_collect_result(buf):
 | 
					 | 
				
			||||||
    """
 | 
					 | 
				
			||||||
    解码CollectResult消息
 | 
					 | 
				
			||||||
    Args:
 | 
					 | 
				
			||||||
        buf (bytes): 编码后的数据
 | 
					 | 
				
			||||||
    Returns:
 | 
					 | 
				
			||||||
        dict: 解码后的消息
 | 
					 | 
				
			||||||
    """
 | 
					 | 
				
			||||||
    result = {'values': []}
 | 
					 | 
				
			||||||
    pos = 0
 | 
					 | 
				
			||||||
    while pos < len(buf):
 | 
					 | 
				
			||||||
        tag, pos = decode_varint(buf, pos)
 | 
					 | 
				
			||||||
        field_number = tag >> 3
 | 
					 | 
				
			||||||
        wire_type = tag & 0x07
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
        if field_number == 1:  # correlation_id
 | 
					 | 
				
			||||||
            if wire_type == 2:
 | 
					 | 
				
			||||||
                value, pos = decode_string(buf, pos)
 | 
					 | 
				
			||||||
                result['correlation_id'] = value
 | 
					 | 
				
			||||||
        elif field_number == 2:  # values (repeated)
 | 
					 | 
				
			||||||
            if wire_type == 5:  # fixed32
 | 
					 | 
				
			||||||
                value = struct.unpack('<f', buf[pos:pos + 4])[0]
 | 
					 | 
				
			||||||
                pos += 4
 | 
					 | 
				
			||||||
                result['values'].append(value)
 | 
					 | 
				
			||||||
        else:
 | 
					 | 
				
			||||||
            if wire_type == 0:
 | 
					 | 
				
			||||||
                _, pos = decode_varint(buf, pos)
 | 
					 | 
				
			||||||
            elif wire_type == 5:
 | 
					 | 
				
			||||||
                pos += 4  # fixed32
 | 
					 | 
				
			||||||
            elif wire_type == 2:
 | 
					 | 
				
			||||||
                length, pos = decode_varint(buf, pos);
 | 
					 | 
				
			||||||
                pos += length
 | 
					 | 
				
			||||||
            else:
 | 
					 | 
				
			||||||
                pos += 1
 | 
					 | 
				
			||||||
    return result
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
def encode_instruction(payload_type, payload_data):
 | 
					 | 
				
			||||||
    """
 | 
					 | 
				
			||||||
    编码Instruction消息 (包含oneof字段)
 | 
					 | 
				
			||||||
    Args:
 | 
					 | 
				
			||||||
        payload_type (str): oneof字段的类型 ('raw_485_command', 'batch_collect_command', 'collect_result')
 | 
					 | 
				
			||||||
        payload_data (dict): 对应类型的消息字典
 | 
					 | 
				
			||||||
    Returns:
 | 
					 | 
				
			||||||
        bytearray: 编码后的数据
 | 
					 | 
				
			||||||
    """
 | 
					 | 
				
			||||||
    result = bytearray()
 | 
					 | 
				
			||||||
    encoded_payload = bytearray()
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
    if payload_type == 'raw_485_command':
 | 
					 | 
				
			||||||
        encoded_payload = encode_raw_485_command(payload_data['bus_number'], payload_data['command_bytes'])
 | 
					 | 
				
			||||||
        result.extend(encode_varint((1 << 3) | 2))  # field 1, wire type 2
 | 
					 | 
				
			||||||
    elif payload_type == 'batch_collect_command':
 | 
					 | 
				
			||||||
        encoded_payload = encode_batch_collect_command(payload_data['correlation_id'], payload_data['tasks'])
 | 
					 | 
				
			||||||
        result.extend(encode_varint((2 << 3) | 2))  # field 2, wire type 2
 | 
					 | 
				
			||||||
    elif payload_type == 'collect_result':
 | 
					 | 
				
			||||||
        encoded_payload = encode_collect_result(payload_data['correlation_id'], payload_data['values'])
 | 
					 | 
				
			||||||
        result.extend(encode_varint((3 << 3) | 2))  # field 3, wire type 2
 | 
					 | 
				
			||||||
    else:
 | 
					 | 
				
			||||||
        raise ValueError("未知的指令负载类型")
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
    result.extend(encode_varint(len(encoded_payload)))
 | 
					 | 
				
			||||||
    result.extend(encoded_payload)
 | 
					 | 
				
			||||||
    return result
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
def decode_instruction(buf):
 | 
					 | 
				
			||||||
    """
 | 
					 | 
				
			||||||
    解码Instruction消息
 | 
					 | 
				
			||||||
    Args:
 | 
					 | 
				
			||||||
        buf (bytes): 编码后的数据
 | 
					 | 
				
			||||||
    Returns:
 | 
					 | 
				
			||||||
        dict: 解码后的消息
 | 
					 | 
				
			||||||
    """
 | 
					 | 
				
			||||||
    result = {}
 | 
					 | 
				
			||||||
    pos = 0
 | 
					 | 
				
			||||||
    while pos < len(buf):
 | 
					 | 
				
			||||||
        tag, pos = decode_varint(buf, pos)
 | 
					 | 
				
			||||||
        field_number = tag >> 3
 | 
					 | 
				
			||||||
        wire_type = tag & 0x07
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
        if wire_type == 2:  # 所有oneof字段都使用长度分隔类型
 | 
					 | 
				
			||||||
            length, pos = decode_varint(buf, pos)
 | 
					 | 
				
			||||||
            value_buf = buf[pos:pos + length]
 | 
					 | 
				
			||||||
            pos += length
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
            if field_number == 1:  # raw_485_command
 | 
					 | 
				
			||||||
                result['raw_485_command'] = decode_raw_485_command(value_buf)
 | 
					 | 
				
			||||||
            elif field_number == 2:  # batch_collect_command
 | 
					 | 
				
			||||||
                result['batch_collect_command'] = decode_batch_collect_command(value_buf)
 | 
					 | 
				
			||||||
            elif field_number == 3:  # collect_result
 | 
					 | 
				
			||||||
                result['collect_result'] = decode_collect_result(value_buf)
 | 
					 | 
				
			||||||
        else:
 | 
					 | 
				
			||||||
            # 跳过未知字段
 | 
					 | 
				
			||||||
            if wire_type == 0:
 | 
					 | 
				
			||||||
                _, pos = decode_varint(buf, pos)
 | 
					 | 
				
			||||||
            elif wire_type == 5:
 | 
					 | 
				
			||||||
                pos += 4
 | 
					 | 
				
			||||||
            elif wire_type == 2:
 | 
					 | 
				
			||||||
                length, pos = decode_varint(buf, pos);
 | 
					 | 
				
			||||||
                pos += length
 | 
					 | 
				
			||||||
            else:
 | 
					 | 
				
			||||||
                pos += 1
 | 
					 | 
				
			||||||
    return result
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
# --- 单元测试与使用范例 ---
 | 
					 | 
				
			||||||
if __name__ == "__main__":
 | 
					 | 
				
			||||||
    print("--- 测试 Raw485Command ---")
 | 
					 | 
				
			||||||
    raw_cmd_data = {'bus_number': 1, 'command_bytes': b'\x01\x03\x00\x00\x00\x02\xc4\x0b'}
 | 
					 | 
				
			||||||
    encoded_raw_cmd = encode_raw_485_command(raw_cmd_data['bus_number'], raw_cmd_data['command_bytes'])
 | 
					 | 
				
			||||||
    print(f"编码后 Raw485Command: {encoded_raw_cmd.hex()}")
 | 
					 | 
				
			||||||
    decoded_raw_cmd = decode_raw_485_command(encoded_raw_cmd)
 | 
					 | 
				
			||||||
    print(f"解码后 Raw485Command: {decoded_raw_cmd}")
 | 
					 | 
				
			||||||
    assert decoded_raw_cmd == raw_cmd_data
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
    print("\n--- 测试 CollectTask ---")
 | 
					 | 
				
			||||||
    collect_task_data = {'command': raw_cmd_data}
 | 
					 | 
				
			||||||
    encoded_collect_task = encode_collect_task(collect_task_data['command'])
 | 
					 | 
				
			||||||
    print(f"编码后 CollectTask: {encoded_collect_task.hex()}")
 | 
					 | 
				
			||||||
    decoded_collect_task = decode_collect_task(encoded_collect_task)
 | 
					 | 
				
			||||||
    print(f"解码后 CollectTask: {decoded_collect_task}")
 | 
					 | 
				
			||||||
    assert decoded_collect_task == collect_task_data
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
    print("\n--- 测试 BatchCollectCommand ---")
 | 
					 | 
				
			||||||
    batch_collect_data = {
 | 
					 | 
				
			||||||
        'correlation_id': 'abc-123',
 | 
					 | 
				
			||||||
        'tasks': [
 | 
					 | 
				
			||||||
            {'command': {'bus_number': 1, 'command_bytes': b'\x01\x03\x00\x00\x00\x02\xc4\x0b'}},
 | 
					 | 
				
			||||||
            {'command': {'bus_number': 2, 'command_bytes': b'\x02\x03\x00\x01\x00\x01\xd5\xfa'}}
 | 
					 | 
				
			||||||
        ]
 | 
					 | 
				
			||||||
    }
 | 
					 | 
				
			||||||
    encoded_batch_collect = encode_batch_collect_command(batch_collect_data['correlation_id'],
 | 
					 | 
				
			||||||
                                                         batch_collect_data['tasks'])
 | 
					 | 
				
			||||||
    print(f"编码后 BatchCollectCommand: {encoded_batch_collect.hex()}")
 | 
					 | 
				
			||||||
    decoded_batch_collect = decode_batch_collect_command(encoded_batch_collect)
 | 
					 | 
				
			||||||
    print(f"解码后 BatchCollectCommand: {decoded_batch_collect}")
 | 
					 | 
				
			||||||
    assert decoded_batch_collect == batch_collect_data
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
    print("\n--- 测试 CollectResult ---")
 | 
					 | 
				
			||||||
    collect_result_data = {
 | 
					 | 
				
			||||||
        'correlation_id': 'res-456',
 | 
					 | 
				
			||||||
        'values': [12.34, 56.78, 90.12]
 | 
					 | 
				
			||||||
    }
 | 
					 | 
				
			||||||
    encoded_collect_result = encode_collect_result(collect_result_data['correlation_id'], collect_result_data['values'])
 | 
					 | 
				
			||||||
    print(f"编码后 CollectResult: {encoded_collect_result.hex()}")
 | 
					 | 
				
			||||||
    decoded_collect_result = decode_collect_result(encoded_collect_result)
 | 
					 | 
				
			||||||
    print(f"解码后 CollectResult: {decoded_collect_result}")
 | 
					 | 
				
			||||||
    # 由于32位浮点数精度问题,直接比较可能会失败,此处设置一个合理的容忍度
 | 
					 | 
				
			||||||
    assert decoded_collect_result['correlation_id'] == collect_result_data['correlation_id']
 | 
					 | 
				
			||||||
    for i in range(len(collect_result_data['values'])):
 | 
					 | 
				
			||||||
        assert abs(decoded_collect_result['values'][i] - collect_result_data['values'][i]) < 1e-5  # 已放宽容忍度
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
    print("\n--- 测试 Instruction (内含Raw485Command) ---")
 | 
					 | 
				
			||||||
    instruction_raw_485 = encode_instruction('raw_485_command', raw_cmd_data)
 | 
					 | 
				
			||||||
    print(f"编码后 Instruction (Raw485Command): {instruction_raw_485.hex()}")
 | 
					 | 
				
			||||||
    decoded_instruction_raw_485 = decode_instruction(instruction_raw_485)
 | 
					 | 
				
			||||||
    print(f"解码后 Instruction (Raw485Command): {decoded_instruction_raw_485}")
 | 
					 | 
				
			||||||
    assert decoded_instruction_raw_485['raw_485_command'] == raw_cmd_data
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
    print("\n--- 测试 Instruction (内含BatchCollectCommand) ---")
 | 
					 | 
				
			||||||
    instruction_batch_collect = encode_instruction('batch_collect_command', batch_collect_data)
 | 
					 | 
				
			||||||
    print(f"编码后 Instruction (BatchCollectCommand): {instruction_batch_collect.hex()}")
 | 
					 | 
				
			||||||
    decoded_instruction_batch_collect = decode_instruction(instruction_batch_collect)
 | 
					 | 
				
			||||||
    print(f"解码后 Instruction (BatchCollectCommand): {decoded_instruction_batch_collect}")
 | 
					 | 
				
			||||||
    assert decoded_instruction_batch_collect['batch_collect_command']['correlation_id'] == batch_collect_data[
 | 
					 | 
				
			||||||
        'correlation_id']
 | 
					 | 
				
			||||||
    assert len(decoded_instruction_batch_collect['batch_collect_command']['tasks']) == len(batch_collect_data['tasks'])
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
    print("\n--- 测试 Instruction (内含CollectResult) ---")
 | 
					 | 
				
			||||||
    instruction_collect_result = encode_instruction('collect_result', collect_result_data)
 | 
					 | 
				
			||||||
    print(f"编码后 Instruction (CollectResult): {instruction_collect_result.hex()}")
 | 
					 | 
				
			||||||
    decoded_instruction_collect_result = decode_instruction(instruction_collect_result)
 | 
					 | 
				
			||||||
    print(f"解码后 Instruction (CollectResult): {decoded_instruction_collect_result}")
 | 
					 | 
				
			||||||
    assert decoded_instruction_collect_result['collect_result']['correlation_id'] == collect_result_data[
 | 
					 | 
				
			||||||
        'correlation_id']
 | 
					 | 
				
			||||||
    for i in range(len(collect_result_data['values'])):
 | 
					 | 
				
			||||||
        assert abs(
 | 
					 | 
				
			||||||
            decoded_instruction_collect_result['collect_result']['values'][i] - collect_result_data['values'][i]) < 1e-5
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
    print("\n所有测试均已通过!")
 | 
					 | 
				
			||||||
		Reference in New Issue
	
	Block a user