From ee8039b301d4d086d6760db4cc4152da166d499a Mon Sep 17 00:00:00 2001 From: huang <1724659546@qq.com> Date: Mon, 29 Sep 2025 18:13:19 +0800 Subject: [PATCH] =?UTF-8?q?=E8=B0=83=E6=95=B4GeneralDeviceService?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../service/device/general_device_service.go | 147 ++++++++---------- internal/infra/models/device.go | 5 +- 2 files changed, 70 insertions(+), 82 deletions(-) diff --git a/internal/app/service/device/general_device_service.go b/internal/app/service/device/general_device_service.go index 964d207..668964c 100644 --- a/internal/app/service/device/general_device_service.go +++ b/internal/app/service/device/general_device_service.go @@ -3,7 +3,6 @@ package device import ( "errors" "fmt" - "strconv" "time" "git.huangwc.com/pig/pig-farm-controller/internal/app/service/device/proto" @@ -42,34 +41,43 @@ func NewGeneralDeviceService( } func (g *GeneralDeviceService) Switch(device *models.Device, action DeviceAction) error { - - // 校验设备参数及生成指令 - if *device.ParentID == 0 { - return fmt.Errorf("设备 %v(id=%v) 的上级区域主控(id=%v) ID不合理, 无法执行指令", device.Name, device.ID, *device.ParentID) + // 1. 依赖模型自身的 SelfCheck 进行全面校验 + if err := device.SelfCheck(); err != nil { + return fmt.Errorf("设备 %v(id=%v) 未通过自检: %w", device.Name, device.ID, err) + } + if err := device.DeviceTemplate.SelfCheck(); err != nil { + return fmt.Errorf("设备 %v(id=%v) 的模板未通过自检: %w", device.Name, device.ID, err) } - if !device.SelfCheck() { - return fmt.Errorf("设备 %v(id=%v) 缺少必要信息, 无法发送指令", device.Name, device.ID) + // 2. 检查预加载的 AreaController 是否有效 + areaController := &device.AreaController + if err := areaController.SelfCheck(); err != nil { + return fmt.Errorf("区域主控 %v(id=%v) 未通过自检: %w", areaController.Name, areaController.ID, err) } - deviceInfo := make(map[string]interface{}) - if err := device.ParseProperties(&deviceInfo); err != nil { - return fmt.Errorf("解析设备 %v(id=%v) 配置失败: %v", device.Name, device.ID, err) + // 3. 使用模型层预定义的 BusProperties 结构体解析设备属性 + var deviceProps models.BusProperties + if err := device.ParseProperties(&deviceProps); err != nil { + return fmt.Errorf("解析设备 %v(id=%v) 的属性失败: %w", device.Name, device.ID, err) } - // 已通过 SelfCheck 保证其为纯数字，此处仅进行类型转换 - busNumber, _ := strconv.Atoi(fmt.Sprintf("%v", deviceInfo[models.BusNumber])) - busAddress, _ := strconv.Atoi(fmt.Sprintf("%v", deviceInfo[models.BusAddress])) - relayChannel, _ := strconv.Atoi(fmt.Sprintf("%v", deviceInfo[models.RelayChannel])) + var command models.SwitchCommands + // 前面的 device.DeviceTemplate.SelfCheck()保障了解析一定成功 + _ = device.DeviceTemplate.ParseCommands(&command) + deviceAction := command.On + if action == DeviceActionStop { + deviceAction = command.Off + } + // 4. 构建 Protobuf 指令 data, err := anypb.New(&proto.Switch{ - DeviceAction: string(action), - BusNumber: int32(busNumber), - BusAddress: int32(busAddress), - RelayChannel: int32(relayChannel), + DeviceAction: deviceAction, + BusNumber: int32(deviceProps.BusNumber), + BusAddress: int32(deviceProps.BusAddress), + RelayChannel: int32(deviceProps.RelayChannel), }) if err != nil { - return fmt.Errorf("创建指令失败: %v", err) + return fmt.Errorf("创建指令失败: %w", err) } instruction := &proto.Instruction{ @@ -77,100 +85,83 @@ func (g *GeneralDeviceService) Switch(device *models.Device, action DeviceAction Data: data, } - // 获取自身LoRa设备ID, 因为可能变更, 所以每次都现获取 - thisDevice, err := g.deviceRepo.FindByID(*device.ParentID) - if err != nil { - return fmt.Errorf("获取区域主控(id=%v)信息失败: %v", *device.ParentID, err) - } - if !thisDevice.SelfCheck() { - return fmt.Errorf("区域主控 %v(id=%v) 缺少必要信息, 无法发送指令", thisDevice.Name, thisDevice.ID) - } - thisDeviceinfo := make(map[string]interface{}) - if err := thisDevice.ParseProperties(&thisDeviceinfo); err != nil { - return fmt.Errorf("解析区域主控 %v(id=%v) 配置失败: %v", device.Name, device.ID, err) - } - loraAddress := fmt.Sprintf("%v", thisDeviceinfo[models.LoRaAddress]) - - // 生成消息 message, err := gproto.Marshal(instruction) if err != nil { - return fmt.Errorf("序列化指令失败: %v", err) + return fmt.Errorf("序列化指令失败: %w", err) } - // 发送指令并获取 SendResult - sendResult, err := g.comm.Send(loraAddress, message) + // 5. 发送指令 + networkID := areaController.NetworkID + sendResult, err := g.comm.Send(networkID, message) if err != nil { - // 发送失败，直接返回错误 - return fmt.Errorf("发送指令到设备 %s 失败: %w", loraAddress, err) + return fmt.Errorf("发送指令到 %s 失败: %w", networkID, err) } - // 创建并保存命令日志 + // 6. 创建并保存命令日志 logRecord := &models.DeviceCommandLog{ MessageID: sendResult.MessageID, - DeviceID: thisDevice.ID, // thisDevice 是我们查出来的区域主控 + DeviceID: areaController.ID, SentAt: time.Now(), } - if err := g.deviceCommandLogRepo.Create(logRecord); err != nil { // 记录日志失败是一个需要关注的问题，但可能不应该中断主流程。 // 我们记录一个错误日志，然后成功返回。 g.logger.Errorf("创建指令日志失败 (MessageID: %s): %v", sendResult.MessageID, err) } - g.logger.Infof("成功发送指令到设备 %s 并创建日志 (MessageID: %s)", loraAddress, sendResult.MessageID) + g.logger.Infof("成功发送指令到 %s 并创建日志 (MessageID: %s)", networkID, sendResult.MessageID) return nil } // Collect 实现了 Service 接口，用于发起对指定区域主控下的多个设备的批量采集请求。 -// 它负责查找区域主控、生成关联ID、创建待处理记录、构建指令并最终发送。 func (g *GeneralDeviceService) Collect(regionalControllerID uint, devicesToCollect []*models.Device) error { - if regionalControllerID == 0 { - return errors.New("区域主控ID不能为空") - } - if len(devicesToCollect) == 0 { - // 如果没有要采集的设备，这不是一个错误，只是一个空操作。 g.logger.Info("待采集设备列表为空，无需执行采集任务。") return nil } - // 1. 查找并自检区域主控设备 - regionalController, err := g.deviceRepo.FindByID(regionalControllerID) - if err != nil { - return fmt.Errorf("查找区域主控 (ID: %d) 失败: %w", regionalControllerID, err) + // 1. 从设备列表中获取预加载的区域主控，并进行校验 + regionalController := &devicesToCollect[0].AreaController + if regionalController.ID != regionalControllerID { + return fmt.Errorf("设备列表与指定的区域主控ID (%d) 不匹配", regionalControllerID) } - if !regionalController.SelfCheck() { - return fmt.Errorf("区域主控 (ID: %d) 未通过自检，缺少必要属性", regionalControllerID) + if err := regionalController.SelfCheck(); err != nil { + return fmt.Errorf("区域主控 (ID: %d) 未通过自检: %w", regionalControllerID, err) } - // 2. 准备采集任务列表和数据库存根，并验证设备 + // 2. 准备采集任务列表 var childDeviceIDs []uint var collectTasks []*proto.CollectTask for _, dev := range devicesToCollect { - // 验证设备是否属于指定的区域主控 - if dev.ParentID == nil || *dev.ParentID != regionalControllerID { - return fmt.Errorf("设备 '%s' (ID: %d) 不属于指定的区域主控 (ID: %d)", dev.Name, dev.ID, regionalControllerID) + // 依赖模型自身的 SelfCheck 进行全面校验 + if err := dev.SelfCheck(); err != nil { + g.logger.Warnf("跳过设备 %d，因其未通过自检: %v", dev.ID, err) + continue } - - // 对每个待采集的设备执行自检 - if !dev.SelfCheck() { - g.logger.Warnf("跳过设备 %d，因其未通过自检", dev.ID) + if err := dev.DeviceTemplate.SelfCheck(); err != nil { + g.logger.Warnf("跳过设备 %d，因其设备模板未通过自检: %v", dev.ID, err) continue } - // 自检已通过，我们可以安全地解析属性 - var props map[string]interface{} - // 此时 ParseProperties 不应失败 - _ = dev.ParseProperties(&props) + // 使用模板的 ParseCommands 方法获取指令 + var sensorCmd models.SensorCommands + if err := dev.DeviceTemplate.ParseCommands(&sensorCmd); err != nil { + g.logger.Warnf("跳过设备 %d，因其模板指令无法解析为 SensorCommands: %v", dev.ID, err) + continue + } - busNumber := props[models.BusNumber].(float64) - busAddress := props[models.BusAddress].(float64) + // 使用模型层预定义的 BusProperties 结构体解析设备属性 + var deviceProps models.BusProperties + if err := dev.ParseProperties(&deviceProps); err != nil { + g.logger.Warnf("跳过设备 %d，因其属性解析失败: %v", dev.ID, err) + continue + } collectTasks = append(collectTasks, &proto.CollectTask{ - DeviceAction: dev.Command, - BusNumber: int32(busNumber), - BusAddress: int32(busAddress), + DeviceAction: sensorCmd.Read, // 使用从模板中获取的指令 + BusNumber: int32(deviceProps.BusNumber), + BusAddress: int32(deviceProps.BusAddress), }) childDeviceIDs = append(childDeviceIDs, dev.ID) } @@ -179,11 +170,8 @@ func (g *GeneralDeviceService) Collect(regionalControllerID uint, devicesToColle return errors.New("经过滤后，没有可通过自检的有效设备") } - // 3. 从区域主控的属性中解析出 DevEui (loraAddress) - var rcProps map[string]interface{} - // SelfCheck 已保证属性可解析 - _ = regionalController.ParseProperties(&rcProps) - loraAddress := rcProps[models.LoRaAddress].(string) + // 3. 构建并发送指令 + networkID := regionalController.NetworkID // 4. 创建待处理请求记录 correlationID := uuid.New().String() @@ -220,12 +208,11 @@ func (g *GeneralDeviceService) Collect(regionalControllerID uint, devicesToColle return err } - // 6. 发送指令 - if _, err := g.comm.Send(loraAddress, payload); err != nil { + if _, err := g.comm.Send(networkID, payload); err != nil { g.logger.DPanicf("待采集请求 (CorrelationID: %s) 已创建，但发送到设备失败: %v。数据可能不一致！", correlationID, err) return err } - g.logger.Infof("成功将采集请求 (CorrelationID: %s) 发送到设备 %s", correlationID, loraAddress) + g.logger.Infof("成功将采集请求 (CorrelationID: %s) 发送到设备 %s", correlationID, networkID) return nil } diff --git a/internal/infra/models/device.go b/internal/infra/models/device.go index cc9da17..187f804 100644 --- a/internal/infra/models/device.go +++ b/internal/infra/models/device.go @@ -30,8 +30,9 @@ type LoraProperties struct { // BusProperties 定义了总线设备的特有属性 type BusProperties struct { - BusID int `json:"bus_id"` // 485 总线号 - BusAddress int `json:"bus_address"` // 485 总线地址 + BusNumber int `json:"bus_number"` // 485 总线号 + BusAddress int `json:"bus_address"` // 485 总线地址 + RelayChannel int `json:"relay_channel"` // 继电器通道号 } // AreaController 是一个LoRa转总线(如485)的通信网关