调整GeneralDeviceService

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
 }