431 lines
17 KiB
Go
431 lines
17 KiB
Go
package transport
|
||
|
||
import (
|
||
"encoding/base64"
|
||
"encoding/json"
|
||
"io"
|
||
"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"
|
||
)
|
||
|
||
// ChirpStackListener 主动发送的请求的event字段, 这个字段代表事件类型
|
||
const (
|
||
eventTypeUp = "up" // 上行数据事件：当接收到设备发送的数据时触发，这是最核心的事件。
|
||
eventTypeStatus = "status" // 设备状态事件：当设备报告其状态时触发（例如电池电量、信号强度）。
|
||
eventTypeJoin = "join" // 入网事件：当设备成功加入网络时触发。
|
||
eventTypeAck = "ack" // 下行确认事件：当设备确认收到下行消息时触发。
|
||
eventTypeTxAck = "txack" // 网关发送确认事件：当网关确认已发送下行消息时触发（不代表设备已收到）。
|
||
eventTypeLog = "log" // 日志事件：当设备或 ChirpStack 产生日志信息时触发。
|
||
eventTypeLocation = "location" // 位置事件：当设备的位置被解析或更新时触发。
|
||
eventTypeIntegration = "integration" // 集成事件：当其他集成（如第三方服务）处理数据后触发。
|
||
)
|
||
|
||
// ChirpStackListener 是一个监听器, 用于监听ChirpStack反馈的设备上行事件
|
||
type ChirpStackListener struct {
|
||
logger *logs.Logger
|
||
sensorDataRepo repository.SensorDataRepository
|
||
deviceRepo repository.DeviceRepository
|
||
areaControllerRepo repository.AreaControllerRepository
|
||
deviceCommandLogRepo repository.DeviceCommandLogRepository
|
||
pendingCollectionRepo repository.PendingCollectionRepository
|
||
}
|
||
|
||
// NewChirpStackListener 创建一个新的 ChirpStackListener 实例
|
||
func NewChirpStackListener(
|
||
logger *logs.Logger,
|
||
sensorDataRepo repository.SensorDataRepository,
|
||
deviceRepo repository.DeviceRepository,
|
||
areaControllerRepo repository.AreaControllerRepository,
|
||
deviceCommandLogRepo repository.DeviceCommandLogRepository,
|
||
pendingCollectionRepo repository.PendingCollectionRepository,
|
||
) ListenHandler { // 返回接口类型
|
||
return &ChirpStackListener{
|
||
logger: logger,
|
||
sensorDataRepo: sensorDataRepo,
|
||
deviceRepo: deviceRepo,
|
||
areaControllerRepo: areaControllerRepo,
|
||
deviceCommandLogRepo: deviceCommandLogRepo,
|
||
pendingCollectionRepo: pendingCollectionRepo,
|
||
}
|
||
}
|
||
|
||
// Handler 监听ChirpStack反馈的事件, 因为这是个Webhook, 所以直接回复掉再慢慢处理信息
|
||
func (c *ChirpStackListener) Handler() http.HandlerFunc {
|
||
return func(w http.ResponseWriter, r *http.Request) {
|
||
defer r.Body.Close()
|
||
|
||
b, err := io.ReadAll(r.Body)
|
||
if err != nil {
|
||
c.logger.Errorf("读取请求体失败: %v", err)
|
||
http.Error(w, "failed to read body", http.StatusBadRequest)
|
||
return
|
||
}
|
||
|
||
event := r.URL.Query().Get("event")
|
||
|
||
w.WriteHeader(http.StatusOK)
|
||
|
||
// 将异步处理逻辑委托给 handler 方法
|
||
go c.handler(b, event)
|
||
}
|
||
}
|
||
|
||
// handler 用于处理 ChirpStack 发送的事件
|
||
func (c *ChirpStackListener) handler(data []byte, eventType string) {
|
||
switch eventType {
|
||
case eventTypeUp:
|
||
var msg UpEvent
|
||
if err := json.Unmarshal(data, &msg); err != nil {
|
||
c.logger.Errorf("解析 'up' 事件失败: %v, data: %s", err, string(data))
|
||
return
|
||
}
|
||
c.handleUpEvent(&msg)
|
||
|
||
case eventTypeJoin:
|
||
var msg JoinEvent
|
||
if err := json.Unmarshal(data, &msg); err != nil {
|
||
c.logger.Errorf("解析 'join' 事件失败: %v, data: %s", err, string(data))
|
||
return
|
||
}
|
||
c.handleJoinEvent(&msg)
|
||
|
||
case eventTypeAck:
|
||
var msg AckEvent
|
||
if err := json.Unmarshal(data, &msg); err != nil {
|
||
c.logger.Errorf("解析 'ack' 事件失败: %v, data: %s", err, string(data))
|
||
return
|
||
}
|
||
c.handleAckEvent(&msg)
|
||
|
||
case eventTypeTxAck:
|
||
var msg TxAckEvent
|
||
if err := json.Unmarshal(data, &msg); err != nil {
|
||
c.logger.Errorf("解析 'txack' 事件失败: %v, data: %s", err, string(data))
|
||
return
|
||
}
|
||
c.handleTxAckEvent(&msg)
|
||
|
||
case eventTypeStatus:
|
||
var msg StatusEvent
|
||
if err := json.Unmarshal(data, &msg); err != nil {
|
||
c.logger.Errorf("解析 'status' 事件失败: %v, data: %s", err, string(data))
|
||
return
|
||
}
|
||
c.handleStatusEvent(&msg)
|
||
|
||
case eventTypeLog:
|
||
var msg LogEvent
|
||
if err := json.Unmarshal(data, &msg); err != nil {
|
||
c.logger.Errorf("解析 'log' 事件失败: %v, data: %s", err, string(data))
|
||
return
|
||
}
|
||
c.handleLogEvent(&msg)
|
||
|
||
case eventTypeLocation:
|
||
var msg LocationEvent
|
||
if err := json.Unmarshal(data, &msg); err != nil {
|
||
c.logger.Errorf("解析 'location' 事件失败: %v, data: %s", err, string(data))
|
||
return
|
||
}
|
||
c.handleLocationEvent(&msg)
|
||
|
||
case eventTypeIntegration:
|
||
var msg IntegrationEvent
|
||
if err := json.Unmarshal(data, &msg); err != nil {
|
||
c.logger.Errorf("解析 'integration' 事件失败: %v, data: %s", err, string(data))
|
||
return
|
||
}
|
||
c.handleIntegrationEvent(&msg)
|
||
|
||
default:
|
||
c.logger.Errorf("未知的ChirpStack事件: %s, data: %s", eventType, string(data))
|
||
}
|
||
}
|
||
|
||
// --- 业务处理函数 ---
|
||
|
||
// handleUpEvent 处理上行数据事件
|
||
func (c *ChirpStackListener) handleUpEvent(event *UpEvent) {
|
||
c.logger.Infof("开始处理 'up' 事件, DevEui: %s", event.DeviceInfo.DevEui)
|
||
|
||
// 1. 查找区域主控设备
|
||
regionalController, err := c.areaControllerRepo.FindByNetworkID(event.DeviceInfo.DevEui)
|
||
if err != nil {
|
||
c.logger.Errorf("处理 'up' 事件失败：无法通过 DevEui '%s' 找到区域主控设备: %v", event.DeviceInfo.DevEui, err)
|
||
return
|
||
}
|
||
// 依赖 SelfCheck 确保区域主控有效
|
||
if err := regionalController.SelfCheck(); err != nil {
|
||
c.logger.Errorf("处理 'up' 事件失败：区域主控 %v(ID: %d) 未通过自检: %v", regionalController.Name, regionalController.ID, 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 结构体
|
||
signalMetrics := models.SignalMetrics{
|
||
RssiDbm: rx.Rssi,
|
||
SnrDb: rx.Snr,
|
||
}
|
||
|
||
// 记录信号强度
|
||
c.recordSensorData(regionalController.ID, regionalController.ID, event.Time, models.SensorTypeSignalMetrics, 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)
|
||
}
|
||
|
||
// 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 解析外层 "信封"
|
||
var instruction proto.Instruction
|
||
if err := gproto.Unmarshal(decodedData, &instruction); err != nil {
|
||
c.logger.Errorf("解析上行 Instruction Protobuf 失败: %v, Decoded Data: %x", err, decodedData)
|
||
return
|
||
}
|
||
|
||
// 3.3 检查是否是采集响应
|
||
if instruction.Method != proto.MethodType_COLLECT {
|
||
c.logger.Infof("收到一个非采集响应的上行指令 (Method: %s)，无需处理。", instruction.Method.String())
|
||
return
|
||
}
|
||
|
||
// 3.4 解包内层 CollectResult
|
||
var collectResp proto.CollectResult
|
||
if err := instruction.Data.UnmarshalTo(&collectResp); err != nil {
|
||
c.logger.Errorf("解包数据信息失败: %v", err)
|
||
return
|
||
}
|
||
|
||
correlationID := collectResp.CorrelationId
|
||
c.logger.Infof("成功解析采集响应 (CorrelationID: %s)，包含 %d 个值。", correlationID, len(collectResp.Values))
|
||
|
||
// 4. 根据 CorrelationID 查找待处理请求
|
||
pendingReq, err := c.pendingCollectionRepo.FindByCorrelationID(correlationID)
|
||
if err != nil {
|
||
c.logger.Errorf("处理采集响应失败：无法找到待处理请求 (CorrelationID: %s): %v", correlationID, err)
|
||
return
|
||
}
|
||
|
||
// 检查状态，防止重复处理
|
||
if pendingReq.Status != models.PendingStatusPending && pendingReq.Status != models.PendingStatusTimedOut {
|
||
c.logger.Warnf("收到一个已处理过的采集响应 (CorrelationID: %s, Status: %s)，将忽略。", correlationID, pendingReq.Status)
|
||
return
|
||
}
|
||
|
||
// 5. 匹配数据并存入数据库
|
||
deviceIDs := pendingReq.CommandMetadata
|
||
values := collectResp.Values
|
||
if len(deviceIDs) != len(values) {
|
||
c.logger.Errorf("数据不匹配：下行指令要求采集 %d 个设备，但上行响应包含 %d 个值 (CorrelationID: %s)", len(deviceIDs), len(values), correlationID)
|
||
// 即使数量不匹配，也更新状态为完成，以防止请求永远 pending
|
||
err = c.pendingCollectionRepo.UpdateStatusToFulfilled(correlationID, event.Time)
|
||
if err != nil {
|
||
c.logger.Errorf("处理采集响应失败：无法更新待处理请求 (CorrelationID: %s) 的状态为完成: %v", correlationID, err)
|
||
}
|
||
return
|
||
}
|
||
|
||
for i, deviceID := range deviceIDs {
|
||
rawSensorValue := values[i] // 这是设备上报的原始值
|
||
|
||
// 5.1 获取设备及其模板
|
||
dev, err := c.deviceRepo.FindByID(deviceID)
|
||
if err != nil {
|
||
c.logger.Errorf("处理采集数据失败：无法找到设备 (ID: %d): %v", deviceID, err)
|
||
continue
|
||
}
|
||
// 依赖 SelfCheck 确保设备和模板有效
|
||
if err := dev.SelfCheck(); err != nil {
|
||
c.logger.Warnf("跳过设备 %d，因其未通过自检: %v", dev.ID, err)
|
||
continue
|
||
}
|
||
if err := dev.DeviceTemplate.SelfCheck(); err != nil {
|
||
c.logger.Warnf("跳过设备 %d，因其设备模板未通过自检: %v", dev.ID, err)
|
||
continue
|
||
}
|
||
|
||
// 5.2 从设备模板中解析 ValueDescriptor
|
||
var valueDescriptors []*models.ValueDescriptor
|
||
if err := dev.DeviceTemplate.ParseValues(&valueDescriptors); err != nil {
|
||
c.logger.Warnf("跳过设备 %d，因其设备模板的 Values 属性解析失败: %v", dev.ID, err)
|
||
continue
|
||
}
|
||
// 根据 DeviceTemplate.SelfCheck，这里应该只有一个 ValueDescriptor
|
||
if len(valueDescriptors) == 0 {
|
||
c.logger.Warnf("跳过设备 %d，因其设备模板缺少 ValueDescriptor 定义", dev.ID)
|
||
continue
|
||
}
|
||
valueDescriptor := valueDescriptors[0]
|
||
|
||
// 5.3 应用乘数和偏移量计算最终值
|
||
parsedValue := float64(rawSensorValue)*valueDescriptor.Multiplier + valueDescriptor.Offset
|
||
|
||
// 5.4 根据传感器类型构建具体的数据结构
|
||
var dataToRecord interface{}
|
||
switch valueDescriptor.Type {
|
||
case models.SensorTypeTemperature:
|
||
dataToRecord = models.TemperatureData{TemperatureCelsius: parsedValue}
|
||
case models.SensorTypeHumidity:
|
||
dataToRecord = models.HumidityData{HumidityPercent: parsedValue}
|
||
case models.SensorTypeWeight:
|
||
dataToRecord = models.WeightData{WeightKilograms: parsedValue}
|
||
default:
|
||
// TODO 未知传感器的数据需要记录吗
|
||
c.logger.Warnf("未知的传感器类型 '%s'，将使用通用格式记录", valueDescriptor.Type)
|
||
dataToRecord = map[string]float64{"value": parsedValue}
|
||
}
|
||
|
||
// 5.5 记录传感器数据
|
||
c.recordSensorData(regionalController.ID, dev.ID, event.Time, valueDescriptor.Type, dataToRecord)
|
||
c.logger.Infof("成功记录传感器数据: 设备ID=%d, 类型=%s, 原始值=%f, 解析值=%.2f", dev.ID, valueDescriptor.Type, rawSensorValue, parsedValue)
|
||
}
|
||
|
||
// 6. 更新请求状态为“已完成”
|
||
if err := c.pendingCollectionRepo.UpdateStatusToFulfilled(correlationID, event.Time); err != nil {
|
||
c.logger.Errorf("更新待采集请求状态为 'fulfilled' 失败 (CorrelationID: %s): %v", correlationID, err)
|
||
} else {
|
||
c.logger.Infof("成功完成并关闭采集请求 (CorrelationID: %s)", correlationID)
|
||
}
|
||
}
|
||
|
||
// handleStatusEvent 处理设备状态事件
|
||
func (c *ChirpStackListener) handleStatusEvent(event *StatusEvent) {
|
||
c.logger.Infof("处接收到理 'status' 事件: %+v", event)
|
||
|
||
// 查找区域主控设备
|
||
regionalController, err := c.areaControllerRepo.FindByNetworkID(event.DeviceInfo.DevEui)
|
||
if err != nil {
|
||
c.logger.Errorf("处理 'status' 事件失败：无法通过 DevEui '%s' 找到区域主控设备: %v", event.DeviceInfo.DevEui, err)
|
||
return
|
||
}
|
||
|
||
// 记录信号强度
|
||
signalMetrics := models.SignalMetrics{
|
||
MarginDb: event.Margin,
|
||
}
|
||
c.recordSensorData(regionalController.ID, regionalController.ID, event.Time, models.SensorTypeSignalMetrics, signalMetrics)
|
||
c.logger.Infof("已记录区域主控 (ID: %d) 的信号状态: %+v", regionalController.ID, signalMetrics)
|
||
|
||
// 记录电量
|
||
batteryLevel := models.BatteryLevel{
|
||
BatteryLevelRatio: event.BatteryLevel,
|
||
BatteryLevelUnavailable: event.BatteryLevelUnavailable,
|
||
ExternalPower: event.ExternalPower,
|
||
}
|
||
c.recordSensorData(regionalController.ID, regionalController.ID, event.Time, models.SensorTypeBatteryLevel, batteryLevel)
|
||
c.logger.Infof("已记录区域主控 (ID: %d) 的电池状态: %+v", regionalController.ID, batteryLevel)
|
||
}
|
||
|
||
// handleAckEvent 处理下行确认事件
|
||
func (c *ChirpStackListener) handleAckEvent(event *AckEvent) {
|
||
c.logger.Infof("接收到 'ack' 事件: %+v", event)
|
||
|
||
// 更新下行任务记录的确认时间及接收成功状态
|
||
err := c.deviceCommandLogRepo.UpdateAcknowledgedAt(event.DeduplicationID, event.Time, event.Acknowledged)
|
||
if err != nil {
|
||
c.logger.Errorf("更新下行任务记录的确认时间及接收成功状态失败 (MessageID: %s, DevEui: %s, Acknowledged: %t): %v",
|
||
event.DeduplicationID, event.DeviceInfo.DevEui, event.Acknowledged, err)
|
||
return
|
||
}
|
||
|
||
c.logger.Infof("成功更新下行任务记录确认时间及接收成功状态 (MessageID: %s, DevEui: %s, Acknowledged: %t, AcknowledgedAt: %s)",
|
||
event.DeduplicationID, event.DeviceInfo.DevEui, event.Acknowledged, event.Time.Format(time.RFC3339))
|
||
}
|
||
|
||
// handleLogEvent 处理日志事件
|
||
func (c *ChirpStackListener) handleLogEvent(event *LogEvent) {
|
||
// 首先，打印完整的事件结构体，用于详细排查
|
||
c.logger.Infof("接收到 'log' 事件的完整内容: %+v", event)
|
||
|
||
// 接着，根据 ChirpStack 日志的级别，使用我们自己的 logger 对应级别来打印核心信息
|
||
logMessage := "ChirpStack 日志: [%s] %s (DevEui: %s)"
|
||
switch event.Level {
|
||
case "INFO":
|
||
c.logger.Infof(logMessage, event.Code, event.Description, event.DeviceInfo.DevEui)
|
||
case "WARNING":
|
||
c.logger.Warnf(logMessage, event.Code, event.Description, event.DeviceInfo.DevEui)
|
||
case "ERROR":
|
||
c.logger.Errorf(logMessage, event.Code, event.Description, event.DeviceInfo.DevEui)
|
||
default:
|
||
// 对于未知级别，使用 Warn 级别打印，并明确指出级别未知
|
||
c.logger.Warnf("ChirpStack 日志: [未知级别: %s] %s %s (DevEui: %s)",
|
||
event.Level, event.Code, event.Description, event.DeviceInfo.DevEui)
|
||
}
|
||
}
|
||
|
||
// handleJoinEvent 处理入网事件
|
||
func (c *ChirpStackListener) handleJoinEvent(event *JoinEvent) {
|
||
c.logger.Infof("接收到 'join' 事件: %+v", event)
|
||
// 在这里添加您的业务逻辑
|
||
}
|
||
|
||
// handleTxAckEvent 处理网关发送确认事件
|
||
func (c *ChirpStackListener) handleTxAckEvent(event *TxAckEvent) {
|
||
c.logger.Infof("接收到 'txack' 事件: %+v", event)
|
||
// 在这里添加您的业务逻辑
|
||
}
|
||
|
||
// handleLocationEvent 处理位置事件
|
||
func (c *ChirpStackListener) handleLocationEvent(event *LocationEvent) {
|
||
c.logger.Infof("接收到 'location' 事件: %+v", event)
|
||
// 在这里添加您的业务逻辑
|
||
}
|
||
|
||
// handleIntegrationEvent 处理集成事件
|
||
func (c *ChirpStackListener) handleIntegrationEvent(event *IntegrationEvent) {
|
||
c.logger.Infof("接收到 'integration' 事件: %+v", event)
|
||
// 在这里添加您的业务逻辑
|
||
}
|
||
|
||
// recordSensorData 是一个通用方法，用于将传感器数据存入数据库。
|
||
// regionalControllerID: 区域主控设备的ID
|
||
// sensorDeviceID: 实际产生传感器数据的普通设备的ID
|
||
// sensorType: 传感器值的类型 (例如 models.SensorTypeTemperature)
|
||
// data: 具体的传感器数据结构体实例 (例如 models.TemperatureData)
|
||
func (c *ChirpStackListener) recordSensorData(regionalControllerID uint, sensorDeviceID uint, eventTime time.Time, sensorType models.SensorType, data interface{}) {
|
||
// 1. 将传入的结构体序列化为 JSON
|
||
jsonData, err := json.Marshal(data)
|
||
if err != nil {
|
||
c.logger.Errorf("记录传感器数据失败：序列化数据为 JSON 时出错: %v", err)
|
||
return
|
||
}
|
||
|
||
// 2. 构建 SensorData 模型
|
||
sensorData := &models.SensorData{
|
||
Time: eventTime,
|
||
DeviceID: sensorDeviceID,
|
||
RegionalControllerID: regionalControllerID,
|
||
SensorType: sensorType,
|
||
Data: datatypes.JSON(jsonData),
|
||
}
|
||
|
||
// 3. 调用仓库创建记录
|
||
if err := c.sensorDataRepo.Create(sensorData); err != nil {
|
||
c.logger.Errorf("记录传感器数据失败：存入数据库时出错: %v", err)
|
||
}
|
||
}
|