fix spark yarn cluster mode bug

apache · Aug 24, 2023 · 2fb31f0 · 2fb31f0
1 parent 743b417
commit 2fb31f0
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/...n/java/org/apache/linkis/manager/am/service/impl/DefaultEngineConnPidCallbackService.java b/...n/java/org/apache/linkis/manager/am/service/impl/DefaultEngineConnPidCallbackService.java
@@ -64,13 +64,12 @@ public void dealPid(ResponseEngineConnPid protocol) {
     }
 
     engineNode.setIdentifier(protocol.pid());
-
+    ServiceInstance oldServiceInstance = engineNode.getServiceInstance();
     if (engineNode.getMark().equals(AMConstant.CLUSTER_PROCESS_MARK)) {
       ServiceInstance serviceInstance = protocol.serviceInstance();
       engineNode.setServiceInstance(serviceInstance);
-      getEngineNodeManager().updateEngineNode(serviceInstance, engineNode);
-      nodeLabelService.labelsFromInstanceToNewInstance(
-          engineNode.getServiceInstance(), serviceInstance);
+      getEngineNodeManager().updateEngineNode(oldServiceInstance, engineNode);
+      nodeLabelService.labelsFromInstanceToNewInstance(oldServiceInstance, serviceInstance);
     }
     defaultEngineNodeManager.updateEngine(engineNode);
   }