From 1c56e74c9e08873874a961117f7efa1b8f0b368b Mon Sep 17 00:00:00 2001 From: Junfan Zhang Date: Tue, 24 Dec 2024 11:54:45 +0800 Subject: [PATCH] [#2298] fix(server): followup to correct the metric of grpc_get_local_shuffle_index_process_latency (#2302) ### What changes were proposed in this pull request? followup to correct the metric of grpc_get_local_shuffle_index ### Why are the changes needed? Fix: #2298 ### Does this PR introduce _any_ user-facing change? No. ### How was this patch tested? Co-authored-by: Junfan Zhang --- .../uniffle/server/ShuffleServerGrpcService.java | 13 ++++++------- 1 file changed, 6 insertions(+), 7 deletions(-) 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) {