Function WithForwardedRequestID

pkg/rpcmetadata/request.go:52–68

WithForwardedRequestID forwards the incoming request ID to outgoing RPCs.

(ctx context.Context)

// WithForwardedRequestID forwards the incoming request ID to outgoing RPCs.
func WithForwardedRequestID(ctx context.Context) context.Context {
	inMD, ok := metadata.FromIncomingContext(ctx)
	if !ok {
		return ctx
	}
	outMD, _ := metadata.FromOutgoingContext(ctx)
	requestID := inMD.Get(requestIDKey)
	if len(requestID) == 0 {
		return ctx
	}
	return metadata.NewOutgoingContext(ctx, metadata.Join(
		outMD,
		metadata.MD{
			requestIDKey: requestID,
		},
	))
}