update download

Signed-off-by: v_wbxiongli <740332065@qq.com>
FederatedAI · Dec 16, 2023 · b1ce806 · b1ce806
1 parent bd02169
commit b1ce806
Showing 1 changed file with 2 additions and 1 deletion.
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,7 +573,8 @@ 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(serverNodeInDb.getServerNodeId().toString(), sessionRankList);
+
+                contentMap.put(String.join(":",serverNodeInDb.getHost(),serverNodeInDb.getPort().toString()), sessionRankList);
                 ErProcessor erProcessor = new ErProcessor();
                 erProcessor.setSessionId(sessionId);
                 erProcessor.setServerNodeId(serverNodeId);