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 ( import (
"context" "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/svc"
"gitlink.org.cn/JointCloud/pcm-coordinator/internal/types" "gitlink.org.cn/JointCloud/pcm-coordinator/internal/types"
"gitlink.org.cn/JointCloud/pcm-coordinator/pkg/repository/result" "gitlink.org.cn/JointCloud/pcm-coordinator/pkg/repository/result"
@@ -13,15 +14,17 @@ import (


type PageListTaskLogic struct { type PageListTaskLogic struct {
logx.Logger 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 { func NewPageListTaskLogic(ctx context.Context, svcCtx *svc.ServiceContext) *PageListTaskLogic {
return &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 // 计算每个任务的运行时间x
for _, model := range list { 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) { func (s *TaskStatus) updateAiTask(aiTaskList []*models.TaskAi) {
s.taskSyncLock.Lock()
defer s.taskSyncLock.Unlock()
var wg sync.WaitGroup var wg sync.WaitGroup
for _, aitask := range aiTaskList { for _, aitask := range aiTaskList {
t := aitask t := aitask


Loading…
Cancel
Save