修改domain包

internal/domain/device/device_service.go

@@ -1,6 +1,10 @@
 package device
 
-import "git.huangwc.com/pig/pig-farm-controller/internal/infra/models"
+import (
+	"context"
+
+	"git.huangwc.com/pig/pig-farm-controller/internal/infra/models"
+)
 
 // 设备行为
 type DeviceAction string
@@ -21,10 +25,10 @@ var (
 type Service interface {
 
 	// Switch 用于切换指定设备的状态, 比如启动和停止
-	Switch(device *models.Device, action DeviceAction) error
+	Switch(ctx context.Context, device *models.Device, action DeviceAction) error
 
 	// Collect 用于发起对指定区域主控下的多个设备的批量采集请求。
-	Collect(regionalControllerID uint, devicesToCollect []*models.Device) error
+	Collect(ctx context.Context, regionalControllerID uint, devicesToCollect []*models.Device) error
 }
 
 // 设备操作指令通用结构(最外层)

internal/domain/device/general_device_service.go

@@ -1,6 +1,7 @@
 package device
 
 import (
+	"context"
 	"errors"
 	"fmt"
 	"time"
@@ -17,31 +18,32 @@ import (
 )
 
 type GeneralDeviceService struct {
+	ctx                   context.Context
 	deviceRepo            repository.DeviceRepository
 	deviceCommandLogRepo  repository.DeviceCommandLogRepository
 	pendingCollectionRepo repository.PendingCollectionRepository
-	logger                *logs.Logger
 	comm                  transport.Communicator
 }
 
 // NewGeneralDeviceService 创建一个通用设备服务
 func NewGeneralDeviceService(
+	ctx context.Context,
 	deviceRepo repository.DeviceRepository,
 	deviceCommandLogRepo repository.DeviceCommandLogRepository,
 	pendingCollectionRepo repository.PendingCollectionRepository,
-	logger *logs.Logger,
 	comm transport.Communicator,
 ) Service {
 	return &GeneralDeviceService{
+		ctx:                   ctx,
 		deviceRepo:            deviceRepo,
 		deviceCommandLogRepo:  deviceCommandLogRepo,
 		pendingCollectionRepo: pendingCollectionRepo,
-		logger:                logger,
 		comm:                  comm,
 	}
 }
 
-func (g *GeneralDeviceService) Switch(device *models.Device, action DeviceAction) error {
+func (g *GeneralDeviceService) Switch(ctx context.Context, device *models.Device, action DeviceAction) error {
+	serviceCtx, logger := logs.Trace(ctx, g.ctx, "Switch")
 	// 1. 依赖模型自身的 SelfCheck 进行全面校验
 	if err := device.SelfCheck(); err != nil {
 		return fmt.Errorf("设备 %v(id=%v) 未通过自检: %w", device.Name, device.ID, err)
@@ -102,7 +104,7 @@ func (g *GeneralDeviceService) Switch(device *models.Device, action DeviceAction
 
 	// 7. 发送指令
 	networkID := areaController.NetworkID
-	sendResult, err := g.comm.Send(networkID, message)
+	sendResult, err := g.comm.Send(serviceCtx, networkID, message)
 	if err != nil {
 		return fmt.Errorf("发送指令到 %s 失败: %w", networkID, err)
 	}
@@ -120,20 +122,21 @@ func (g *GeneralDeviceService) Switch(device *models.Device, action DeviceAction
 		logRecord.ReceivedSuccess = *sendResult.ReceivedSuccess
 	}
 
-	if err := g.deviceCommandLogRepo.Create(logRecord); err != nil {
+	if err := g.deviceCommandLogRepo.Create(serviceCtx, logRecord); err != nil {
 		// 记录日志失败是一个需要关注的问题，但可能不应该中断主流程。
 		// 我们记录一个错误日志，然后成功返回。
-		g.logger.Errorf("创建指令日志失败 (MessageID: %s): %v", sendResult.MessageID, err)
+		logger.Errorf("创建指令日志失败 (MessageID: %s): %v", sendResult.MessageID, err)
 	}
 
-	g.logger.Infof("成功发送指令到 %s 并创建日志 (MessageID: %s)", networkID, sendResult.MessageID)
+	logger.Infof("成功发送指令到 %s 并创建日志 (MessageID: %s)", networkID, sendResult.MessageID)
 	return nil
 }
 
 // Collect 实现了 Service 接口，用于发起对指定区域主控下的多个设备的批量采集请求。
-func (g *GeneralDeviceService) Collect(regionalControllerID uint, devicesToCollect []*models.Device) error {
+func (g *GeneralDeviceService) Collect(ctx context.Context, regionalControllerID uint, devicesToCollect []*models.Device) error {
+	serviceCtx, logger := logs.Trace(ctx, g.ctx, "Collect")
 	if len(devicesToCollect) == 0 {
-		g.logger.Info("待采集设备列表为空，无需执行采集任务。")
+		logger.Info("待采集设备列表为空，无需执行采集任务。")
 		return nil
 	}
 
@@ -153,25 +156,25 @@ func (g *GeneralDeviceService) Collect(regionalControllerID uint, devicesToColle
 	for _, dev := range devicesToCollect {
 		// 依赖模型自身的 SelfCheck 进行全面校验
 		if err := dev.SelfCheck(); err != nil {
-			g.logger.Warnf("跳过设备 %d，因其未通过自检: %v", dev.ID, err)
+			logger.Warnf("跳过设备 %d，因其未通过自检: %v", dev.ID, err)
 			continue
 		}
 		if err := dev.DeviceTemplate.SelfCheck(); err != nil {
-			g.logger.Warnf("跳过设备 %d，因其设备模板未通过自检: %v", dev.ID, err)
+			logger.Warnf("跳过设备 %d，因其设备模板未通过自检: %v", dev.ID, err)
 			continue
 		}
 
 		// 使用模板的 ParseCommands 方法获取传感器指令参数
 		var sensorCmd models.SensorCommands
 		if err := dev.DeviceTemplate.ParseCommands(&sensorCmd); err != nil {
-			g.logger.Warnf("跳过设备 %d，因其模板指令无法解析为 SensorCommands: %v", dev.ID, err)
+			logger.Warnf("跳过设备 %d，因其模板指令无法解析为 SensorCommands: %v", dev.ID, err)
 			continue
 		}
 
 		// 使用模型层预定义的 Bus485Properties 结构体解析设备属性
 		var deviceProps models.Bus485Properties
 		if err := dev.ParseProperties(&deviceProps); err != nil {
-			g.logger.Warnf("跳过设备 %d，因其属性解析失败: %v", dev.ID, err)
+			logger.Warnf("跳过设备 %d，因其属性解析失败: %v", dev.ID, err)
 			continue
 		}
 
@@ -183,10 +186,10 @@ func (g *GeneralDeviceService) Collect(regionalControllerID uint, devicesToColle
 			sensorCmd.ModbusQuantity,
 		)
 		if err != nil {
-			g.logger.Warnf("跳过设备 %d，因生成Modbus RTU读取指令失败: %v", dev.ID, err)
+			logger.Warnf("跳过设备 %d，因生成Modbus RTU读取指令失败: %v", dev.ID, err)
 			continue
 		}
-		g.logger.Debugf("生成485指令: %v", modbusCommandBytes)
+		logger.Debugf("生成485指令: %v", modbusCommandBytes)
 
 		// 构建 Raw485Command，包含总线号
 		raw485Cmd := &proto.Raw485Command{
@@ -216,11 +219,11 @@ func (g *GeneralDeviceService) Collect(regionalControllerID uint, devicesToColle
 		Status:          models.PendingStatusPending,
 		CreatedAt:       time.Now(),
 	}
-	if err := g.pendingCollectionRepo.Create(pendingReq); err != nil {
-		g.logger.Errorf("创建待采集请求失败 (CorrelationID: %s): %v", correlationID, err)
+	if err := g.pendingCollectionRepo.Create(serviceCtx, pendingReq); err != nil {
+		logger.Errorf("创建待采集请求失败 (CorrelationID: %s): %v", correlationID, err)
 		return err
 	}
-	g.logger.Infof("成功创建待采集请求 (CorrelationID: %s, DeviceID: %d)", correlationID, regionalController.ID)
+	logger.Infof("成功创建待采集请求 (CorrelationID: %s, DeviceID: %d)", correlationID, regionalController.ID)
 
 	// 5. 构建最终的空中载荷
 	batchCmd := &proto.BatchCollectCommand{
@@ -234,15 +237,15 @@ func (g *GeneralDeviceService) Collect(regionalControllerID uint, devicesToColle
 	}
 	payload, err := gproto.Marshal(instruction)
 	if err != nil {
-		g.logger.Errorf("序列化采集指令失败 (CorrelationID: %s): %v", correlationID, err)
+		logger.Errorf("序列化采集指令失败 (CorrelationID: %s): %v", correlationID, err)
 		return err
 	}
-	g.logger.Infof("构造空中载荷成功: networkID: %v, payload: %v", networkID, instruction)
-	if _, err := g.comm.Send(networkID, payload); err != nil {
-		g.logger.DPanicf("待采集请求 (CorrelationID: %s) 已创建，但发送到设备失败: %v。数据可能不一致！", correlationID, err)
+	logger.Infof("构造空中载荷成功: networkID: %v, payload: %v", networkID, instruction)
+	if _, err := g.comm.Send(serviceCtx, networkID, payload); err != nil {
+		logger.DPanicf("待采集请求 (CorrelationID: %s) 已创建，但发送到设备失败: %v。数据可能不一致！", correlationID, err)
 		return err
 	}
 
-	g.logger.Infof("成功将采集请求 (CorrelationID: %s) 发送到设备 %s", correlationID, networkID)
+	logger.Infof("成功将采集请求 (CorrelationID: %s) 发送到设备 %s", correlationID, networkID)
 	return nil
 }