记录信号强度数据

internal/app/service/transport/chirp_stack.go

@@ -4,8 +4,12 @@ import (
 	"encoding/json"
 	"io"
 	"net/http"
+	"time"
 
 	"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"
+	"gorm.io/datatypes"
 )
 
 // ChirpStackListener 主动发送的请求的event字段, 这个字段代表事件类型
@@ -22,12 +26,16 @@ const (
 
 // ChirpStackListener 是一个监听器, 用于监听ChirpStack反馈的设备上行事件
 type ChirpStackListener struct {
-	logger *logs.Logger
+	logger         *logs.Logger
+	sensorDataRepo repository.SensorDataRepository
+	deviceRepo     repository.DeviceRepository
 }
 
-func NewChirpStackListener(logger *logs.Logger) *ChirpStackListener {
+func NewChirpStackListener(logger *logs.Logger, sensorDataRepo repository.SensorDataRepository, deviceRepo repository.DeviceRepository) *ChirpStackListener {
 	return &ChirpStackListener{
-		logger: logger,
+		logger:         logger,
+		sensorDataRepo: sensorDataRepo,
+		deviceRepo:     deviceRepo,
 	}
 }
 
@@ -129,13 +137,44 @@ func (c *ChirpStackListener) handler(data []byte, eventType string) {
 // handleUpEvent 处理上行数据事件
 func (c *ChirpStackListener) handleUpEvent(event *UpEvent) {
 	c.logger.Infof("处理 'up' 事件: %+v", event)
+
+	// 记录信号强度
+	// 根据业务逻辑，一个猪场只有一个网关，所以 RxInfo 中通常只有一个元素，或者 gateway_id 都是相同的。
+	// 因此，我们只取第一个 RxInfo 中的信号数据即可。
+	if len(event.RxInfo) > 0 {
+		rx := event.RxInfo[0] // 取第一个接收到的网关信息
+
+		// 构建 SignalMetrics 结构体
+		signalMetrics := models.SignalMetrics{
+			RSSI: rx.Rssi,
+			SNR:  rx.Snr,
+		}
+		c.recordSensorData(event.DeviceInfo.DevEui, event.Time, models.SensorDataTypeSignalMetrics, signalMetrics)
+	} else {
+		c.logger.Warnf("处理 'up' 事件时未找到 RxInfo，无法记录信号数据。DevEui: %s", event.DeviceInfo.DevEui)
+	}
+
 	// 在这里添加您的业务逻辑
 }
 
 // handleStatusEvent 处理设备状态事件
 func (c *ChirpStackListener) handleStatusEvent(event *StatusEvent) {
 	c.logger.Infof("处接收到理 'status' 事件: %+v", event)
-	// 在这里添加您的业务逻辑
+
+	// 记录信号强度
+	signalMetrics := models.SignalMetrics{
+		Margin: event.Margin,
+	}
+	c.recordSensorData(event.DeviceInfo.DevEui, event.Time, models.SensorDataTypeSignalMetrics, signalMetrics)
+
+	// 记录 电量
+	batteryLevel := models.BatteryLevel{
+		BatteryLevel:            event.BatteryLevel,
+		BatteryLevelUnavailable: event.BatteryLevelUnavailable,
+		ExternalPower:           event.ExternalPower,
+	}
+	c.recordSensorData(event.DeviceInfo.DevEui, event.Time, models.SensorDataTypeBatteryLevel, batteryLevel)
+
 }
 
 // handleAckEvent 处理下行确认事件
@@ -163,8 +202,6 @@ func (c *ChirpStackListener) handleLogEvent(event *LogEvent) {
 		c.logger.Warnf("ChirpStack 日志: [未知级别: %s] %s %s (DevEui: %s)",
 			event.Level, event.Code, event.Description, event.DeviceInfo.DevEui)
 	}
-
-	// 在这里添加您的业务逻辑
 }
 
 // handleJoinEvent 处理入网事件
@@ -190,3 +227,34 @@ func (c *ChirpStackListener) handleIntegrationEvent(event *IntegrationEvent) {
 	c.logger.Infof("接收到 'integration' 事件: %+v", event)
 	// 在这里添加您的业务逻辑
 }
+
+// recordSensorData 是一个通用方法，用于将传感器数据存入数据库。
+func (c *ChirpStackListener) recordSensorData(devEui string, eventTime time.Time, dataType models.SensorDataType, data interface{}) {
+	// 1. 查找设备
+	device, err := c.deviceRepo.FindByDevEui(devEui)
+	if err != nil {
+		c.logger.Warnf("记录传感器数据失败：无法通过 DevEui '%s' 找到设备: %v", devEui, err)
+		return
+	}
+
+	// 2. 序列化数据结构体为 JSON
+	jsonData, err := json.Marshal(data)
+	if err != nil {
+		c.logger.Errorf("记录传感器数据失败：序列化数据为 JSON 时出错: %v", err)
+		return
+	}
+
+	// 3. 构建 SensorData 模型
+	sensorData := &models.SensorData{
+		Time:                 eventTime,
+		DeviceID:             device.ID,
+		RegionalControllerID: *device.ParentID,
+		SensorDataType:       dataType, // 设置传感器数据类型
+		Data:                 datatypes.JSON(jsonData),
+	}
+
+	// 4. 调用仓库创建记录
+	if err := c.sensorDataRepo.Create(sensorData); err != nil {
+		c.logger.Errorf("记录传感器数据失败：存入数据库时出错: %v", err)
+	}
+}