From 4b63e81ae123fc75ba1f4c52281dd186bf0ea780 Mon Sep 17 00:00:00 2001 From: liugq Date: Mon, 30 Oct 2023 17:17:29 +0800 Subject: [PATCH] fix get task worker (#208) Reviewed-on: https://git.infini.ltd/infini/console/pulls/208 Co-authored-by: liugq Co-committed-by: liugq --- plugin/task_manager/common_api.go | 2 +- plugin/task_manager/scheduler/scheduler.go | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/plugin/task_manager/common_api.go b/plugin/task_manager/common_api.go index 73cb9b26..a5444044 100644 --- a/plugin/task_manager/common_api.go +++ b/plugin/task_manager/common_api.go @@ -681,7 +681,7 @@ func (h *APIHandler) getChildPipelineInfosFromGateway(pipelineTaskIDs map[string for instID, taskIDs := range pipelineTaskIDs { inst := &model2.TaskWorker{} inst.ID = instID - _, err = orm.Get(inst) + _, err = orm.Get(&inst.Instance) if err != nil { log.Errorf("failed to get instance info, id: %s, err: %v", instID, err) continue diff --git a/plugin/task_manager/scheduler/scheduler.go b/plugin/task_manager/scheduler/scheduler.go index 66ebea41..2a03fbca 100644 --- a/plugin/task_manager/scheduler/scheduler.go +++ b/plugin/task_manager/scheduler/scheduler.go @@ -66,7 +66,7 @@ func (p *scheduler) GetPreferenceInstance(config migration_model.ExecutionConfig if p.CheckInstanceAvailable { tempInst := model.TaskWorker{} tempInst.ID = node.ID - _, err = orm.Get(&tempInst) + _, err = orm.Get(&tempInst.Instance) if err != nil { log.Errorf("failed to get instance, err: %v", err) continue @@ -102,7 +102,7 @@ func (p *scheduler) GetInstance(instanceID string) (*model.TaskWorker, error) { instance := model.TaskWorker{} instance.ID = instanceID - _, err := orm.Get(&instance) + _, err := orm.Get(&instance.Instance) if err != nil { log.Errorf("failed to get instance [%s] from orm, err: %v", instance.ID, err) return nil, err