update download

Signed-off-by: v_wbxiongli <740332065@qq.com>
FederatedAI · Dec 18, 2023 · 3e13151 · 3e13151
1 parent f7d0724
commit 3e13151
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/...cluster_manager/src/main/java/org/fedai/eggroll/clustermanager/job/JobServiceHandler.java b/...cluster_manager/src/main/java/org/fedai/eggroll/clustermanager/job/JobServiceHandler.java
@@ -573,8 +573,7 @@ public PrepareJobDownloadResponse prepareJobDownload(Context context, PrepareJob
                 for (SessionRanksTemp rank : sessionRanks) {
                     sessionRankList.add(Arrays.asList(rank.getServerNodeId(),rank.getContainerId(),rank.getGlobalRank(),rank.getLocalRank(),rank.getIndex()));
                 }
-
-                contentMap.put(String.join(":",serverNodeInDb.getHost(),serverNodeInDb.getPort().toString()), sessionRankList);
+                contentMap.put(serverNodeInDb.getServerNodeId().toString(), sessionRankList);
                 ErProcessor erProcessor = new ErProcessor();
                 erProcessor.setSessionId(sessionId);
                 erProcessor.setServerNodeId(serverNodeId);
@@ -603,7 +602,7 @@ public PrepareJobDownloadResponse prepareJobDownload(Context context, PrepareJob
 
         newErSessionMeta.getProcessors().forEach(p -> {
             if (contentMap.containsKey(p.getServerNodeId())) {
-                contentMap.put(p.getTransferEndpoint().toString(), p.getServerNodeId());
+                contentMap.put(p.getTransferEndpoint().toString(), contentMap.get(p.getServerNodeId()));
                 contentMap.remove(p.getServerNodeId());
             } else {
                 log.info("download cannot found node {}", p.getServerNodeId());