update GetMaxExecTime

nm/task_handler.go

@@ -433,14 +433,7 @@ func (op *TaskOp) checkContainerHealthy(internalIp string, internalPort uint16)
 }
 
 func (op *TaskOp) waitContainerRunning(handler *TaskWorker, imageId string) error {
-	maxExecTime, err := strconv.ParseInt(op.taskParam.Headers[models.MaxExecTime][0], 10, 64)
-	if err != nil {
-		log.Errorf("%s-%s", "Parse max exec time", err.Error())
-		return fmt.Errorf("%s-%s", "Parse max exec time", err.Error())
-	}
-	if maxExecTime == 0 {
-		maxExecTime = models.DefaultMaxExecTime
-	}
+	maxExecTime := op.GetMaxExecTime()
 	log.WithField("maxExecTime", maxExecTime).Info("Waiting for container running", imageId)
 	for {
 		select {
@@ -498,14 +491,7 @@ func (op *TaskOp) validateWebHook() error {
 }
 
 func (op *TaskOp) waitReqContainerOk(dockerOp *operate.DockerOp) error {
-	maxExecTime, err := strconv.ParseInt(op.taskParam.Headers[models.MaxExecTime][0], 10, 64)
-	if err != nil {
-		log.Errorf("%s-%s", "Parse max exec time", err.Error())
-		return fmt.Errorf("%s-%s", "Parse max exec time", err.Error())
-	}
-	if maxExecTime == 0 {
-		maxExecTime = models.DefaultMaxExecTime
-	}
+	maxExecTime := op.GetMaxExecTime()
 	log.WithField("maxExecTime", maxExecTime).Info("Waiting for request container success")
 	for {
 		select {
@@ -750,6 +736,18 @@ func (op *TaskOp) waitReqContainerOk(dockerOp *operate.DockerOp) error {
 	}
 }
 
+func (op *TaskOp) GetMaxExecTime() int64 {
+	maxExecTimeArr := op.taskParam.Headers[models.MaxExecTime]
+	var maxExecTime int64
+	if len(maxExecTimeArr) > 0 {
+		maxExecTime, _ = strconv.ParseInt(maxExecTimeArr[0], 10, 64)
+	}
+	if maxExecTime == 0 {
+		maxExecTime = models.DefaultMaxExecTime
+	}
+	return maxExecTime
+}
+
 func parseData(readBody []byte) interface{} {
 	var m map[string]json.RawMessage
 	if err := json.Unmarshal(readBody, &m); err != nil {