pig-farm-controller/internal/domain/task/task.go

package task

import (
	"context"
	"fmt"

	"git.huangwc.com/pig/pig-farm-controller/internal/domain/alarm"
	"git.huangwc.com/pig/pig-farm-controller/internal/domain/device"
	"git.huangwc.com/pig/pig-farm-controller/internal/domain/notify"
	"git.huangwc.com/pig/pig-farm-controller/internal/domain/plan"
	"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"
)

const (
	CompNameDelayTask                   = "DelayTask"
	CompNameReleaseFeedWeight           = "ReleaseFeedWeightTask"
	CompNameFullCollectionTask          = "FullCollectionTask"
	CompNameAlarmNotification           = "AlarmNotificationTask"
	CompNameHeartbeatTask               = "HeartbeatTask"
	CompNameOtaCheck                    = "OtaCheckTask"
	CompNameDeviceThresholdCheck        = "DeviceThresholdCheckTask"
	CompNameAreaCollectorThresholdCheck = "AreaCollectorThresholdCheckTask"
	CompNameNotificationRefresh         = "NotificationRefreshTask"
)

type taskFactory struct {
	ctx context.Context

	sensorDataRepo     repository.SensorDataRepository
	deviceRepo         repository.DeviceRepository
	alarmRepo          repository.AlarmRepository
	areaControllerRepo repository.AreaControllerRepository
	otaRepo            repository.OtaRepository

	deviceOperator      device.DeviceOperator
	deviceCommunicator  device.DeviceCommunicator
	notificationService notify.Service
	alarmService        alarm.AlarmService
}

func NewTaskFactory(
	ctx context.Context,
	sensorDataRepo repository.SensorDataRepository,
	deviceRepo repository.DeviceRepository,
	alarmRepo repository.AlarmRepository,
	areaControllerRepo repository.AreaControllerRepository,
	otaRepo repository.OtaRepository,
	deviceOperator device.DeviceOperator,
	deviceCommunicator device.DeviceCommunicator,
	notifyService notify.Service,
	alarmService alarm.AlarmService,
) plan.TaskFactory {
	return &taskFactory{
		ctx:                 ctx,
		sensorDataRepo:      sensorDataRepo,
		deviceRepo:          deviceRepo,
		alarmRepo:           alarmRepo,
		areaControllerRepo:  areaControllerRepo,
		otaRepo:             otaRepo,
		deviceOperator:      deviceOperator,
		deviceCommunicator:  deviceCommunicator,
		notificationService: notifyService,
		alarmService:        alarmService,
	}
}

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(logs.AddCompName(baseCtx, CompNameDelayTask), claimedLog)
	case models.TaskTypeReleaseFeedWeight:
		return NewReleaseFeedWeightTask(logs.AddCompName(baseCtx, CompNameReleaseFeedWeight), claimedLog, t.sensorDataRepo, t.deviceRepo, t.deviceOperator)
	case models.TaskTypeFullCollection:
		return NewFullCollectionTask(logs.AddCompName(baseCtx, CompNameFullCollectionTask), claimedLog, t.deviceRepo, t.deviceOperator)
	case models.TaskTypeHeartbeat:
		return NewHeartbeatTask(logs.AddCompName(baseCtx, CompNameHeartbeatTask), claimedLog, t.areaControllerRepo, t.deviceCommunicator)
	case models.TaskTypeAlarmNotification:
		return NewAlarmNotificationTask(logs.AddCompName(baseCtx, CompNameAlarmNotification), claimedLog, t.notificationService, t.alarmRepo)
	case models.TaskTypeDeviceThresholdCheck:
		return NewDeviceThresholdCheckTask(logs.AddCompName(baseCtx, CompNameDeviceThresholdCheck), claimedLog, t.sensorDataRepo, t.alarmService)
	case models.TaskTypeAreaCollectorThresholdCheck:
		return NewAreaThresholdCheckTask(logs.AddCompName(baseCtx, CompNameAreaCollectorThresholdCheck), claimedLog, t.sensorDataRepo, t.deviceRepo, t.alarmService)
	case models.TaskTypeNotificationRefresh:
		return NewRefreshNotificationTask(logs.AddCompName(baseCtx, CompNameNotificationRefresh), claimedLog, t.alarmService)
	case models.TaskTypeOTACheck:
		return NewOtaCheckTask(logs.AddCompName(baseCtx, CompNameOtaCheck), claimedLog, t.otaRepo)
	default:
		logger.Panicf("不支持的任务类型: %s", claimedLog.Task.Type)
		panic("不支持的任务类型") // 显式panic防编译器报错
	}
}

// CreateTaskFromModel 实现了 TaskFactory 接口，用于从模型创建任务实例。
func (t *taskFactory) CreateTaskFromModel(ctx context.Context, taskModel *models.Task) (plan.TaskDeviceIDResolver, error) {
	tempLog := &models.TaskExecutionLog{Task: *taskModel}
	baseCtx := context.Background()

	switch taskModel.Type {
	case models.TaskTypeWaiting:
		return NewDelayTask(logs.AddCompName(baseCtx, CompNameDelayTask), tempLog), nil
	case models.TaskTypeReleaseFeedWeight:
		return NewReleaseFeedWeightTask(
			logs.AddCompName(baseCtx, CompNameReleaseFeedWeight),
			tempLog,
			t.sensorDataRepo,
			t.deviceRepo,
			t.deviceOperator,
		), nil
	case models.TaskTypeFullCollection:
		return NewFullCollectionTask(logs.AddCompName(baseCtx, CompNameFullCollectionTask), tempLog, t.deviceRepo, t.deviceOperator), nil
	case models.TaskTypeHeartbeat:
		return NewHeartbeatTask(logs.AddCompName(baseCtx, CompNameHeartbeatTask), tempLog, t.areaControllerRepo, t.deviceCommunicator), nil
	case models.TaskTypeAlarmNotification:
		return NewAlarmNotificationTask(logs.AddCompName(baseCtx, CompNameAlarmNotification), tempLog, t.notificationService, t.alarmRepo), nil
	case models.TaskTypeDeviceThresholdCheck:
		return NewDeviceThresholdCheckTask(logs.AddCompName(baseCtx, CompNameDeviceThresholdCheck), tempLog, t.sensorDataRepo, t.alarmService), nil
	case models.TaskTypeAreaCollectorThresholdCheck:
		return NewAreaThresholdCheckTask(logs.AddCompName(baseCtx, CompNameAreaCollectorThresholdCheck), tempLog, t.sensorDataRepo, t.deviceRepo, t.alarmService), nil
	case models.TaskTypeNotificationRefresh:
		return NewRefreshNotificationTask(logs.AddCompName(baseCtx, CompNameNotificationRefresh), tempLog, t.alarmService), nil
	case models.TaskTypeOTACheck:
		return NewOtaCheckTask(logs.AddCompName(baseCtx, CompNameOtaCheck), tempLog, t.otaRepo), nil
	default:
		return nil, fmt.Errorf("不支持为类型 '%s' 的任务创建模型实例", taskModel.Type)
	}
}