2019-12-24 09:03:04 +02:00
|
|
|
package internal
|
|
|
|
|
|
|
|
import (
|
|
|
|
"sync"
|
|
|
|
"sync/atomic"
|
|
|
|
|
|
|
|
"go.opentelemetry.io/otel/api/metric"
|
|
|
|
"go.opentelemetry.io/otel/api/trace"
|
|
|
|
)
|
|
|
|
|
|
|
|
type (
|
|
|
|
traceProviderHolder struct {
|
|
|
|
tp trace.Provider
|
|
|
|
}
|
|
|
|
|
|
|
|
meterProviderHolder struct {
|
|
|
|
mp metric.Provider
|
|
|
|
}
|
|
|
|
)
|
|
|
|
|
|
|
|
var (
|
|
|
|
globalTracer = defaultTracerValue()
|
|
|
|
globalMeter = defaultMeterValue()
|
|
|
|
|
|
|
|
delegateMeterOnce sync.Once
|
2020-01-02 23:20:38 +02:00
|
|
|
delegateTraceOnce sync.Once
|
2019-12-24 09:03:04 +02:00
|
|
|
)
|
|
|
|
|
|
|
|
// TraceProvider is the internal implementation for global.TraceProvider.
|
|
|
|
func TraceProvider() trace.Provider {
|
|
|
|
return globalTracer.Load().(traceProviderHolder).tp
|
|
|
|
}
|
|
|
|
|
|
|
|
// SetTraceProvider is the internal implementation for global.SetTraceProvider.
|
|
|
|
func SetTraceProvider(tp trace.Provider) {
|
2020-01-02 23:20:38 +02:00
|
|
|
delegateTraceOnce.Do(func() {
|
|
|
|
current := TraceProvider()
|
|
|
|
if current == tp {
|
|
|
|
// Setting the provider to the prior default is nonsense, panic.
|
|
|
|
// Panic is acceptable because we are likely still early in the
|
|
|
|
// process lifetime.
|
|
|
|
panic("invalid Provider, the global instance cannot be reinstalled")
|
|
|
|
} else if def, ok := current.(*traceProvider); ok {
|
|
|
|
def.setDelegate(tp)
|
|
|
|
}
|
|
|
|
|
|
|
|
})
|
2019-12-24 09:03:04 +02:00
|
|
|
globalTracer.Store(traceProviderHolder{tp: tp})
|
|
|
|
}
|
|
|
|
|
|
|
|
// MeterProvider is the internal implementation for global.MeterProvider.
|
|
|
|
func MeterProvider() metric.Provider {
|
|
|
|
return globalMeter.Load().(meterProviderHolder).mp
|
|
|
|
}
|
|
|
|
|
|
|
|
// SetMeterProvider is the internal implementation for global.SetMeterProvider.
|
|
|
|
func SetMeterProvider(mp metric.Provider) {
|
|
|
|
delegateMeterOnce.Do(func() {
|
|
|
|
current := MeterProvider()
|
|
|
|
|
|
|
|
if current == mp {
|
|
|
|
// Setting the provider to the prior default is nonsense, panic.
|
|
|
|
// Panic is acceptable because we are likely still early in the
|
|
|
|
// process lifetime.
|
|
|
|
panic("invalid Provider, the global instance cannot be reinstalled")
|
|
|
|
} else if def, ok := current.(*meterProvider); ok {
|
|
|
|
def.setDelegate(mp)
|
|
|
|
}
|
|
|
|
})
|
|
|
|
globalMeter.Store(meterProviderHolder{mp: mp})
|
|
|
|
}
|
|
|
|
|
|
|
|
func defaultTracerValue() *atomic.Value {
|
|
|
|
v := &atomic.Value{}
|
2020-01-02 23:20:38 +02:00
|
|
|
v.Store(traceProviderHolder{tp: &traceProvider{}})
|
2019-12-24 09:03:04 +02:00
|
|
|
return v
|
|
|
|
}
|
|
|
|
|
|
|
|
func defaultMeterValue() *atomic.Value {
|
|
|
|
v := &atomic.Value{}
|
|
|
|
v.Store(meterProviderHolder{mp: &meterProvider{}})
|
|
|
|
return v
|
|
|
|
}
|
|
|
|
|
|
|
|
// ResetForTest restores the initial global state, for testing purposes.
|
|
|
|
func ResetForTest() {
|
|
|
|
globalTracer = defaultTracerValue()
|
|
|
|
globalMeter = defaultMeterValue()
|
|
|
|
delegateMeterOnce = sync.Once{}
|
2020-01-02 23:20:38 +02:00
|
|
|
delegateTraceOnce = sync.Once{}
|
2019-12-24 09:03:04 +02:00
|
|
|
}
|