Skip to content

Commit

Permalink
Address feedback and add documentation.
Browse files Browse the repository at this point in the history
  • Loading branch information
Tyler Yahn committed Jan 2, 2020
1 parent cb93e28 commit f111547
Showing 1 changed file with 50 additions and 10 deletions.
60 changes: 50 additions & 10 deletions api/global/internal/trace.go
Original file line number Diff line number Diff line change
@@ -1,28 +1,56 @@
package internal

/*
This file contains the forwarding implementation of the trace.Provider used as
the default global instance. Prior to initialization of an SDK, Tracers
returned by the global Provider will provide no-op functionality. This means
that all Span created prior to initialization are no-op Spans.
Once an SDK has been initialized, all provided no-op Tracers are swapped for
Tracers provided by the SDK defined Provider. However, any Span started prior
to this initialization does not change its behavior. Meaning, the Span remains
a no-op Span.
The implementation to track and swap Tracers locks all new Tracer creation
until the swap is complete. This assumes that this operation is not
performance-critical. If that assumption is incorrect, be sure to configure an
SDK prior to any Tracer creation.
*/

import (
"context"
"sync"

"go.opentelemetry.io/otel/api/trace"
)

// traceProvider is a placeholder for a configured SDK Provider.
//
// All Provider functionality is forwarded to a delegate once configured.
type traceProvider struct {
lock sync.Mutex
mtx sync.Mutex
tracers []*tracer

delegate trace.Provider
}

// Compile time guarantee that traceProvider implements the trace.Provider interface.
var _ trace.Provider = &traceProvider{}

// setDelegate configures p to delegate all Provider functionality to provider.
//
// All Tracers provided prior to this function call are switched out to be
// Tracers provided by provider.
//
// Delegation only happens on the first call to this method. All subsequent
// calls result in no delegation changes.
func (p *traceProvider) setDelegate(provider trace.Provider) {
if p.delegate != nil {
return
}

p.lock.Lock()
defer p.lock.Unlock()
p.mtx.Lock()
defer p.mtx.Unlock()

p.delegate = provider
for _, t := range p.tracers {
Expand All @@ -32,12 +60,10 @@ func (p *traceProvider) setDelegate(provider trace.Provider) {
p.tracers = nil
}

// Tracer creates a trace.Tracer with the given name. When a delegate is set,
// all previously returned trace.Tracers will be swapped to equivalent
// trace.Tracers created from the delegate.
// Tracer implements trace.Provider.
func (p *traceProvider) Tracer(name string) trace.Tracer {
p.lock.Lock()
defer p.lock.Unlock()
p.mtx.Lock()
defer p.mtx.Unlock()

if p.delegate != nil {
return p.delegate.Tracer(name)
Expand All @@ -48,28 +74,42 @@ func (p *traceProvider) Tracer(name string) trace.Tracer {
return t
}

// tracer is a placeholder for a trace.Tracer.
//
// All Tracer functionality is forwarded to a delegate once configured.
// Otherwise, all functionality is forwarded to a NoopTracer.
type tracer struct {
once sync.Once
name string

delegate trace.Tracer
}

// Compile time guarantee that tracer implements the trace.Tracer interface.
var _ trace.Tracer = &tracer{}

// setDelegate configures t to delegate all Tracer functionality to Tracers
// created by provider.
//
// All subsequent calls to the Tracer methods will be passed to the delegate.
//
// Delegation only happens on the first call to this method. All subsequent
// calls result in no delegation changes.
func (t *tracer) setDelegate(provider trace.Provider) {
t.once.Do(func() { t.delegate = provider.Tracer(t.name) })
}

// WithSpan wraps around execution of func with delegated Tracer.
// WithSpan implements trace.Tracer by forwarding the call to t.delegate if
// set, otherwise it forwards the call to a NoopTracer.
func (t *tracer) WithSpan(ctx context.Context, name string, body func(context.Context) error) error {
if t.delegate != nil {
return t.delegate.WithSpan(ctx, name, body)
}
return trace.NoopTracer{}.WithSpan(ctx, name, body)
}

// Start starts a span from the delegated tracer.
// Start implements trace.Tracer by forwarding the call to t.delegate if
// set, otherwise it forwards the call to a NoopTracer.
func (t *tracer) Start(ctx context.Context, name string, opts ...trace.StartOption) (context.Context, trace.Span) {
if t.delegate != nil {
return t.delegate.Start(ctx, name, opts...)
Expand Down

0 comments on commit f111547

Please sign in to comment.