update param parse

nm/model_handler.go

@@ -86,12 +86,12 @@ func reportModelInfo(nodeManager *models.NodeManagerClient,
 	msgRespWorker *RespMsgWorker, dockerOp *operate.DockerOp) {
 	for {
 		select {
-		case taskId := <-dockerOp.ModelTaskIdIndexesChan:
+		case taskIdIndexes := <-dockerOp.ModelTaskIdIndexesChan:
 			if !nodeManager.Status {
 				log.WithField("endpoint", nodeManager.Endpoint).Error("Node manager is down , stop report model info")
 				return
 			}
-			params := buildParams(taskId)
+			params := buildParams(taskIdIndexes)
 			msgRespWorker.RegisterMsgResp(nodeManager, worker, SubmitResourceMapRes, params)
 		}
 	}

nm/msg_resp.go

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