update waitContainerRunning

nm/task_handler.go

@@ -187,16 +187,16 @@ func (t *TaskWorker) ComputeTaskHandler(taskMsg *nodeManagerV1.PushTaskMessage)
 			return
 		}
 		log.Infof("Started container with ID %s", containerId)
-		if err = taskOp.waitContainerRunning(t, imageId); err != nil {
-			taskOp.taskExecResult.TaskExecError = fmt.Sprintf("%s", err.Error())
-			t.ExecTaskIdIsSuccess.Store(taskMsg.TaskId, true)
-			return
-		}
 	} else {
 		taskOp.taskCmd.ApiUrl = fmt.Sprintf("http://%s:%d%s", internalIp, internalPort, taskOp.taskCmd.ApiUrl)
 		log.Info("Container ports:", internalPort)
 		log.WithField("ApiUrl", taskOp.taskCmd.ApiUrl).Info("The image is running")
 	}
+	if err = taskOp.waitContainerRunning(t, imageId); err != nil {
+		taskOp.taskExecResult.TaskExecError = fmt.Sprintf("%s", err.Error())
+		t.ExecTaskIdIsSuccess.Store(taskMsg.TaskId, true)
+		return
+	}
 	if err = taskOp.waitReqContainerOk(t.DockerOp); err != nil {
 		taskOp.taskExecResult.TaskExecError = fmt.Sprintf("%s", err.Error())
 		t.ExecTaskIdIsSuccess.Store(taskMsg.TaskId, true)