From 21fb9c7e57621e558ef53ca155c43c9484d8edf3 Mon Sep 17 00:00:00 2001 From: huang <1724659546@qq.com> Date: Wed, 24 Sep 2025 21:53:18 +0800 Subject: [PATCH] =?UTF-8?q?=E8=AE=B0=E5=BD=95=E4=BF=A1=E5=8F=B7=E5=BC=BA?= =?UTF-8?q?=E5=BA=A6=E6=95=B0=E6=8D=AE?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- internal/app/api/api.go | 2 + internal/app/service/transport/chirp_stack.go | 80 ++++++- .../service/transport/chirp_stack_types.go | 214 +++++++++--------- internal/core/application.go | 7 +- internal/infra/models/SensorData.go | 30 --- internal/infra/models/sensor_data.go | 56 +++++ .../infra/repository/device_repository.go | 13 ++ .../repository/sensor_data_repository.go | 27 +++ 8 files changed, 284 insertions(+), 145 deletions(-) delete mode 100644 internal/infra/models/SensorData.go create mode 100644 internal/infra/models/sensor_data.go create mode 100644 internal/infra/repository/sensor_data_repository.go diff --git a/internal/app/api/api.go b/internal/app/api/api.go index 6d2dbf9..89c48d7 100644 --- a/internal/app/api/api.go +++ b/internal/app/api/api.go @@ -53,6 +53,8 @@ func NewAPI(cfg config.ServerConfig, userRepo repository.UserRepository, deviceRepository repository.DeviceRepository, planRepository repository.PlanRepository, + sensorDataRepository repository.SensorDataRepository, + executionLogRepository repository.ExecutionLogRepository, tokenService token.TokenService, listenHandler transport.ListenHandler, analysisTaskManager *task.AnalysisPlanTaskManager) *API { diff --git a/internal/app/service/transport/chirp_stack.go b/internal/app/service/transport/chirp_stack.go index 60f949b..9cd1831 100644 --- a/internal/app/service/transport/chirp_stack.go +++ b/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) + } +} diff --git a/internal/app/service/transport/chirp_stack_types.go b/internal/app/service/transport/chirp_stack_types.go index defc48a..a8b5bdd 100644 --- a/internal/app/service/transport/chirp_stack_types.go +++ b/internal/app/service/transport/chirp_stack_types.go @@ -10,189 +10,189 @@ import ( // DeviceInfo 包含了所有事件中通用的设备信息。 // 基于 aiserver.proto v4 (integration) type DeviceInfo struct { - TenantID string `json:"tenantId"` - TenantName string `json:"tenantName"` - ApplicationID string `json:"applicationId"` - ApplicationName string `json:"applicationName"` - DeviceProfileID string `json:"deviceProfileId"` - DeviceProfileName string `json:"deviceProfileName"` - DeviceName string `json:"deviceName"` - DevEui string `json:"devEui"` - DeviceClassEnabled string `json:"deviceClassEnabled,omitempty"` // Class A, B, or C - Tags map[string]string `json:"tags"` + TenantID string `json:"tenant_id"` // 租户ID + TenantName string `json:"tenant_name"` // 租户名称 + ApplicationID string `json:"application_id"` // 应用ID + ApplicationName string `json:"application_name"` // 应用名称 + DeviceProfileID string `json:"device_profile_id"` // 设备配置文件ID + DeviceProfileName string `json:"device_profile_name"` // 设备配置文件名称 + DeviceName string `json:"device_name"` // 设备名称 + DevEui string `json:"dev_eui"` // 设备EUI (十六进制编码) + DeviceClassEnabled string `json:"device_class_enabled,omitempty"` // 设备启用的LoRaWAN类别 (A, B, 或 C) + Tags map[string]string `json:"tags"` // 用户定义的标签 } // Location 包含了地理位置信息。 type Location struct { - Latitude float64 `json:"latitude"` - Longitude float64 `json:"longitude"` - Altitude float64 `json:"altitude"` + Latitude float64 `json:"latitude"` // 纬度 + Longitude float64 `json:"longitude"` // 经度 + Altitude float64 `json:"altitude"` // 海拔 } // --- 可复用的子结构体 --- // UplinkRelayRxInfo 包含了上行中继接收信息。 type UplinkRelayRxInfo struct { - DevEui string `json:"devEui"` - Frequency uint32 `json:"frequency"` - Dr uint32 `json:"dr"` - Snr int32 `json:"snr"` - Rssi int32 `json:"rssi"` - WorChannel uint32 `json:"worChannel"` + DevEui string `json:"dev_eui"` // 中继设备的DevEUI + Frequency uint32 `json:"frequency"` // 接收频率 + Dr uint32 `json:"dr"` // 数据速率 + Snr int32 `json:"snr"` // 信噪比 + Rssi int32 `json:"rssi"` // 接收信号强度指示 + WorChannel uint32 `json:"wor_channel"` // Work-on-Relay 通道 } // KeyEnvelope 包装了一个加密的密钥。 // 基于 common.proto type KeyEnvelope struct { - KEKLabel string `json:"kekLabel,omitempty"` - AESKey string `json:"aesKey,omitempty"` // Base64 编码的加密密钥 + KEKLabel string `json:"kek_label,omitempty"` // 密钥加密密钥 (KEK) 标签 + AESKey string `json:"aes_key,omitempty"` // Base64 编码的加密密钥 } // JoinServerContext 包含了 Join-Server 上下文。 // 基于 common.proto type JoinServerContext struct { - SessionKeyID string `json:"sessionKeyId"` - AppSKey *KeyEnvelope `json:"appSKey,omitempty"` + SessionKeyID string `json:"session_key_id"` // 会话密钥ID + AppSKey *KeyEnvelope `json:"app_s_key,omitempty"` // 应用会话密钥 } // UplinkRxInfo 包含了上行接收信息。 type UplinkRxInfo struct { - GatewayID string `json:"gatewayId"` - UplinkID uint32 `json:"uplinkId"` - Time time.Time `json:"time"` - Rssi int `json:"rssi"` - Snr float64 `json:"snr"` - Channel int `json:"channel"` - Location *Location `json:"location"` - Context string `json:"context"` - Metadata map[string]string `json:"metadata"` + GatewayID string `json:"gateway_id"` // 接收到上行数据的网关ID + UplinkID uint32 `json:"uplink_id"` // 上行ID + Time time.Time `json:"time"` // 接收时间 + Rssi int `json:"rssi"` // 接收信号强度指示 + Snr float64 `json:"snr"` // 信噪比 + Channel int `json:"channel"` // 接收通道 + Location *Location `json:"location"` // 网关位置 + Context string `json:"context"` // 上下文信息 + Metadata map[string]string `json:"metadata"` // 元数据 } // LoraModulationInfo 包含了 LoRa 调制的具体参数。 type LoraModulationInfo struct { - Bandwidth int `json:"bandwidth"` - SpreadingFactor int `json:"spreadingFactor"` - CodeRate string `json:"codeRate"` - Polarization bool `json:"polarizationInvert,omitempty"` + Bandwidth int `json:"bandwidth"` // 带宽 + SpreadingFactor int `json:"spreading_factor"` // 扩频因子 + CodeRate string `json:"code_rate"` // 编码率 + Polarization bool `json:"polarization_invert,omitempty"` // 极化反转 } // Modulation 包含了具体的调制信息。 type Modulation struct { - Lora LoraModulationInfo `json:"lora"` + Lora LoraModulationInfo `json:"lora"` // LoRa 调制信息 } // UplinkTxInfo 包含了上行发送信息。 type UplinkTxInfo struct { - Frequency int `json:"frequency"` - Modulation Modulation `json:"modulation"` + Frequency int `json:"frequency"` // 发送频率 + Modulation Modulation `json:"modulation"` // 调制信息 } // DownlinkTxInfo 包含了下行发送信息。 type DownlinkTxInfo struct { - Frequency int `json:"frequency"` - Power int `json:"power"` - Modulation Modulation `json:"modulation"` + Frequency int `json:"frequency"` // 发送频率 + Power int `json:"power"` // 发送功率 + Modulation Modulation `json:"modulation"` // 调制信息 } // ResolvedLocation 包含了地理位置解析结果。 type ResolvedLocation struct { - Latitude float64 `json:"latitude"` - Longitude float64 `json:"longitude"` - Altitude float64 `json:"altitude"` - Source string `json:"source"` - Accuracy int `json:"accuracy"` + Latitude float64 `json:"latitude"` // 纬度 + Longitude float64 `json:"longitude"` // 经度 + Altitude float64 `json:"altitude"` // 海拔 + Source string `json:"source"` // 位置来源 + Accuracy int `json:"accuracy"` // 精度 } // --- 事件专属结构体 --- // UpEvent 对应 ChirpStack 的 "up" 事件。 type UpEvent struct { - DeduplicationID string `json:"deduplicationId"` - Time time.Time `json:"time"` - DeviceInfo DeviceInfo `json:"deviceInfo"` - DevAddr string `json:"devAddr"` - ADR bool `json:"adr"` - DR int `json:"dr"` - FCnt uint32 `json:"fCnt"` - FPort uint8 `json:"fPort"` - Confirmed bool `json:"confirmed"` - Data string `json:"data"` - Object json.RawMessage `json:"object"` - RxInfo []UplinkRxInfo `json:"rxInfo"` - TxInfo UplinkTxInfo `json:"txInfo"` - RelayRxInfo *UplinkRelayRxInfo `json:"relayRxInfo,omitempty"` - JoinServerContext *JoinServerContext `json:"joinServerContext,omitempty"` - RegionConfigID string `json:"regionConfigId,omitempty"` + DeduplicationID string `json:"deduplication_id"` // 去重ID + Time time.Time `json:"time"` // 事件时间 + DeviceInfo DeviceInfo `json:"device_info"` // 设备信息 + DevAddr string `json:"dev_addr"` // 设备地址 + ADR bool `json:"adr"` // 自适应数据速率 (ADR) 是否启用 + DR int `json:"dr"` // 数据速率 + FCnt uint32 `json:"f_cnt"` // 帧计数器 + FPort uint8 `json:"f_port"` // 端口 + Confirmed bool `json:"confirmed"` // 是否是确认帧 + Data string `json:"data"` // Base64 编码的原始负载数据 + Object json.RawMessage `json:"object"` // 解码后的JSON对象负载 + RxInfo []UplinkRxInfo `json:"rx_info"` // 接收信息列表 + TxInfo UplinkTxInfo `json:"tx_info"` // 发送信息 + RelayRxInfo *UplinkRelayRxInfo `json:"relay_rx_info,omitempty"` // 中继接收信息 + JoinServerContext *JoinServerContext `json:"join_server_context,omitempty"` // Join-Server 上下文 + RegionConfigID string `json:"region_config_id,omitempty"` // 区域配置ID } // JoinEvent 对应 ChirpStack 的 "join" 事件。 type JoinEvent struct { - DeduplicationID string `json:"deduplicationId"` - Time time.Time `json:"time"` - DeviceInfo DeviceInfo `json:"deviceInfo"` - DevAddr string `json:"devAddr"` - RelayRxInfo *UplinkRelayRxInfo `json:"relayRxInfo,omitempty"` - JoinServerContext *JoinServerContext `json:"joinServerContext,omitempty"` - RegionConfigID string `json:"regionConfigId,omitempty"` + DeduplicationID string `json:"deduplication_id"` // 去重ID + Time time.Time `json:"time"` // 事件时间 + DeviceInfo DeviceInfo `json:"device_info"` // 设备信息 + DevAddr string `json:"dev_addr"` // 设备地址 + RelayRxInfo *UplinkRelayRxInfo `json:"relay_rx_info,omitempty"` // 中继接收信息 + JoinServerContext *JoinServerContext `json:"join_server_context,omitempty"` // Join-Server 上下文 + RegionConfigID string `json:"region_config_id,omitempty"` // 区域配置ID } // AckEvent 对应 ChirpStack 的 "ack" 事件。 type AckEvent struct { - DeduplicationID string `json:"deduplicationId"` - Time time.Time `json:"time"` - DeviceInfo DeviceInfo `json:"deviceInfo"` - Acknowledged bool `json:"acknowledged"` - FCntDown uint32 `json:"fCntDown"` - QueueItemID string `json:"queueItemId"` + DeduplicationID string `json:"deduplication_id"` // 去重ID + Time time.Time `json:"time"` // 事件时间 + DeviceInfo DeviceInfo `json:"device_info"` // 设备信息 + Acknowledged bool `json:"acknowledged"` // 是否已确认 + FCntDown uint32 `json:"f_cnt_down"` // 下行帧计数器 + QueueItemID string `json:"queue_item_id"` // 队列项ID } // TxAckEvent 对应 ChirpStack 的 "txack" 事件。 type TxAckEvent struct { - DownlinkID uint32 `json:"downlinkId"` // 修改: 替换 DeduplicationID - Time time.Time `json:"time"` - DeviceInfo DeviceInfo `json:"deviceInfo"` - FCntDown uint32 `json:"fCntDown"` - GatewayID string `json:"gatewayId"` - QueueItemID string `json:"queueItemId"` - TxInfo DownlinkTxInfo `json:"txInfo"` + DownlinkID uint32 `json:"downlink_id"` // 下行ID + Time time.Time `json:"time"` // 事件时间 + DeviceInfo DeviceInfo `json:"device_info"` // 设备信息 + FCntDown uint32 `json:"f_cnt_down"` // 下行帧计数器 + GatewayID string `json:"gateway_id"` // 网关ID + QueueItemID string `json:"queue_item_id"` // 队列项ID + TxInfo DownlinkTxInfo `json:"tx_info"` // 下行发送信息 } // StatusEvent 对应 ChirpStack 的 "status" 事件。 type StatusEvent struct { - DeduplicationID string `json:"deduplicationId"` - Time time.Time `json:"time"` - DeviceInfo DeviceInfo `json:"deviceInfo"` - Margin int `json:"margin"` - ExternalPower bool `json:"externalPowerSource"` - BatteryLevel float32 `json:"batteryLevel"` - BatteryLevelUnavailable bool `json:"batteryLevelUnavailable"` + DeduplicationID string `json:"deduplication_id"` // 去重ID + Time time.Time `json:"time"` // 事件时间 + DeviceInfo DeviceInfo `json:"device_info"` // 设备信息 + Margin int `json:"margin"` // 链路预算余量 (dB) + ExternalPower bool `json:"external_power_source"` // 设备是否连接外部电源 + BatteryLevel float32 `json:"battery_level"` // 电池剩余电量 + BatteryLevelUnavailable bool `json:"battery_level_unavailable"` // 电池电量是否不可用 } // LogEvent 对应 ChirpStack 的 "log" 事件。 type LogEvent struct { - DeduplicationID string `json:"deduplicationId"` - Time time.Time `json:"time"` - DeviceInfo DeviceInfo `json:"deviceInfo"` - Level string `json:"level"` - Code string `json:"code"` - Description string `json:"description"` - Context map[string]string `json:"context"` + DeduplicationID string `json:"deduplication_id"` // 去重ID + Time time.Time `json:"time"` // 事件时间 + DeviceInfo DeviceInfo `json:"device_info"` // 设备信息 + Level string `json:"level"` // 日志级别 (e.g., INFO, WARNING, ERROR) + Code string `json:"code"` // 日志代码 + Description string `json:"description"` // 日志描述 + Context map[string]string `json:"context"` // 上下文信息 } // LocationEvent 对应 ChirpStack 的 "location" 事件。 type LocationEvent struct { - DeduplicationID string `json:"deduplicationId"` - Time time.Time `json:"time"` - DeviceInfo DeviceInfo `json:"deviceInfo"` - Location ResolvedLocation `json:"location"` + DeduplicationID string `json:"deduplication_id"` // 去重ID + Time time.Time `json:"time"` // 事件时间 + DeviceInfo DeviceInfo `json:"device_info"` // 设备信息 + Location ResolvedLocation `json:"location"` // 解析后的位置信息 } // IntegrationEvent 对应 ChirpStack 的 "integration" 事件。 type IntegrationEvent struct { - DeduplicationID string `json:"deduplicationId"` - Time time.Time `json:"time"` - DeviceInfo DeviceInfo `json:"deviceInfo"` - IntegrationName string `json:"integrationName"` - EventType string `json:"eventType,omitempty"` - Object json.RawMessage `json:"object"` + DeduplicationID string `json:"deduplication_id"` // 去重ID + Time time.Time `json:"time"` // 事件时间 + DeviceInfo DeviceInfo `json:"device_info"` // 设备信息 + IntegrationName string `json:"integration_name"` // 集成名称 + EventType string `json:"event_type,omitempty"` // 事件类型 + Object json.RawMessage `json:"object"` // 集成事件的原始JSON负载 } diff --git a/internal/core/application.go b/internal/core/application.go index a446e11..80ee2ae 100644 --- a/internal/core/application.go +++ b/internal/core/application.go @@ -69,8 +69,11 @@ func NewApplication(configPath string) (*Application, error) { // 初始化执行日志仓库 executionLogRepo := repository.NewGormExecutionLogRepository(storage.GetDB()) + // 初始化传感器数据仓库 + sensorDataRepo := repository.NewGormSensorDataRepository(storage.GetDB()) + // 初始化设备上行监听器 - listenHandler := transport.NewChirpStackListener(logger) + listenHandler := transport.NewChirpStackListener(logger, sensorDataRepo, deviceRepo) // 初始化计划触发器管理器 analysisPlanTaskManager := task.NewAnalysisPlanTaskManager(planRepo, pendingTaskRepo, executionLogRepo, logger) @@ -79,7 +82,7 @@ func NewApplication(configPath string) (*Application, error) { executor := task.NewScheduler(pendingTaskRepo, executionLogRepo, planRepo, analysisPlanTaskManager, task.TaskFactory, logger, time.Duration(cfg.Task.Interval)*time.Second, cfg.Task.NumWorkers) // 初始化 API 服务器 - apiServer := api.NewAPI(cfg.Server, logger, userRepo, deviceRepo, planRepo, tokenService, listenHandler, analysisPlanTaskManager) + apiServer := api.NewAPI(cfg.Server, logger, userRepo, deviceRepo, planRepo, sensorDataRepo, executionLogRepo, tokenService, listenHandler, analysisPlanTaskManager) // 组装 Application 对象 app := &Application{ diff --git a/internal/infra/models/SensorData.go b/internal/infra/models/SensorData.go deleted file mode 100644 index 0697553..0000000 --- a/internal/infra/models/SensorData.go +++ /dev/null @@ -1,30 +0,0 @@ -package models - -import ( - "time" - - "gorm.io/datatypes" -) - -// SensorData 存储所有类型的传感器数据，对应数据库中的 'sensor_data' 表。 -type SensorData struct { - // Time 是数据记录的时间戳，作为复合主键的一部分。 - // GORM 会将其映射到 'time' TIMESTAMPTZ 列。 - Time time.Time `gorm:"primaryKey" json:"time"` - - // DeviceID 是传感器的唯一标识符，作为复合主键的另一部分。 - // GORM 会将其映射到 'device_id' VARCHAR(50) 列。 - DeviceID uint `gorm:"primaryKey" json:"device_id"` - - // RegionalControllerID 是上报此数据的区域主控的ID。 - // 我们为其添加了数据库索引以优化按区域查询的性能。 - RegionalControllerID uint `json:"regional_controller_id"` - - // Data 存储一个或多个传感器读数，格式为 JSON。 - // GORM 会使用 'jsonb' 类型来创建此列。 - Data datatypes.JSON `gorm:"type:jsonb" json:"data"` -} - -func (SensorData) TableName() string { - return "sensor_data" -} diff --git a/internal/infra/models/sensor_data.go b/internal/infra/models/sensor_data.go new file mode 100644 index 0000000..93898ed --- /dev/null +++ b/internal/infra/models/sensor_data.go @@ -0,0 +1,56 @@ +package models + +import ( + "time" + + "gorm.io/datatypes" +) + +// SensorDataType 定义了 SensorData 记录中 Data 字段的整体类型 +type SensorDataType string + +const ( + SensorDataTypeSignalMetrics SensorDataType = "signal_metrics" // 信号强度 + SensorDataTypeBatteryLevel SensorDataType = "battery_level" // 电池电量 +) + +// SignalMetrics 存储信号强度数据 +type SignalMetrics struct { + RSSI int `json:"rssi"` // 绝对信号强度（dBm），受距离、障碍物影响 + SNR float64 `json:"snr"` // 信号与噪声的相对比率（dB），由 RSSI 减去噪声地板（Noise Floor） + Sensitivity int `json:"sensitivity"` // 网关的最低检测阈值(dBm) + Margin int `json:"margin"` // SNR 相对于接收器灵敏度的余量, Margin = SNR - Sensitivity +} + +// BatteryLevel 存储电池电量数据 +type BatteryLevel struct { + BatteryLevel float32 `json:"battery_level"` // 电量剩余百分比 + BatteryLevelUnavailable bool `json:"battery_level_unavailable"` // 电量数据不可用 + ExternalPower bool `json:"external_power"` // 是否使用外部电源 +} + +// SensorData 存储所有类型的传感器数据，对应数据库中的 'sensor_data' 表。 +type SensorData struct { + // Time 是数据记录的时间戳，作为复合主键的一部分。 + // GORM 会将其映射到 'time' TIMESTAMPTZ 列。 + Time time.Time `gorm:"primaryKey" json:"time"` + + // DeviceID 是传感器的唯一标识符，作为复合主键的另一部分。 + // GORM 会将其映射到 'device_id' VARCHAR(50) 列。 + DeviceID uint `gorm:"primaryKey" json:"device_id"` + + // RegionalControllerID 是上报此数据的区域主控的ID。 + // 我们为其添加了数据库索引以优化按区域查询的性能。 + RegionalControllerID uint `json:"regional_controller_id"` + + // SensorDataType 是传感数据的类型 + SensorDataType SensorDataType `gorm:"not null;index" json:"sensor_data_type"` + + // Data 存储一个或多个传感器读数，格式为 JSON。 + // GORM 会使用 'jsonb' 类型来创建此列。 + Data datatypes.JSON `gorm:"type:jsonb" json:"data"` +} + +func (SensorData) TableName() string { + return "sensor_data" +} diff --git a/internal/infra/repository/device_repository.go b/internal/infra/repository/device_repository.go index b5554b3..5cc3ee8 100644 --- a/internal/infra/repository/device_repository.go +++ b/internal/infra/repository/device_repository.go @@ -32,6 +32,9 @@ type DeviceRepository interface { // Delete 根据主键 ID 删除一个设备 Delete(id uint) error + + // FindByDevEui 根据 DevEui (存储在 properties JSONB 中的 lora_address) 查找设备 (新增) + FindByDevEui(devEui string) (*models.Device, error) } // gormDeviceRepository 是 DeviceRepository 的 GORM 实现 @@ -108,3 +111,13 @@ func (r *gormDeviceRepository) Update(device *models.Device) error { func (r *gormDeviceRepository) Delete(id uint) error { return r.db.Delete(&models.Device{}, id).Error } + +// FindByDevEui 根据 DevEui (存储在 properties JSONB 中的 lora_address) 查找设备 +func (r *gormDeviceRepository) FindByDevEui(devEui string) (*models.Device, error) { + var device models.Device + // 使用 GORM 的 JSONB 查询语法: properties->>'lora_address' + if err := r.db.Where("properties->>'lora_address' = ?", devEui).First(&device).Error; err != nil { + return nil, err // 如果找不到或发生其他错误，GORM 会返回错误 + } + return &device, nil +} diff --git a/internal/infra/repository/sensor_data_repository.go b/internal/infra/repository/sensor_data_repository.go new file mode 100644 index 0000000..b9b80a9 --- /dev/null +++ b/internal/infra/repository/sensor_data_repository.go @@ -0,0 +1,27 @@ +package repository + +import ( + "git.huangwc.com/pig/pig-farm-controller/internal/infra/models" + "gorm.io/gorm" +) + +// SensorDataRepository 定义了与传感器数据相关的数据库操作接口。 +type SensorDataRepository interface { + Create(sensorData *models.SensorData) error +} + +// gormSensorDataRepository 是 SensorDataRepository 的 GORM 实现。 +type gormSensorDataRepository struct { + db *gorm.DB +} + +// NewGormSensorDataRepository 创建一个新的 SensorDataRepository GORM 实现实例。 +// 它直接接收一个 *gorm.DB 实例作为依赖，完全遵循项目中的既定模式。 +func NewGormSensorDataRepository(db *gorm.DB) SensorDataRepository { + return &gormSensorDataRepository{db: db} +} + +// Create 将一条新的传感器数据记录插入数据库。 +func (r *gormSensorDataRepository) Create(sensorData *models.SensorData) error { + return r.db.Create(sensorData).Error +}