实现 AlarmNotificationTask

internal/domain/task/task.go

@@ -5,6 +5,7 @@ import (
 	"fmt"
 
 	"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"
@@ -19,23 +20,31 @@ const (
 )
 
 type taskFactory struct {
-	ctx            context.Context
+	ctx context.Context
+
 	sensorDataRepo repository.SensorDataRepository
 	deviceRepo     repository.DeviceRepository
-	deviceService  device.Service
+	alarmRepo      repository.AlarmRepository
+
+	deviceService       device.Service
+	notificationService notify.Service
 }
 
 func NewTaskFactory(
 	ctx context.Context,
 	sensorDataRepo repository.SensorDataRepository,
 	deviceRepo repository.DeviceRepository,
+	alarmRepo repository.AlarmRepository,
 	deviceService device.Service,
+	notifyService notify.Service,
 ) plan.TaskFactory {
 	return &taskFactory{
-		ctx:            ctx,
-		sensorDataRepo: sensorDataRepo,
-		deviceRepo:     deviceRepo,
-		deviceService:  deviceService,
+		ctx:                 ctx,
+		sensorDataRepo:      sensorDataRepo,
+		deviceRepo:          deviceRepo,
+		alarmRepo:           alarmRepo,
+		deviceService:       deviceService,
+		notificationService: notifyService,
 	}
 }
 
@@ -50,7 +59,7 @@ func (t *taskFactory) Production(ctx context.Context, claimedLog *models.TaskExe
 	case models.TaskTypeFullCollection:
 		return NewFullCollectionTask(logs.AddCompName(baseCtx, CompNameFullCollectionTask), claimedLog, t.deviceRepo, t.deviceService)
 	case models.TaskTypeAlarmNotification:
-		return NewAlarmNotificationTask(logs.AddCompName(baseCtx, CompNameAlarmNotification), claimedLog)
+		return NewAlarmNotificationTask(logs.AddCompName(baseCtx, CompNameAlarmNotification), claimedLog, t.notificationService, t.alarmRepo)
 	default:
 		// TODO 这里直接panic合适吗? 不过这个场景确实不该出现任何异常的任务类型
 		logger.Panicf("不支持的任务类型: %s", claimedLog.Task.Type)
@@ -79,7 +88,7 @@ func (t *taskFactory) CreateTaskFromModel(ctx context.Context, taskModel *models
 	case models.TaskTypeFullCollection:
 		return NewFullCollectionTask(logs.AddCompName(baseCtx, CompNameFullCollectionTask), tempLog, t.deviceRepo, t.deviceService), nil
 	case models.TaskTypeAlarmNotification:
-		return NewAlarmNotificationTask(logs.AddCompName(baseCtx, CompNameAlarmNotification), tempLog), nil
+		return NewAlarmNotificationTask(logs.AddCompName(baseCtx, CompNameAlarmNotification), tempLog, t.notificationService, t.alarmRepo), nil
 	default:
 		return nil, fmt.Errorf("不支持为类型 '%s' 的任务创建模型实例", taskModel.Type)
 	}