log enhance

apache · Jun 19, 2024 · d63d3ce · d63d3ce
1 parent 5b26a32
commit d63d3ce
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/...rk/common/src/main/java/org/apache/uniffle/shuffle/manager/ShuffleManagerGrpcService.java b/...rk/common/src/main/java/org/apache/uniffle/shuffle/manager/ShuffleManagerGrpcService.java
@@ -171,13 +171,18 @@ public void reportShuffleFetchFailure(
               String.format(
                   "Activate stage retry on stage(%d:%d), taskFailuresCount:(%d)",
                   stageId, stageAttempt, rssShuffleStatus.getTaskFailureAttemptCount());
-          if (shuffleManager.reassignOnStageResubmit(stageId, stageAttempt, shuffleId, -1)) {
+          int partitionNum = shuffleManager.getPartitionNum(shuffleId);
+          if (shuffleManager.reassignOnStageResubmit(
+              stageId, stageAttempt, shuffleId, partitionNum)) {
             LOG.info(
-                "{} from executorId({}), task({}:{})",
+                "{} from executorId({}), task({}:{}) on stageId({}:{}), shuffleId({})",
                 msg,
                 executorId,
                 taskAttemptId,
-                taskAttemptNumber);
+                taskAttemptNumber,
+                stageId,
+                stageAttempt,
+                shuffleId);
           }
         } else {
           reSubmitWholeStage = false;