issue-18初步实现

internal/app/service/transport/chirp_stack.go

@@ -7,9 +7,11 @@ import (
 	"net/http"
 	"time"
 
+	"git.huangwc.com/pig/pig-farm-controller/internal/app/service/device/proto"
 	"git.huangwc.com/pig/pig-farm-controller/internal/infra/logs"
 	"git.huangwc.com/pig/pig-farm-controller/internal/infra/models"
 	"git.huangwc.com/pig/pig-farm-controller/internal/infra/repository"
+	gproto "google.golang.org/protobuf/proto"
 	"gorm.io/datatypes"
 )
 
@@ -151,12 +153,20 @@ type GenericSensorReading struct {
 
 // handleUpEvent 处理上行数据事件
 func (c *ChirpStackListener) handleUpEvent(event *UpEvent) {
-	c.logger.Infof("处理 'up' 事件: %+v", event)
+	c.logger.Infof("开始处理 'up' 事件, DevEui: %s", event.DeviceInfo.DevEui)
 
-	// 记录信号强度
-	// 根据业务逻辑，一个猪场只有一个网关，所以 RxInfo 中通常只有一个元素，或者 gateway_id 都是相同的。
-	// 因此，我们只取第一个 RxInfo 中的信号数据即可。
+	// 1. 查找区域主控设备
+	regionalController, err := c.deviceRepo.FindByDevEui(event.DeviceInfo.DevEui)
+	if err != nil {
+		c.logger.Errorf("处理 'up' 事件失败：无法通过 DevEui '%s' 找到区域主控设备: %v", event.DeviceInfo.DevEui, err)
+		return
+	}
+	c.logger.Infof("找到区域主控: %s (ID: %d)", regionalController.Name, regionalController.ID)
+
+	// 2. 记录区域主控的信号强度 (如果存在)
 	if len(event.RxInfo) > 0 {
+		// 根据业务逻辑，一个猪场只有一个网关，所以 RxInfo 中通常只有一个元素，或者 gateway_id 都是相同的。
+		// 因此，我们只取第一个 RxInfo 中的信号数据即可。
 		rx := event.RxInfo[0] // 取第一个接收到的网关信息
 
 		// 构建 SignalMetrics 结构体
@@ -164,73 +174,73 @@ func (c *ChirpStackListener) handleUpEvent(event *UpEvent) {
 			RssiDbm: rx.Rssi,
 			SnrDb:   rx.Snr,
 		}
-
-		// 这里的 event.DeviceInfo.DevEui 对应的是区域主控的 DevEui
-		regionalController, err := c.deviceRepo.FindByDevEui(event.DeviceInfo.DevEui)
-		if err != nil {
-			c.logger.Errorf("处理 'up' 事件失败：无法通过 DevEui '%s' 找到区域主控设备: %v", event.DeviceInfo.DevEui, err)
-			return
-		}
-		// 记录区域主控的信号强度
 		c.recordSensorData(regionalController.ID, regionalController.ID, event.Time, models.SensorDataTypeSignalMetrics, signalMetrics)
+		c.logger.Infof("已记录区域主控 (ID: %d) 的信号强度: RSSI=%d, SNR=%.2f", regionalController.ID, rx.Rssi, rx.Snr)
 	} else {
 		c.logger.Warnf("处理 'up' 事件时未找到 RxInfo，无法记录信号数据。DevEui: %s", event.DeviceInfo.DevEui)
 	}
 
-	// 解析并记录传感器数据 (温度、湿度、重量)
-	// 假设 event.Data (frmPayload) 是 Base64 编码的 JSON 数组字符串
-	if event.Data != "" {
-		decodedData, err := base64.StdEncoding.DecodeString(event.Data)
+	// 3. 处理上报的传感器数据
+	if event.Data == "" {
+		c.logger.Warnf("处理 'up' 事件时 Data 字段为空，无需记录上行数据。DevEui: %s", event.DeviceInfo.DevEui)
+		return
+	}
+
+	// 3.1 Base64 解码
+	decodedData, err := base64.StdEncoding.DecodeString(event.Data)
+	if err != nil {
+		c.logger.Errorf("Base64 解码 'up' 事件的 Data 失败: %v, Data: %s", err, event.Data)
+		return
+	}
+
+	// 3.2 Protobuf 反序列化
+	var payload proto.UplinkPayload
+	if err := gproto.Unmarshal(decodedData, &payload); err != nil {
+		c.logger.Errorf("Protobuf 反序列化 'up' 事件的解码后 Data 失败: %v, Decoded Data: %x", err, decodedData)
+		return
+	}
+	c.logger.Infof("成功解析 Protobuf 数据, 包含 %d 条读数", len(payload.Readings))
+
+	// 3.3 遍历处理每一条读数
+	for _, reading := range payload.Readings {
+		// 3.3.1 根据物理地址查找对应的传感器设备
+		sensorDevice, err := c.deviceRepo.FindByParentAndPhysicalAddress(regionalController.ID, reading.BusNumber, reading.BusAddress)
 		if err != nil {
-			c.logger.Errorf("Base64 解码 'up' 事件的 Data 失败: %v, Data: %s", err, event.Data)
-			return
+			c.logger.Errorf("查找传感器设备失败: %v", err)
+			continue // 继续处理下一条读数
 		}
 
-		var readings []GenericSensorReading
-		if err := json.Unmarshal(decodedData, &readings); err != nil {
-			c.logger.Errorf("解析 'up' 事件的解码后 Data (JSON 数组) 失败: %v, Decoded Data: %s", err, string(decodedData))
-			return
+		// ✨ 核心修正: 直接从 models 包的公开 map 中查找转换关系 ✨
+		sensorDataType, ok := models.DeviceSubTypeToSensorDataTypeMap[sensorDevice.SubType]
+		if !ok {
+			// 如果一个设备子类型不在 map 中, 说明它不是一个需要记录数据的传感器, 这属于正常情况, 无需记录日志.
+			continue
 		}
 
-		// 查找区域主控设备，以便记录其ID
-		regionalController, err := c.deviceRepo.FindByDevEui(event.DeviceInfo.DevEui)
-		if err != nil {
-			c.logger.Errorf("处理 'up' 事件失败：无法通过 DevEui '%s' 找到区域主控设备: %v", event.DeviceInfo.DevEui, err)
-			return
-		}
-
-		for _, reading := range readings {
-			// 根据类型构建具体的传感器数据结构体
-			var sensorData interface{}
-			var sensorDataType models.SensorDataType
-
-			switch reading.Type {
-			case models.SensorDataTypeTemperature: // 使用枚举常量
-				sensorData = models.TemperatureData{
-					TemperatureCelsius: reading.Value,
-				}
-				sensorDataType = models.SensorDataTypeTemperature
-			case models.SensorDataTypeHumidity: // 使用枚举常量
-				sensorData = models.HumidityData{
-					HumidityPercent: reading.Value,
-				}
-				sensorDataType = models.SensorDataTypeHumidity
-			case models.SensorDataTypeWeight: // 使用枚举常量
-				sensorData = models.WeightData{
-					WeightKilograms: reading.Value,
-				}
-				sensorDataType = models.SensorDataTypeWeight
-			default:
-				c.logger.Warnf("处理 'up' 事件时遇到未知传感器类型: %s, Value: %f. 区域主控DevEui: %s, 传感器设备ID: %d",
-					reading.Type, reading.Value, event.DeviceInfo.DevEui, reading.DeviceID)
-				continue // 跳过未知类型
+		// 3.3.2 根据转换后的 sensorDataType 构建具体的数据结构
+		var sensorData interface{}
+		switch sensorDataType {
+		case models.SensorDataTypeTemperature:
+			sensorData = models.TemperatureData{
+				TemperatureCelsius: float64(reading.Value),
 			}
-
-			// 记录普通设备的传感器数据
-			c.recordSensorData(regionalController.ID, reading.DeviceID, event.Time, sensorDataType, sensorData)
+		case models.SensorDataTypeHumidity:
+			sensorData = models.HumidityData{
+				HumidityPercent: float64(reading.Value),
+			}
+		case models.SensorDataTypeWeight:
+			sensorData = models.WeightData{
+				WeightKilograms: float64(reading.Value),
+			}
+		default:
+			// 这个 case 理论上不会被触发
+			c.logger.Warnf("未处理的传感器数据类型 '%s' (设备ID: %d)", sensorDataType, sensorDevice.ID)
+			continue
 		}
-	} else {
-		c.logger.Warnf("处理 'up' 事件时 Data 字段为空，无法记录传感器数据。DevEui: %s", event.DeviceInfo.DevEui)
+
+		// 3.3.3 记录传感器数据
+		c.recordSensorData(regionalController.ID, sensorDevice.ID, event.Time, sensorDataType, sensorData)
+		c.logger.Infof("成功记录传感器数据: 设备ID=%d, 类型=%s, 值=%.2f", sensorDevice.ID, sensorDataType, reading.Value)
 	}
 }