update image name

nm/task_handler.go

@@ -155,6 +155,7 @@ func (t *TaskWorker) ComputeTaskHandler(taskMsg *nodemanagerV2.PushTaskMessage)
 		t.ExecTaskIdIsFinished.Store(taskMsg.TaskId, true)
 		return
 	}
+	taskOp.taskCmd.ImageName = fmt.Sprintf("%s-%s", taskOp.taskCmd.ImageName, conf.GetConfig().OpSys)
 	model, err := db.GetModel(taskOp.taskCmd.ImageName)
 	if err != nil {
 		log.Errorf("failed to unmarshal task cmd: %s", err.Error())
@@ -164,7 +165,6 @@ func (t *TaskWorker) ComputeTaskHandler(taskMsg *nodemanagerV2.PushTaskMessage)
 	}
 	model.LastWorkTime = time.Now().Unix()
 	model.TotalRunCount++
-	taskOp.taskCmd.ImageName = fmt.Sprintf("%s-%s", taskOp.taskCmd.ImageName, conf.GetConfig().OpSys)
 	log.Info("received task cmd :", taskOp.taskCmd)
 	log.WithField("t.lastExecTaskImageName", t.lastExecTaskImageName).WithField("newTaskImageName", taskOp.taskCmd.ImageName).Info("task image info")
 	if taskMsg.TaskKind != baseV1.TaskKind_StandardTask && conf.GetConfig().IsStopLastContainer {
@@ -236,6 +236,7 @@ func (t *TaskWorker) GetAckResp(taskMsg *nodemanagerV2.PushTaskMessage) (isCanEx
 		log.Errorf("failed to unmarshal task cmd: %s", err.Error())
 		return
 	}
+	taskCmd.ImageName = fmt.Sprintf("%s-%s", taskCmd.ImageName, conf.GetConfig().OpSys)
 	value, ok := t.ExecTaskIdIsFinished.Load(t.lastExecTaskId)
 	if !ok {
 		log.WithField("task id", t.lastExecTaskId).Warn("task exec is not finished")