diff --git a/server/src/main/java/org/apache/uniffle/server/ShuffleServerGrpcService.java b/server/src/main/java/org/apache/uniffle/server/ShuffleServerGrpcService.java index 2a4f77e921..9c32e8621b 100644 --- a/server/src/main/java/org/apache/uniffle/server/ShuffleServerGrpcService.java +++ b/server/src/main/java/org/apache/uniffle/server/ShuffleServerGrpcService.java @@ -1322,6 +1322,12 @@ public void getLocalShuffleIndex( ShuffleServerMetrics.gaugeReadLocalIndexFileBufferSize.inc(assumedFileSize); GetLocalShuffleIndexResponse.Builder builder = GetLocalShuffleIndexResponse.newBuilder().setStatus(status.toProto()).setRetMsg(msg); + builder.setIndexData(UnsafeByteOperations.unsafeWrap(data)); + builder.setDataFileLen(shuffleIndexResult.getDataFileLen()); + builder.addAllStorageIds( + Arrays.stream(shuffleIndexResult.getStorageIds()) + .boxed() + .collect(Collectors.toList())); long readTime = System.currentTimeMillis() - start; shuffleServer .getGrpcMetrics() @@ -1331,13 +1337,6 @@ public void getLocalShuffleIndex( readTime, data.remaining(), requestInfo); - - builder.setIndexData(UnsafeByteOperations.unsafeWrap(data)); - builder.setDataFileLen(shuffleIndexResult.getDataFileLen()); - builder.addAllStorageIds( - Arrays.stream(shuffleIndexResult.getStorageIds()) - .boxed() - .collect(Collectors.toList())); auditContext.withReturnValue("len=" + shuffleIndexResult.getDataFileLen()); reply = builder.build(); } catch (FileNotFoundException indexFileNotFoundException) {