chore: support initializing request id (#59)

* chore: support initializing request id

* ci: fix codecov

Author: aaron-steinfeld

.github/workflows/pr-test.yml

@@ -21,11 +21,12 @@ jobs:
           args: build jacocoTestReport
 
       - name: Upload coverage to Codecov
-        uses: codecov/codecov-action@v3
+        uses: codecov/codecov-action@v4
         with:
           name: unit test reports
           fail_ci_if_error: true
           flags: unit
+          token: ${{ secrets.CODECOV_TOKEN }}
 
       - name: copy test reports
         uses: hypertrace/github-actions/gradle@main

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

@@ -6,7 +6,10 @@
 import io.grpc.ServerCall;
 import io.grpc.ServerCallHandler;
 import io.grpc.ServerInterceptor;
+import java.util.Optional;
+import java.util.UUID;
 import org.hypertrace.core.grpcutils.context.RequestContext;
+import org.hypertrace.core.grpcutils.context.RequestContextConstants;
 
 /**
  * Interceptor which intercepts the request headers to extract request context and sets it in the
@@ -20,9 +23,18 @@ public <ReqT, RespT> ServerCall.Listener<ReqT> interceptCall(
       ServerCall<ReqT, RespT> serverCall,
       Metadata metadata,
       ServerCallHandler<ReqT, RespT> serverCallHandler) {
-    RequestContext requestContext = RequestContext.fromMetadata(metadata);
-    Context ctx = Context.current().withValue(RequestContext.CURRENT, requestContext);
+    RequestContext currentContext =
+        Optional.ofNullable(RequestContext.CURRENT.get())
+            .orElseGet(() -> RequestContext.fromMetadata(metadata));
+    if (currentContext.getHeaderValue(RequestContextConstants.REQUEST_ID_HEADER_KEY).isEmpty()) {
+      currentContext.put(
+          RequestContextConstants.REQUEST_ID_HEADER_KEY, UUID.randomUUID().toString());
+    }
 
-    return Contexts.interceptCall(ctx, serverCall, metadata, serverCallHandler);
+    return Contexts.interceptCall(
+        Context.current().withValue(RequestContext.CURRENT, currentContext),
+        serverCall,
+        metadata,
+        serverCallHandler);
   }
 }