|
|
|
@@ -2,6 +2,7 @@ package core |
|
|
|
|
|
|
|
import ( |
|
|
|
"context" |
|
|
|
"gitlink.org.cn/JointCloud/pcm-coordinator/internal/scheduler/service/utils/status" |
|
|
|
"gitlink.org.cn/JointCloud/pcm-coordinator/internal/svc" |
|
|
|
"gitlink.org.cn/JointCloud/pcm-coordinator/internal/types" |
|
|
|
"gitlink.org.cn/JointCloud/pcm-coordinator/pkg/repository/result" |
|
|
|
@@ -13,15 +14,17 @@ import ( |
|
|
|
|
|
|
|
type PageListTaskLogic struct { |
|
|
|
logx.Logger |
|
|
|
ctx context.Context |
|
|
|
svcCtx *svc.ServiceContext |
|
|
|
ctx context.Context |
|
|
|
svcCtx *svc.ServiceContext |
|
|
|
taskStatus *status.TaskStatus |
|
|
|
} |
|
|
|
|
|
|
|
func NewPageListTaskLogic(ctx context.Context, svcCtx *svc.ServiceContext) *PageListTaskLogic { |
|
|
|
return &PageListTaskLogic{ |
|
|
|
Logger: logx.WithContext(ctx), |
|
|
|
ctx: ctx, |
|
|
|
svcCtx: svcCtx, |
|
|
|
Logger: logx.WithContext(ctx), |
|
|
|
ctx: ctx, |
|
|
|
svcCtx: svcCtx, |
|
|
|
taskStatus: status.NewTaskStatus(svcCtx.Scheduler.AiStorages, svcCtx.Scheduler.AiService.AiCollectorAdapterMap, &svcCtx.Config), |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
@@ -61,9 +64,8 @@ func (l *PageListTaskLogic) PageListTask(req *types.PageTaskReq) (resp *types.Pa |
|
|
|
} |
|
|
|
|
|
|
|
// 更新智算任务状态 |
|
|
|
//taskStatus := status.NewTaskStatus(l.svcCtx.Scheduler.AiStorages, l.svcCtx.Scheduler.AiService.AiCollectorAdapterMap, &l.svcCtx.Config) |
|
|
|
//go taskStatus.UpdateTaskStatus(list) |
|
|
|
//go taskStatus.UpdateAiTaskStatus(list) |
|
|
|
go l.taskStatus.UpdateTaskStatus(list) |
|
|
|
go l.taskStatus.UpdateAiTaskStatus(list) |
|
|
|
|
|
|
|
// 计算每个任务的运行时间x |
|
|
|
for _, model := range list { |
|
|
|
|