update bit map submit

nm/model_handler.go

@@ -50,7 +50,7 @@ func monitorModelInfo(dockerOp *operate.DockerOp) {
 				continue
 			}
 			reportTaskIds := make([]uint64, 0)
-			reportTaskIds = append(reportTaskIds, uint64(len(modelInfosResp)))
+			maxLong := uint64(0)
 			for _, modelInfo := range modelInfosResp {
 				if modelInfo.ImageName == "" {
 					continue
@@ -72,9 +72,13 @@ func monitorModelInfo(dockerOp *operate.DockerOp) {
 				} else {
 					reportTaskIds = append(reportTaskIds, modelInfo.TaskId)
 				}
+				if modelInfo.TaskId > maxLong {
+					maxLong = modelInfo.TaskId
+				}
 				dockerOp.SignApi[modelInfo.ImageName] = modelInfo.SignUrl
 				dockerOp.ModelsInfo = append(dockerOp.ModelsInfo, modelInfo)
 			}
+			reportTaskIds = append(reportTaskIds, maxLong)
 			dockerOp.ModelTaskIdIndexesChan <- reportTaskIds
 			ticker = time.NewTicker(time.Minute * 10)
 		}

nm/msg_resp.go

@@ -76,9 +76,9 @@ func HeartbeatResp(params ...interface{}) *nodemanagerV1.WorkerMessage {
 func SubmitResourceMapRes(params ...interface{}) *nodemanagerV1.WorkerMessage {
 	log.Info("Submit resource map response received params: ", params)
 	taskIdIndexes := params[0].([]uint64)
-	taskIdLength := taskIdIndexes[0]
+	taskIdLength := taskIdIndexes[len(taskIdIndexes)-1]
 	b := bitmap.New(taskIdLength)
-	for i := 1; i < len(taskIdIndexes); i++ {
+	for i := 0; i < len(taskIdIndexes)-1; i++ {
 		taskIdIndex := taskIdIndexes[i]
 		err := b.Set(taskIdIndex)
 		if err != nil {