Browse Source

updated task status sync

pull/460/head
tzwang 7 months ago
parent
commit
cfdaebc9fb
2 changed files with 10 additions and 10 deletions
  1. +10
    -8
      internal/logic/core/pagelisttasklogic.go
  2. +0
    -2
      internal/scheduler/service/utils/status/statusSync.go

+ 10
- 8
internal/logic/core/pagelisttasklogic.go View File

@@ -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 {


+ 0
- 2
internal/scheduler/service/utils/status/statusSync.go View File

@@ -119,8 +119,6 @@ func (s *TaskStatus) UpdateTaskStatus(tasklist []*types.TaskModel) {
}

func (s *TaskStatus) updateAiTask(aiTaskList []*models.TaskAi) {
s.taskSyncLock.Lock()
defer s.taskSyncLock.Unlock()
var wg sync.WaitGroup
for _, aitask := range aiTaskList {
t := aitask


Loading…
Cancel
Save