修改domain包

internal/domain/task/task.go

@@ -1,6 +1,7 @@
 package task
 
 import (
+	"context"
 	"fmt"
 
 	"git.huangwc.com/pig/pig-farm-controller/internal/domain/device"
@@ -10,61 +11,70 @@ import (
 	"git.huangwc.com/pig/pig-farm-controller/internal/infra/repository"
 )
 
+const (
+	CompNameDelayTask          = "DelayTask"
+	CompNameReleaseFeedWeight  = "ReleaseFeedWeightTask"
+	CompNameFullCollectionTask = "FullCollectionTask"
+)
+
 type taskFactory struct {
-	logger         *logs.Logger
+	ctx            context.Context
 	sensorDataRepo repository.SensorDataRepository
 	deviceRepo     repository.DeviceRepository
 	deviceService  device.Service
 }
 
 func NewTaskFactory(
-	logger *logs.Logger,
+	ctx context.Context,
 	sensorDataRepo repository.SensorDataRepository,
 	deviceRepo repository.DeviceRepository,
 	deviceService device.Service,
 ) plan.TaskFactory {
 	return &taskFactory{
-		logger:         logger,
+		ctx:            ctx,
 		sensorDataRepo: sensorDataRepo,
 		deviceRepo:     deviceRepo,
 		deviceService:  deviceService,
 	}
 }
 
-func (t *taskFactory) Production(claimedLog *models.TaskExecutionLog) plan.Task {
+func (t *taskFactory) Production(ctx context.Context, claimedLog *models.TaskExecutionLog) plan.Task {
+	logger := logs.TraceLogger(ctx, t.ctx, "Production")
+	baseCtx := context.Background()
 	switch claimedLog.Task.Type {
 	case models.TaskTypeWaiting:
-		return NewDelayTask(t.logger, claimedLog)
+		return NewDelayTask(logs.AddCompName(baseCtx, CompNameDelayTask), claimedLog)
 	case models.TaskTypeReleaseFeedWeight:
-		return NewReleaseFeedWeightTask(claimedLog, t.sensorDataRepo, t.deviceRepo, t.deviceService, t.logger)
+		return NewReleaseFeedWeightTask(logs.AddCompName(baseCtx, CompNameReleaseFeedWeight), claimedLog, t.sensorDataRepo, t.deviceRepo, t.deviceService)
 	case models.TaskTypeFullCollection:
-		return NewFullCollectionTask(claimedLog, t.deviceRepo, t.deviceService, t.logger)
+		return NewFullCollectionTask(logs.AddCompName(baseCtx, CompNameFullCollectionTask), claimedLog, t.deviceRepo, t.deviceService)
 	default:
 		// TODO 这里直接panic合适吗? 不过这个场景确实不该出现任何异常的任务类型
-		t.logger.Panicf("不支持的任务类型: %s", claimedLog.Task.Type)
+		logger.Panicf("不支持的任务类型: %s", claimedLog.Task.Type)
 		panic("不支持的任务类型") // 显式panic防编译器报错
 	}
 }
 
 // CreateTaskFromModel 实现了 TaskFactory 接口，用于从模型创建任务实例。
-func (t *taskFactory) CreateTaskFromModel(taskModel *models.Task) (plan.TaskDeviceIDResolver, error) {
+func (t *taskFactory) CreateTaskFromModel(ctx context.Context, taskModel *models.Task) (plan.TaskDeviceIDResolver, error) {
 	// 这个方法不关心 claimedLog 的其他字段，所以可以构造一个临时的
 	// 它只用于访问那些不依赖于执行日志的方法，比如 ResolveDeviceIDs
 	tempLog := &models.TaskExecutionLog{Task: *taskModel}
+	baseCtx := context.Background()
 
 	switch taskModel.Type {
 	case models.TaskTypeWaiting:
-		return NewDelayTask(t.logger, tempLog), nil
+		return NewDelayTask(logs.AddCompName(baseCtx, CompNameDelayTask), tempLog), nil
 	case models.TaskTypeReleaseFeedWeight:
 		return NewReleaseFeedWeightTask(
+			logs.AddCompName(baseCtx, CompNameReleaseFeedWeight),
 			tempLog,
 			t.sensorDataRepo,
 			t.deviceRepo,
 			t.deviceService,
-			t.logger,
 		), nil
 	case models.TaskTypeFullCollection:
-		return NewFullCollectionTask(tempLog, t.deviceRepo, t.deviceService, t.logger), nil
+		return NewFullCollectionTask(logs.AddCompName(baseCtx, CompNameFullCollectionTask), tempLog, t.deviceRepo, t.deviceService), nil
 	default:
 		return nil, fmt.Errorf("不支持为类型 '%s' 的任务创建模型实例", taskModel.Type)
 	}