Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

stats/opentelemetry: introduce tracing propagator and carrier #7677

Merged

Conversation

purnesh42H
Copy link
Contributor

@purnesh42H purnesh42H commented Sep 27, 2024

This PR adds the GRPCTraceBinPropagator that implements TextMapPropagator as proposed in gRFC A72 which outlines a custom propagator to handle both binary and text formats for trace context propagation. This will allow gRPC to keep using grpc-trace-bin header for context propagation and also support any other text propagators. When using grpc-trace-bin the OpenCensus SpanContext and OpenTelemetry SpanContext are identical, therefore a gRPC OpenCensus client can speak with a gRPC OpenTelemetry server and vice versa. It is encouraged to use GRPCTraceBinPropagator for the migration. Using the same header greatly simplifies rollout.

RELEASE NOTES: None

Copy link

codecov bot commented Sep 27, 2024

Codecov Report

Attention: Patch coverage is 77.77778% with 18 lines in your changes missing coverage. Please review.

Project coverage is 81.86%. Comparing base (e2b98f9) to head (5c7b69a).
Report is 24 commits behind head on master.

Files with missing lines Patch % Lines
stats/opentelemetry/internal/tracing/carrier.go 68.29% 11 Missing and 2 partials ⚠️
stats/opentelemetry/grpc_trace_bin_propagator.go 87.50% 4 Missing and 1 partial ⚠️
Additional details and impacted files
@@            Coverage Diff             @@
##           master    #7677      +/-   ##
==========================================
- Coverage   81.91%   81.86%   -0.05%     
==========================================
  Files         374      377       +3     
  Lines       37930    38065     +135     
==========================================
+ Hits        31071    31163      +92     
- Misses       5566     5591      +25     
- Partials     1293     1311      +18     
Files with missing lines Coverage Δ
stats/opentelemetry/grpc_trace_bin_propagator.go 87.50% <87.50%> (ø)
stats/opentelemetry/internal/tracing/carrier.go 68.29% <68.29%> (ø)

... and 46 files with indirect coverage changes

---- 🚨 Try these New Features:

@purnesh42H purnesh42H force-pushed the otel-tracing-grpc-trace-bin-propagator branch from 92de02a to 9a9336b Compare September 27, 2024 17:21
@purnesh42H purnesh42H force-pushed the otel-tracing-grpc-trace-bin-propagator branch 2 times, most recently from d0ddcc0 to 5e524a8 Compare September 30, 2024 08:27
@purnesh42H purnesh42H added the Type: Feature New features or improvements in behavior label Sep 30, 2024
@purnesh42H purnesh42H added this to the 1.68 Release milestone Sep 30, 2024
@purnesh42H purnesh42H added the Area: Observability Includes Stats, Tracing, Channelz, Healthz, Binlog, Reflection, Admin, GCP Observability label Sep 30, 2024
@dfawley dfawley removed their assignment Oct 1, 2024
@dfawley
Copy link
Member

dfawley commented Oct 1, 2024

I'll let @zasweq do the primary review here and just take a 2nd pass.

Copy link
Contributor

@zasweq zasweq left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Some comments on implementation; didn't review tests yet. My next pass will be more on correctness, this pass was more general Go style/API and docstring semantics.

stats/opentelemetry/internal/grpc_trace_bin_propagator.go Outdated Show resolved Hide resolved
Comment on lines 32 to 34
// GRPCTraceBinPropagator is TextMapPropagator to propagate cross-cutting
// concerns as both text and binary key-value pairs within a carrier that
// travels in-band across process boundaries.
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Here and elsewhere: I don't know what these comments semantically refer to. What does "cross-cutting concerns" mean? I don't think this is a strict requirement "within a carrier that travels in-band across process boundaries."

Copy link
Contributor Author

@purnesh42H purnesh42H Oct 6, 2024

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

both "cross-cutting concerns" and "within a carrier that travels in-band across process boundaries." are taken directly from OpenTelemetry propagation.TextMapCarrier interface https://pkg.go.dev/go.opentelemetry.io/otel/propagation#TextMapPropagator.

"cross-cutting concerns" simply refers to functionalities that are not central to the core logic of your application but are needed across different parts of it (eg. tracing, logging etc.) which are transported "in-band" (within the main data flow) across process boundaries (separate applications).

However, you are right to point out that we should keep the docstrings simplified for anyone to understand the purpose of the method. I have simplified the comments everwhere now. Let me know what you think.

stats/opentelemetry/internal/grpc_trace_bin_propagator.go Outdated Show resolved Hide resolved
stats/opentelemetry/internal/grpc_trace_bin_propagator.go Outdated Show resolved Hide resolved
stats/opentelemetry/internal/grpc_trace_bin_propagator.go Outdated Show resolved Hide resolved
stats/opentelemetry/internal/tracing/custom_carrier.go Outdated Show resolved Hide resolved
stats/opentelemetry/internal/tracing/custom_carrier.go Outdated Show resolved Hide resolved
func (c CustomCarrier) GetBinary() ([]byte, error) {
values := stats.Trace(*c.Ctx)
if len(values) == 0 {
return nil, errors.New("`grpc-trace-bin` header not found")
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Is this the correct error message? I think both in the case it's not present in md and if it's set to an empty byte string it'll be not found. Is an empty byte string distinct from nil in the return? Is the empty byte string a valid thing to transmit for this header? Do we want this to return an error at all?

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Removed returning error from here to be consistent with Get(..)

Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I'm fine with the error. What is the intended usage of this function? Do you intended to treat not set and set to an empty string equivalent? Is empty string a valid key for this? In that case you should make the error message fail with that. But that relates to the intended usage of this, and the failure modes (i.e. what happens in the error case).

Copy link
Contributor Author

@purnesh42H purnesh42H Oct 9, 2024

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Here we are only looking trace context which needs to conform to following struct. So the propagator will not allow to set any such value (like empty) as it won't be a valid SpanContext. Hence, the only reason for returning nil is client didn't set the grpc-trace-bin header.

// Fields:
        //
        // TraceId: (field_id = 0, len = 16, default = "0000000000000000") -    
        // 16-byte array representing the trace_id.
        //
        // SpanId: (field_id = 1, len = 8, default = "00000000") - 8-byte array   
        // representing the span_id.
        //
        // TraceOptions: (field_id = 2, len = 1, default = "0") - 1-byte array 
           representing the trace_options.

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Is empty string a valid key for this?

Binary methods will not deal with string keys because we are using existing stats package

func SetTrace(ctx context.Context, b []byte) context.Context {
to set bytes which sets a custom struct key


// SetBinary sets the binary value to the gRPC context, which will be sent in
// the outgoing RPC with the header grpc-trace-bin.
func (c CustomCarrier) SetBinary(value []byte) {
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I don't get the point of this API. It seems to do things by deferring to operations on a context either to the stats package or the metadata package? What is the function of this API?

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

yeah this requires a bit more background on migration path from opencensus to opentelemetry (see A72).

The design proposes that while gRPC OpenCensus directly interacts with metadata API, gRPC Open Telemetry will use standardized https://pkg.go.dev/go.opentelemetry.io/otel/propagation package for context propagation by encoding them in metadata for the following benefits:

  • Full integration with OpenTelemetry APIs that is easier for users to reason about.
  • Make it possible to plugin other propagators that the community supports.
  • Flexible API that allows clean and simple migration paths to a different propagator.

As of today, OpenTelemetry propagator API only supports https://pkg.go.dev/go.opentelemetry.io/otel/propagation#TextMapPropagator, that is to send string key/value pairs between the client and server, which is different from the binary header that gRPC currently uses. The future roadmap to support binary propagators at OpenTelemetry is unclear. So, gRPC will use propagator API in TextMap format with optimization path to work around the binary propagator API. Once the Open Telemetry binary propagator API is available in the future, we can continuously integrate with those API with little effort.

Therefore, we need a custom implementation for the Carrier that supports both binary and text values. For binary header grpc-trace-bin we are reusing the existing stats package to conform with existing grpc design.

stats/opentelemetry/internal/tracing/custom_carrier.go Outdated Show resolved Hide resolved
@zasweq zasweq assigned purnesh42H and unassigned zasweq Oct 2, 2024
@purnesh42H purnesh42H requested a review from zasweq October 6, 2024 18:22
@purnesh42H purnesh42H assigned zasweq and unassigned purnesh42H Oct 6, 2024
Comment on lines 53 to 54
// context. It returns an empty string if the key is not present in the
// context.
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

It also returns empty string if it's set in the context but has no values. But I don't know if this is the right behavior, should we return an error? What is the intended usage of this?

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Actually, I replied above the reason why empty value is not possible. We check if the SpanContext is valid in propagator's Inject() method before setting in carrier

span := oteltrace.SpanFromContext(ctx)
	if !span.SpanContext().IsValid() {
		return
	}
	```

Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Please add to documentation the case if it's set in context but has no values. We don't assume this operation comes after that in the callsite, this is a documentation for this layer/operation.

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Added

Comment on lines 54 to 84
t.Run("Fast path with CustomCarrier", func(t *testing.T) {
ctx, cancel := context.WithCancel(context.Background())
carrier := internaltracing.NewCustomCarrier(metadata.NewOutgoingContext(ctx, metadata.MD{}))
propagator.Inject(traceCtx, carrier)

got := stats.OutgoingTrace(*carrier.Context())
want := Binary(spanContext)
if string(got) != string(want) {
t.Fatalf("got = %v, want %v", got, want)
}
cancel()
})

t.Run("Slow path with TextMapCarrier", func(t *testing.T) {
carrier := otelpropagation.MapCarrier{}
propagator.Inject(traceCtx, carrier)

got := carrier.Get(internaltracing.GRPCTraceBinHeaderKey)
want := base64.StdEncoding.EncodeToString(Binary(spanContext))
if got != want {
t.Fatalf("got = %v, want %v", got, want)
}
})
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

yeah i had tabular test before but both tests have different veriffication logic so I wrote in this way

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

yeah i had them as tabular test before but verification logic is significantly different because of binary and string format so i wrote it this way

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I saw there are some existing tests written in similar way

Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

You can declare the want in the t-test as a variable.

Copy link
Contributor Author

@purnesh42H purnesh42H Oct 13, 2024

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

If you see logic to retrieve got is different as well. For fast path which will use SetBinary we get it through stats package because we are using CustomCarrier where as for slow path which will use Set, we just retrieve directly using carrier's Get because we are using a different carrier (not implemented by us). Would you prefer to have separate tests altogether like TestInject_FastPath, TestInject_SlowPath?

Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Yeah separate tests are preferred over t.Run(). Thanks.

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Separated the fast and slow path tests

}
cancel()
})
}
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Same comments on this test as above.

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Done

@purnesh42H purnesh42H assigned dfawley and unassigned purnesh42H Nov 20, 2024
@purnesh42H purnesh42H force-pushed the otel-tracing-grpc-trace-bin-propagator branch from 02196c9 to d708a8b Compare November 20, 2024 19:30
stats/opentelemetry/grpc_trace_bin_propagator_test.go Outdated Show resolved Hide resolved
stats/opentelemetry/grpc_trace_bin_propagator_test.go Outdated Show resolved Hide resolved
stats/opentelemetry/grpc_trace_bin_propagator_test.go Outdated Show resolved Hide resolved
stats/opentelemetry/grpc_trace_bin_propagator_test.go Outdated Show resolved Hide resolved
stats/opentelemetry/grpc_trace_bin_propagator_test.go Outdated Show resolved Hide resolved
stats/opentelemetry/grpc_trace_bin_propagator_test.go Outdated Show resolved Hide resolved
stats/opentelemetry/internal/tracing/carrier.go Outdated Show resolved Hide resolved
@dfawley dfawley assigned purnesh42H and unassigned dfawley Nov 20, 2024
@purnesh42H purnesh42H force-pushed the otel-tracing-grpc-trace-bin-propagator branch from 1e22ed3 to 0c80222 Compare November 21, 2024 10:01
@purnesh42H purnesh42H requested a review from dfawley November 21, 2024 10:02
@purnesh42H purnesh42H assigned dfawley and unassigned purnesh42H Nov 21, 2024
@purnesh42H purnesh42H force-pushed the otel-tracing-grpc-trace-bin-propagator branch from 0c80222 to a3c8693 Compare November 21, 2024 14:34
@purnesh42H purnesh42H requested a review from aranjans November 21, 2024 16:40
@dfawley dfawley assigned purnesh42H and unassigned dfawley Nov 22, 2024
@purnesh42H purnesh42H requested a review from dfawley November 22, 2024 17:10
@purnesh42H purnesh42H assigned dfawley and unassigned purnesh42H Nov 22, 2024
Comment on lines 72 to 77
if !test.injectSC.IsValid() {
if len(gotH) > 0 {
t.Fatalf("got %v value from Carrier's context metadata grpc-trace-bin header, want empty", gotH)
}
return
}
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

When your test cases diverge in behavior so much, it really just feels like two different tests instead of one table driven test. Optionally rewrite that way to make it easier to understand.

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Wrote separate tests for valid and invalid span context injection

@dfawley dfawley assigned purnesh42H and unassigned dfawley Nov 22, 2024
@purnesh42H purnesh42H merged commit 8b70aeb into grpc:master Nov 25, 2024
14 of 15 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
Area: Observability Includes Stats, Tracing, Channelz, Healthz, Binlog, Reflection, Admin, GCP Observability Type: Feature New features or improvements in behavior
Projects
None yet
Development

Successfully merging this pull request may close these issues.

4 participants