增加任务增删改查时对设备任务关联表的维护
This commit is contained in:
		@@ -1,6 +1,8 @@
 | 
			
		||||
package task
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"fmt"
 | 
			
		||||
 | 
			
		||||
	"git.huangwc.com/pig/pig-farm-controller/internal/domain/device"
 | 
			
		||||
	"git.huangwc.com/pig/pig-farm-controller/internal/domain/plan"
 | 
			
		||||
	"git.huangwc.com/pig/pig-farm-controller/internal/infra/logs"
 | 
			
		||||
@@ -43,3 +45,27 @@ func (t *taskFactory) Production(claimedLog *models.TaskExecutionLog) plan.Task
 | 
			
		||||
		panic("不支持的任务类型") // 显式panic防编译器报错
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// CreateTaskFromModel 实现了 TaskFactory 接口,用于从模型创建任务实例。
 | 
			
		||||
func (t *taskFactory) CreateTaskFromModel(taskModel *models.Task) (plan.TaskDeviceIDResolver, error) {
 | 
			
		||||
	// 这个方法不关心 claimedLog 的其他字段,所以可以构造一个临时的
 | 
			
		||||
	// 它只用于访问那些不依赖于执行日志的方法,比如 ResolveDeviceIDs
 | 
			
		||||
	tempLog := &models.TaskExecutionLog{Task: *taskModel}
 | 
			
		||||
 | 
			
		||||
	switch taskModel.Type {
 | 
			
		||||
	case models.TaskTypeWaiting:
 | 
			
		||||
		return NewDelayTask(t.logger, tempLog), nil
 | 
			
		||||
	case models.TaskTypeReleaseFeedWeight:
 | 
			
		||||
		return NewReleaseFeedWeightTask(
 | 
			
		||||
			tempLog,
 | 
			
		||||
			t.sensorDataRepo,
 | 
			
		||||
			t.deviceRepo,
 | 
			
		||||
			t.deviceService,
 | 
			
		||||
			t.logger,
 | 
			
		||||
		), nil
 | 
			
		||||
	case models.TaskTypeFullCollection:
 | 
			
		||||
		return NewFullCollectionTask(tempLog, t.deviceRepo, t.deviceService, t.logger), nil
 | 
			
		||||
	default:
 | 
			
		||||
		return nil, fmt.Errorf("不支持为类型 '%s' 的任务创建模型实例", taskModel.Type)
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 
 | 
			
		||||
		Reference in New Issue
	
	Block a user