expose-latency-percentile-metrics-grpc (#92)

Co-authored-by: Dhamodhar Reddy  Dakannagari <[email protected]>

Author: Dhamodhar-DDR

platform-grpc-service-framework/src/main/java/org/hypertrace/core/serviceframework/grpc/GrpcPlatformServiceContainer.java

@@ -6,6 +6,7 @@
 
 import io.grpc.Server;
 import io.grpc.ServerBuilder;
+import io.grpc.Status;
 import io.grpc.health.v1.HealthCheckRequest;
 import io.grpc.health.v1.HealthCheckResponse.ServingStatus;
 import io.grpc.health.v1.HealthGrpc;
@@ -25,6 +26,7 @@
 import java.util.concurrent.ScheduledExecutorService;
 import java.util.concurrent.ScheduledFuture;
 import java.util.function.Function;
+import java.util.function.UnaryOperator;
 import java.util.stream.Collectors;
 import java.util.stream.Stream;
 import lombok.Value;
@@ -63,7 +65,11 @@ protected void doInit() {
     final ServerBuilder<?> inProcessServerBuilder =
         InProcessServerBuilder.forName(this.getInProcessServerName())
             .intercept(
-                new MetricCollectingServerInterceptor(PlatformMetricsRegistry.getMeterRegistry()))
+                new MetricCollectingServerInterceptor(
+                    PlatformMetricsRegistry.getMeterRegistry(),
+                    UnaryOperator.identity(),
+                    timerBuilder -> timerBuilder.publishPercentiles(0.5, 0.75, 0.90, 0.95, 0.99),
+                    Status.Code.OK))
             .addService(this.healthStatusManager.getHealthService());
     final GrpcServiceContainerEnvironment serviceContainerEnvironment =
         this.buildContainerEnvironment(this.grpcChannelRegistry, this.healthStatusManager);
@@ -198,7 +204,11 @@ protected InProcessGrpcChannelRegistry buildChannelRegistry() {
         this.getAuthorityInProcessOverrideMap(),
         GrpcRegistryConfig.builder()
             .defaultInterceptor(
-                new MetricCollectingClientInterceptor(PlatformMetricsRegistry.getMeterRegistry()))
+                new MetricCollectingClientInterceptor(
+                    PlatformMetricsRegistry.getMeterRegistry(),
+                    UnaryOperator.identity(),
+                    timerBuilder -> timerBuilder.publishPercentiles(0.5, 0.75, 0.90, 0.95, 0.99),
+                    Status.Code.OK))
             .build());
   }
 
@@ -244,7 +254,11 @@ private ServerBuilder<?> initializeBuilder(GrpcPlatformServerDefinition serverDe
     }
     // add micrometer-grpc interceptor to collect server metrics.
     builder.intercept(
-        new MetricCollectingServerInterceptor(PlatformMetricsRegistry.getMeterRegistry()));
+        new MetricCollectingServerInterceptor(
+            PlatformMetricsRegistry.getMeterRegistry(),
+            UnaryOperator.identity(),
+            timerBuilder -> timerBuilder.publishPercentiles(0.5, 0.75, 0.90, 0.95, 0.99),
+            Status.Code.OK));
 
     serverDefinition.getServerInterceptors().forEach(builder::intercept);
     return builder;