[AST-494] request-id distributed logging interceptor (#88)

Author: iamajais

grpc-context-utils/src/main/java/org/hypertrace/core/grpcutils/context/RequestContextConstants.java

@@ -12,6 +12,7 @@
 public class RequestContextConstants {
   public static final String TENANT_ID_HEADER_KEY = "x-tenant-id";
   public static final String REQUEST_ID_HEADER_KEY = "request-id";
+  public static final String CONTEXT_ID_HEADER_KEY = "context-id";
 
   public static final Metadata.Key<String> TENANT_ID_METADATA_KEY =
       Metadata.Key.of(TENANT_ID_HEADER_KEY, ASCII_STRING_MARSHALLER);
@@ -22,6 +23,7 @@ public class RequestContextConstants {
   public static final Set<String> HEADER_PREFIXES_TO_BE_PROPAGATED =
       Set.of(
           TENANT_ID_HEADER_KEY,
+          CONTEXT_ID_HEADER_KEY,
           "X-B3-",
           "grpc-trace-bin",
           "traceparent",

grpc-server-utils/src/main/java/org/hypertrace/core/grpcutils/server/RequestContextLoggingServerInterceptor.java

@@ -1,6 +1,8 @@
 package org.hypertrace.core.grpcutils.server;
 
+import static org.hypertrace.core.grpcutils.context.RequestContextConstants.CONTEXT_ID_HEADER_KEY;
 import static org.hypertrace.core.grpcutils.context.RequestContextConstants.REQUEST_ID_HEADER_KEY;
+import static org.hypertrace.core.grpcutils.context.RequestContextConstants.TENANT_ID_HEADER_KEY;
 
 import io.grpc.Context;
 import io.grpc.Contexts;
@@ -27,6 +29,8 @@ public <ReqT, RespT> ServerCall.Listener<ReqT> interceptCall(
         Optional.ofNullable(RequestContext.CURRENT.get())
             .orElseGet(() -> RequestContext.fromMetadata(metadata));
     Optional<String> opRequestId = currentContext.getHeaderValue(REQUEST_ID_HEADER_KEY);
+    Optional<String> opTenantId = currentContext.getHeaderValue(TENANT_ID_HEADER_KEY);
+    Optional<String> opContextId = currentContext.getHeaderValue(CONTEXT_ID_HEADER_KEY);
     if (opRequestId.isEmpty()) {
       opRequestId = Optional.of(FastUUIDGenerator.randomUUID().toString());
     }
@@ -63,6 +67,8 @@ public void onComplete() {
       public void onMessage(ReqT message) {
         try {
           MDC.put(REQUEST_ID_HEADER_KEY, requestId);
+          opTenantId.ifPresent(s -> MDC.put(TENANT_ID_HEADER_KEY, s));
+          opContextId.ifPresent(s -> MDC.put(CONTEXT_ID_HEADER_KEY, s));
         } catch (Exception e) {
           log.error("Error while setting request context details in MDC params", e);
         }