update reconnect nm problem and start image

nm/model_handler.go

@@ -44,10 +44,6 @@ func monitorModelInfo(dockerOp *operate.DockerOp) {
 				continue
 			}
 			modelInfosResp := resp.Data
-			modelInfoMap := make(map[uint64]*models.ModelInfo, 0)
-			for _, modelInfo := range modelInfosResp {
-				modelInfoMap[modelInfo.TaskId] = modelInfo
-			}
 			imageNameMap, err := dockerOp.PsImageNameMap()
 			if err != nil {
 				log.Error("Docker op ps images failed:", err)

nm/msg_resp.go

@@ -48,7 +48,7 @@ func (o *RespMsgWorker) SendMsg() {
 				if err != nil {
 					log.Error("Send heartbeat msg error:", err)
 					pool.nodeManager.UpdateStatus(false)
-					continue
+					return
 				}
 				log.Info("Worker client send message successfully")
 			}

nm/task_msg.go

@@ -112,19 +112,23 @@ func (t *TaskHandler) ComputeTaskHandler(taskMsg *nodeManagerV1.PushTaskMessage)
 			if tag == taskCmd.ImageName {
 				imageId = image.ID
 				isFound = true
+				log.Info("The image found success:", image.ID)
 				break
 			}
 		}
-		log.Println(image.ID)
+	}
+	if !isFound {
+		log.Error("The image is not found:", taskCmd.ImageName)
+		return
 	}
 	containers := t.DockerOp.ListContainer()
 	isImageRunExist := false
 	for _, container := range containers {
-		if container.ImageID == imageId {
+		if container.ImageID == imageId && container.Status == "running" {
 			networks := container.NetworkSettings.Networks
 			ip := ""
-			for _, settings := range networks {
-				ip = settings.IPAddress
+			for _, endPoint := range networks {
+				ip = endPoint.IPAddress
 				log.Warn("Container network ip:", ip)
 			}
 			//taskCmd.ApiUrl = fmt.Sprintf(taskCmd.ApiUrl, container.Ports[0].PublicPort)