@@ -2,6 +2,7 @@ package lora
import (
"bytes"
"context"
"encoding/binary"
"encoding/json"
"fmt"
@@ -17,6 +18,7 @@ import (
"git.huangwc.com/pig/pig-farm-controller/internal/infra/repository"
"git.huangwc.com/pig/pig-farm-controller/internal/infra/transport"
"git.huangwc.com/pig/pig-farm-controller/internal/infra/transport/proto"
"github.com/google/uuid"
"github.com/tarm/serial"
gproto "google.golang.org/protobuf/proto"
@@ -41,8 +43,8 @@ type message struct {
// LoRaMeshUartPassthroughTransport 实现了 transport.Communicator 和 transport.Listener 接口
type LoRaMeshUartPassthroughTransport struct {
ctx context . Context
config config . LoraMeshConfig
logger * logs . Logger
port * serial . Port
mu sync . Mutex // 用于保护对外的公共方法( 如Send) 的并发调用
@@ -87,8 +89,8 @@ type reassemblyBuffer struct {
// NewLoRaMeshUartPassthroughTransport 创建一个新的 LoRaMeshUartPassthroughTransport 实例
func NewLoRaMeshUartPassthroughTransport (
ctx context . Context ,
config config . LoraMeshConfig ,
logger * logs . Logger ,
areaControllerRepo repository . AreaControllerRepository ,
pendingCollectionRepo repository . PendingCollectionRepository ,
deviceRepo repository . DeviceRepository ,
@@ -106,8 +108,8 @@ func NewLoRaMeshUartPassthroughTransport(
}
t := & LoRaMeshUartPassthroughTransport {
ctx : ctx ,
config : config ,
logger : logger ,
port : port ,
state : stateIdle ,
stopChan : make ( chan struct { } ) ,
@@ -126,15 +128,16 @@ func NewLoRaMeshUartPassthroughTransport(
}
// Listen 启动后台监听协程(非阻塞)
func ( t * LoRaMeshUartPassthroughTransport ) Listen ( ) error {
func ( t * LoRaMeshUartPassthroughTransport ) Listen ( ctx context . Context ) error {
loraCtx , logger := logs . Trace ( ctx , t . ctx , "Listen" )
t . wg . Add ( 1 )
go t . workerLoop ( )
t . logger. Info ( "LoRa传输层工作协程已启动" )
go t . workerLoop ( loraCtx )
logger . Info ( "LoRa传输层工作协程已启动" )
return nil
}
// Send 将发送任务提交给worker协程
func ( t * LoRaMeshUartPassthroughTransport ) Send ( address string , payload [ ] byte ) ( * transport . SendResult , error ) {
func ( t * LoRaMeshUartPassthroughTransport ) Send ( ctx context . Context , address string , payload [ ] byte ) ( * transport . SendResult , error ) {
t . mu . Lock ( )
defer t . mu . Unlock ( )
@@ -156,14 +159,16 @@ func (t *LoRaMeshUartPassthroughTransport) Send(address string, payload []byte)
}
// Stop 停止传输层
func ( t * LoRaMeshUartPassthroughTransport ) Stop ( ) error {
func ( t * LoRaMeshUartPassthroughTransport ) Stop ( ctx context . Context ) error {
close ( t . stopChan )
t . wg . Wait ( )
return t . port . Close ( )
}
// workerLoop 是核心的状态机和调度器
func ( t * LoRaMeshUartPassthroughTransport ) workerLoop ( ) {
func ( t * LoRaMeshUartPassthroughTransport ) workerLoop ( ctx context . Context ) {
loraCtx , logger := logs . Trace ( ctx , t . ctx , "workerLoop" )
defer t . wg . Done ( )
readBuffer := make ( [ ] byte , 1024 )
@@ -176,7 +181,7 @@ func (t *LoRaMeshUartPassthroughTransport) workerLoop() {
if t . reassemblyTimeout != nil {
t . reassemblyTimeout . Stop ( )
}
t . logger. Info ( "LoRa传输层工作协程已停止" )
logger . Info ( "LoRa传输层工作协程已停止" )
return
default :
}
@@ -188,7 +193,7 @@ func (t *LoRaMeshUartPassthroughTransport) workerLoop() {
}
if err != nil && err != io . EOF {
// 忽略预期的超时错误(io.EOF), 只记录真正的IO错误
t . logger. Errorf ( "从串口读取数据时发生错误: %v" , err )
logger . Errorf ( "从串口读取数据时发生错误: %v" , err )
}
// 3. 循环解析缓冲区中的完整物理帧
@@ -197,27 +202,29 @@ func (t *LoRaMeshUartPassthroughTransport) workerLoop() {
if frame == nil {
break // 缓冲区中没有更多完整帧了
}
t . handleFrame ( frame )
t . handleFrame ( loraCtx , frame)
}
// 4. 根据当前状态执行主要逻辑
switch t . state {
case stateIdle :
t . runIdleState ( )
t . runIdleState ( loraCtx )
case stateReceiving :
t . runReceivingState ( )
t . runReceivingState ( loraCtx )
default :
}
}
}
// runIdleState 处理空闲状态下的逻辑,主要是检查并启动发送任务
func ( t * LoRaMeshUartPassthroughTransport ) runIdleState ( ) {
func ( t * LoRaMeshUartPassthroughTransport ) runIdleState ( ctx context . Context ) {
loraCtx := logs . AddFuncName ( ctx , t . ctx , "Listen" )
select {
case req := <- t . sendChan :
t . state = stateSending
// 此处为阻塞式发送
result , err := t . executeSend ( req )
result , err := t . executeSend ( loraCtx , req)
req . result <- & sendResultTuple { result : result , err : err }
t . state = stateIdle
default :
@@ -226,10 +233,11 @@ func (t *LoRaMeshUartPassthroughTransport) runIdleState() {
}
// runReceivingState 处理接收状态下的逻辑,主要是检查超时
func ( t * LoRaMeshUartPassthroughTransport ) runReceivingState ( ) {
func ( t * LoRaMeshUartPassthroughTransport ) runReceivingState ( ctx context . Context ) {
logger := logs . TraceLogger ( ctx , t . ctx , "runReceivingState" )
select {
case sourceAddr := <- t . reassemblyTimeoutCh :
t . logger. Warnf ( "接收来自 0x%04X 的消息超时" , sourceAddr )
logger . Warnf ( "接收来自 0x%04X 的消息超时" , sourceAddr )
delete ( t . reassemblyBuffers , sourceAddr )
t . state = stateIdle
default :
@@ -238,7 +246,8 @@ func (t *LoRaMeshUartPassthroughTransport) runReceivingState() {
}
// executeSend 执行完整的发送流程(分片、构建、写入)
func ( t * LoRaMeshUartPassthroughTransport ) executeSend ( req * sendRequest ) ( * transport . SendResult , error ) {
func ( t * LoRaMeshUartPassthroughTransport ) executeSend ( ctx context . Context , req * sendRequest ) ( * transport . SendResult , error ) {
logger := logs . TraceLogger ( ctx , t . ctx , "executeSend" )
chunks := splitPayload ( req . payload , t . config . MaxChunkSize )
totalChunks := uint8 ( len ( chunks ) )
@@ -257,7 +266,7 @@ func (t *LoRaMeshUartPassthroughTransport) executeSend(req *sendRequest) (*trans
frame . WriteByte ( currentChunk ) // 当前包序号
frame . Write ( chunk ) // 数据块
t . logger. Infof ( "构建LoRa数据包: %v" , frame . Bytes ( ) )
logger . Infof ( "构建LoRa数据包: %v" , frame . Bytes ( ) )
_ , err := t . port . Write ( frame . Bytes ( ) )
if err != nil {
return nil , fmt . Errorf ( "写入串口失败: %w" , err )
@@ -272,9 +281,10 @@ func (t *LoRaMeshUartPassthroughTransport) executeSend(req *sendRequest) (*trans
}
// handleFrame 处理一个从串口解析出的完整物理帧
func ( t * LoRaMeshUartPassthroughTransport ) handleFrame ( frame [ ] byte ) {
func ( t * LoRaMeshUartPassthroughTransport ) handleFrame ( ctx context . Context , frame [ ] byte ) {
loraCtx , logger := logs . Trace ( ctx , t . ctx , "handleFrame" )
if len ( frame ) < 8 {
t . logger. Warnf ( "收到了一个无效长度的帧: %d" , len ( frame ) )
logger . Warnf ( "收到了一个无效长度的帧: %d" , len ( frame ) )
return
}
@@ -291,7 +301,7 @@ func (t *LoRaMeshUartPassthroughTransport) handleFrame(frame []byte) {
DestAddr : fmt . Sprintf ( "%04X" , destAddr ) ,
Payload : chunkData ,
}
go t . handleUpstreamMessage ( msg )
go t . handleUpstreamMessage ( loraCtx , msg)
return
}
@@ -316,18 +326,18 @@ func (t *LoRaMeshUartPassthroughTransport) handleFrame(frame []byte) {
t . reassemblyTimeoutCh <- sourceAddr
} )
} else {
t . logger. Warnf ( "在空闲状态下收到了一个来自 0x%04X 的非首包分片,已忽略。" , sourceAddr )
logger . Warnf ( "在空闲状态下收到了一个来自 0x%04X 的非首包分片,已忽略。" , sourceAddr )
}
case stateReceiving :
if sourceAddr != t . currentRecvSource {
t . logger. Warnf ( "正在接收来自 0x%04X 的数据时,收到了另一个源 0x%04X 的分片,已忽略。" , t . currentRecvSource , sourceAddr )
logger . Warnf ( "正在接收来自 0x%04X 的数据时,收到了另一个源 0x%04X 的分片,已忽略。" , t . currentRecvSource , sourceAddr )
return
}
buffer , ok := t . reassemblyBuffers [ sourceAddr ]
if ! ok {
t . logger. Errorf ( "内部错误: 处于接收状态,但没有为 0x%04X 找到缓冲区" , sourceAddr )
logger . Errorf ( "内部错误: 处于接收状态,但没有为 0x%04X 找到缓冲区" , sourceAddr )
t . state = stateIdle // 重置状态
return
}
@@ -352,23 +362,27 @@ func (t *LoRaMeshUartPassthroughTransport) handleFrame(frame []byte) {
DestAddr : fmt . Sprintf ( "%04X" , destAddr ) ,
Payload : fullPayload . Bytes ( ) ,
}
go t . handleUpstreamMessage ( msg )
go t . handleUpstreamMessage ( loraCtx , msg)
// 清理并返回空闲状态
delete ( t . reassemblyBuffers , sourceAddr )
t . state = stateIdle
}
default :
logger . Errorf ( "内部错误: 状态机处于未知状态 %d" , t . state )
}
}
// handleUpstreamMessage 在独立的协程中处理单个上行的、完整的消息。
func ( t * LoRaMeshUartPassthroughTransport ) handleUpstreamMessage ( msg * message ) {
t . logger . Infof ( "开始处理来自 %s 的上行消息" , msg . SourceAddr )
func ( t * LoRaMeshUartPassthroughTransport ) handleUpstreamMessage ( ctx context . Context , msg * message ) {
loraCtx , logger := logs . Trace ( ctx , t . ctx , "handleUpstreamMessage" )
logger . Infof ( "开始处理来自 %s 的上行消息" , msg . SourceAddr )
// 1. 解析外层 "信封"
var instruction proto . Instruction
if err := gproto . Unmarshal ( msg . Payload , & instruction ) ; err != nil {
t . logger. Errorf ( "解析上行 Instruction Protobuf 失败: %v, 源地址: %s, 原始数据: %x" , err , msg . SourceAddr , msg . Payload )
logger . Errorf ( "解析上行 Instruction Protobuf 失败: %v, 源地址: %s, 原始数据: %x" , err , msg . SourceAddr , msg . Payload )
return
}
@@ -379,39 +393,39 @@ func (t *LoRaMeshUartPassthroughTransport) handleUpstreamMessage(msg *message) {
collectResp = p . CollectResult
default :
// 如果上行的数据不是采集结果,记录日志并忽略
t . logger. Infof ( "收到一个非采集响应的上行指令 (类型: %T),无需处理。源地址: %s" , p , msg . SourceAddr )
logger . Infof ( "收到一个非采集响应的上行指令 (类型: %T),无需处理。源地址: %s" , p , msg . SourceAddr )
return
}
if collectResp == nil {
t . logger. Errorf ( "从 Instruction 中提取的 CollectResult 为 nil。源地址: %s" , msg . SourceAddr )
logger . Errorf ( "从 Instruction 中提取的 CollectResult 为 nil。源地址: %s" , msg . SourceAddr )
return
}
correlationID := collectResp . CorrelationId
t . logger. Infof ( "成功解析采集响应 (CorrelationID: %s),包含 %d 个值。" , correlationID , len ( collectResp . Values ) )
logger . Infof ( "成功解析采集响应 (CorrelationID: %s),包含 %d 个值。" , correlationID , len ( collectResp . Values ) )
// 3. 查找区域主控 (注意: LoRa Mesh 的 SourceAddr 对应于区域主控的 NetworkID)
regionalController , err := t . areaControllerRepo . FindByNetworkID ( msg . SourceAddr )
regionalController , err := t . areaControllerRepo . FindByNetworkID ( loraCtx , msg. SourceAddr )
if err != nil {
t . logger. Errorf ( "处理上行消息失败:无法通过源地址 '%s' 找到区域主控设备: %v" , msg . SourceAddr , err )
logger . Errorf ( "处理上行消息失败:无法通过源地址 '%s' 找到区域主控设备: %v" , msg . SourceAddr , err )
return
}
if err := regionalController . SelfCheck ( ) ; err != nil {
t . logger. Errorf ( "处理上行消息失败:区域主控 %v(ID: %d) 未通过自检: %v" , regionalController . Name , regionalController . ID , err )
logger . Errorf ( "处理上行消息失败:区域主控 %v(ID: %d) 未通过自检: %v" , regionalController . Name , regionalController . ID , err )
return
}
// 4. 根据 CorrelationID 查找待处理请求
pendingReq , err := t . pendingCollectionRepo . FindByCorrelationID ( correlationID )
pendingReq , err := t . pendingCollectionRepo . FindByCorrelationID ( loraCtx , correlationID)
if err != nil {
t . logger. Errorf ( "处理采集响应失败:无法找到待处理请求 (CorrelationID: %s): %v" , correlationID , err )
logger . Errorf ( "处理采集响应失败:无法找到待处理请求 (CorrelationID: %s): %v" , correlationID , err )
return
}
// 检查状态,防止重复处理
if pendingReq . Status != models . PendingStatusPending && pendingReq . Status != models . PendingStatusTimedOut {
t . logger. Warnf ( "收到一个已处理过的采集响应 (CorrelationID: %s, Status: %s),将忽略。" , correlationID , pendingReq . Status )
logger . Warnf ( "收到一个已处理过的采集响应 (CorrelationID: %s, Status: %s),将忽略。" , correlationID , pendingReq . Status )
return
}
@@ -419,10 +433,10 @@ func (t *LoRaMeshUartPassthroughTransport) handleUpstreamMessage(msg *message) {
deviceIDs := pendingReq . CommandMetadata
values := collectResp . Values
if len ( deviceIDs ) != len ( values ) {
t . logger. Errorf ( "数据不匹配:下行指令要求采集 %d 个设备,但上行响应包含 %d 个值 (CorrelationID: %s)" , len ( deviceIDs ) , len ( values ) , correlationID )
err = t . pendingCollectionRepo . UpdateStatusToFulfilled ( correlationID , time . Now ( ) )
logger . Errorf ( "数据不匹配:下行指令要求采集 %d 个设备,但上行响应包含 %d 个值 (CorrelationID: %s)" , len ( deviceIDs ) , len ( values ) , correlationID )
err = t . pendingCollectionRepo . UpdateStatusToFulfilled ( loraCtx , correlationID, time . Now ( ) )
if err != nil {
t . logger. Errorf ( "处理采集响应失败:无法更新待处理请求 (CorrelationID: %s) 的状态为完成: %v" , correlationID , err )
logger . Errorf ( "处理采集响应失败:无法更新待处理请求 (CorrelationID: %s) 的状态为完成: %v" , correlationID , err )
}
return
}
@@ -431,31 +445,31 @@ func (t *LoRaMeshUartPassthroughTransport) handleUpstreamMessage(msg *message) {
rawSensorValue := values [ i ]
if math . IsNaN ( float64 ( rawSensorValue ) ) {
t . logger. Warnf ( "设备 (ID: %d) 上报了一个无效的 NaN 值,已跳过当前值的记录。" , deviceID )
logger . Warnf ( "设备 (ID: %d) 上报了一个无效的 NaN 值,已跳过当前值的记录。" , deviceID )
continue
}
dev , err := t . deviceRepo . FindByID ( deviceID )
dev , err := t . deviceRepo . FindByID ( loraCtx , deviceID)
if err != nil {
t . logger. Errorf ( "处理采集数据失败:无法找到设备 (ID: %d): %v" , deviceID , err )
logger . Errorf ( "处理采集数据失败:无法找到设备 (ID: %d): %v" , deviceID , err )
continue
}
if err := dev . SelfCheck ( ) ; err != nil {
t . logger. Warnf ( "跳过设备 %d, 因其未通过自检: %v" , dev . ID , err )
logger . Warnf ( "跳过设备 %d, 因其未通过自检: %v" , dev . ID , err )
continue
}
if err := dev . DeviceTemplate . SelfCheck ( ) ; err != nil {
t . logger. Warnf ( "跳过设备 %d, 因其设备模板未通过自检: %v" , dev . ID , err )
logger . Warnf ( "跳过设备 %d, 因其设备模板未通过自检: %v" , dev . ID , err )
continue
}
var valueDescriptors [ ] * models . ValueDescriptor
if err := dev . DeviceTemplate . ParseValues ( & valueDescriptors ) ; err != nil {
t . logger. Warnf ( "跳过设备 %d, 因其设备模板的 Values 属性解析失败: %v" , dev . ID , err )
logger . Warnf ( "跳过设备 %d, 因其设备模板的 Values 属性解析失败: %v" , dev . ID , err )
continue
}
if len ( valueDescriptors ) == 0 {
t . logger. Warnf ( "跳过设备 %d, 因其设备模板缺少 ValueDescriptor 定义" , dev . ID )
logger . Warnf ( "跳过设备 %d, 因其设备模板缺少 ValueDescriptor 定义" , dev . ID )
continue
}
valueDescriptor := valueDescriptors [ 0 ]
@@ -471,27 +485,29 @@ func (t *LoRaMeshUartPassthroughTransport) handleUpstreamMessage(msg *message) {
case models . SensorTypeWeight :
dataToRecord = models . WeightData { WeightKilograms : parsedValue }
default :
t . logger. Warnf ( "未知的传感器类型 '%s',将使用通用格式记录" , valueDescriptor . Type )
logger . Warnf ( "未知的传感器类型 '%s',将使用通用格式记录" , valueDescriptor . Type )
dataToRecord = map [ string ] float64 { "value" : parsedValue }
}
t . recordSensorData ( regionalController . ID , dev . ID , time . Now ( ) , valueDescriptor . Type , dataToRecord )
t . logger. Infof ( "成功记录传感器数据: 设备ID=%d, 类型=%s, 原始值=%f, 解析值=%.2f" , dev . ID , valueDescriptor . Type , rawSensorValue , parsedValue )
t . recordSensorData ( loraCtx , regionalController. ID , dev . ID , time . Now ( ) , valueDescriptor . Type , dataToRecord )
logger . Infof ( "成功记录传感器数据: 设备ID=%d, 类型=%s, 原始值=%f, 解析值=%.2f" , dev . ID , valueDescriptor . Type , rawSensorValue , parsedValue )
}
// 6. 更新请求状态为“已完成”
if err := t . pendingCollectionRepo . UpdateStatusToFulfilled ( correlationID , time . Now ( ) ) ; err != nil {
t . logger. Errorf ( "更新待采集请求状态为 'fulfilled' 失败 (CorrelationID: %s): %v" , correlationID , err )
if err := t . pendingCollectionRepo . UpdateStatusToFulfilled ( loraCtx , correlationID, time . Now ( ) ) ; err != nil {
logger . Errorf ( "更新待采集请求状态为 'fulfilled' 失败 (CorrelationID: %s): %v" , correlationID , err )
} else {
t . logger. Infof ( "成功完成并关闭采集请求 (CorrelationID: %s)" , correlationID )
logger . Infof ( "成功完成并关闭采集请求 (CorrelationID: %s)" , correlationID )
}
}
// recordSensorData 是一个通用方法,用于将传感器数据存入数据库。
func ( t * LoRaMeshUartPassthroughTransport ) recordSensorData ( regionalControllerID uint , sensorDeviceID uint , eventTime time . Time , sensorType models . SensorType , data interface { } ) {
func ( t * LoRaMeshUartPassthroughTransport ) recordSensorData ( ctx context . Context , regionalControllerID uint , sensorDeviceID uint , eventTime time . Time , sensorType models . SensorType , data interface { } ) {
loraCtx , logger := logs . Trace ( ctx , t . ctx , "recordSensorData" )
jsonData , err := json . Marshal ( data )
if err != nil {
t . logger. Errorf ( "记录传感器数据失败:序列化数据为 JSON 时出错: %v" , err )
logger . Errorf ( "记录传感器数据失败:序列化数据为 JSON 时出错: %v" , err )
return
}
@@ -503,8 +519,8 @@ func (t *LoRaMeshUartPassthroughTransport) recordSensorData(regionalControllerID
Data : datatypes . JSON ( jsonData ) ,
}
if err := t . sensorDataRepo . Create ( sensorData ) ; err != nil {
t . logger. Errorf ( "记录传感器数据失败:存入数据库时出错: %v" , err )
if err := t . sensorDataRepo . Create ( loraCtx , sensorData) ; err != nil {
logger . Errorf ( "记录传感器数据失败:存入数据库时出错: %v" , err )
}
}