1
0
mirror of https://github.com/open-telemetry/opentelemetry-go.git synced 2025-03-21 21:17:35 +02:00

Enable golint & gofmt, resolve issues (#214)

* Add golint to linters and resolve issues.

I decided to remove constructors for some of the propagation types
because the constructors can be reduced to either using the zero value
or a single, non optional member.

* Enable gofmt and commit fixes
This commit is contained in:
Edward Muller 2019-10-16 10:24:38 -07:00 committed by rghetia
parent 0fd077287b
commit 9f03360a84
22 changed files with 105 additions and 99 deletions

View File

@ -7,6 +7,21 @@ linters:
enable: enable:
- misspell - misspell
- goimports - goimports
- golint
- gofmt
issues:
exclude-rules:
# helpers in tests often (rightfully) pass a *testing.T as their first argument
- path: _test\.go
text: "context.Context should be the first parameter of a function"
linters:
- golint
# Yes, they are, but it's okay in a test
- path: _test\.go
text: "exported func.*returns unexported type.*which can be annoying to use"
linters:
- golint
linters-settings: linters-settings:
misspell: misspell:

View File

@ -661,6 +661,7 @@ func TestObserver(t *testing.T) {
} }
func checkBatches(t *testing.T, ctx context.Context, labels LabelSet, meter *mockMeter, descriptor *Descriptor) { func checkBatches(t *testing.T, ctx context.Context, labels LabelSet, meter *mockMeter, descriptor *Descriptor) {
t.Helper()
if len(meter.measurementBatches) != 3 { if len(meter.measurementBatches) != 3 {
t.Errorf("Expected 3 recorded measurement batches, got %d", len(meter.measurementBatches)) t.Errorf("Expected 3 recorded measurement batches, got %d", len(meter.measurementBatches))
} }

View File

@ -47,6 +47,6 @@ func (as alwaysSampleSampler) Description() string {
var _ Sampler = alwaysSampleSampler{} var _ Sampler = alwaysSampleSampler{}
func AlwaysSampleSampler() alwaysSampleSampler { func AlwaysSampleSampler() Sampler {
return alwaysSampleSampler{} return alwaysSampleSampler{}
} }

View File

@ -47,6 +47,6 @@ func (ns neverSampleSampler) Description() string {
var _ Sampler = neverSampleSampler{} var _ Sampler = neverSampleSampler{}
func NeverSampleSampler() neverSampleSampler { func NeverSampleSampler() Sampler {
return neverSampleSampler{} return neverSampleSampler{}
} }

View File

@ -250,15 +250,15 @@ type coin3Value struct{}
func newContextIntactTest() *contextIntactTest { func newContextIntactTest() *contextIntactTest {
return &contextIntactTest{ return &contextIntactTest{
contextKeyValues: []internal.MockContextKeyValue{ contextKeyValues: []internal.MockContextKeyValue{
internal.MockContextKeyValue{ {
Key: coin1Key{}, Key: coin1Key{},
Value: coin1Value{}, Value: coin1Value{},
}, },
internal.MockContextKeyValue{ {
Key: coin2Key{}, Key: coin2Key{},
Value: coin2Value{}, Value: coin2Value{},
}, },
internal.MockContextKeyValue{ {
Key: coin3Key{}, Key: coin3Key{},
Value: coin3Value{}, Value: coin3Value{},
}, },

View File

@ -51,6 +51,7 @@ type Observer interface {
} }
//go:generate stringer -type=EventType //go:generate stringer -type=EventType
// nolint:golint
const ( const (
INVALID EventType = iota INVALID EventType = iota
START_SPAN START_SPAN

View File

@ -153,8 +153,8 @@ func measurementCompare(m1, m2 metric.Measurement) bool {
func diffEvents(t *testing.T, got, want []exporter.Event, extraIgnoredFields ...string) bool { func diffEvents(t *testing.T, got, want []exporter.Event, extraIgnoredFields ...string) bool {
ignoredPaths := map[string]struct{}{ ignoredPaths := map[string]struct{}{
"Sequence": struct{}{}, "Sequence": {},
"Context": struct{}{}, "Context": {},
} }
for _, field := range extraIgnoredFields { for _, field := range extraIgnoredFields {
ignoredPaths[field] = struct{}{} ignoredPaths[field] = struct{}{}

View File

@ -24,8 +24,8 @@ import (
"go.opentelemetry.io/experimental/streaming/exporter" "go.opentelemetry.io/experimental/streaming/exporter"
) )
// TODO These should move somewhere in the api, right?
var ( var (
// TODO These should move somewhere in the api, right?
ErrorKey = key.New("error") ErrorKey = key.New("error")
SpanIDKey = key.New("span_id") SpanIDKey = key.New("span_id")
TraceIDKey = key.New("trace_id") TraceIDKey = key.New("trace_id")

View File

@ -61,7 +61,7 @@ func Test_spanDataToThrift(t *testing.T) {
StartTime: now, StartTime: now,
EndTime: now, EndTime: now,
Links: []apitrace.Link{ Links: []apitrace.Link{
apitrace.Link{ {
SpanContext: core.SpanContext{ SpanContext: core.SpanContext{
TraceID: linkTraceID, TraceID: linkTraceID,
SpanID: linkSpanID, SpanID: linkSpanID,
@ -96,7 +96,7 @@ func Test_spanDataToThrift(t *testing.T) {
{Key: "status.message", VType: gen.TagType_STRING, VStr: &statusMessage}, {Key: "status.message", VType: gen.TagType_STRING, VStr: &statusMessage},
}, },
References: []*gen.SpanRef{ References: []*gen.SpanRef{
&gen.SpanRef{ {
RefType: gen.SpanRefType_CHILD_OF, RefType: gen.SpanRefType_CHILD_OF,
TraceIdLow: int64(linkTraceID.Low), TraceIdLow: int64(linkTraceID.Low),
TraceIdHigh: int64(linkTraceID.High), TraceIdHigh: int64(linkTraceID.High),

View File

@ -25,7 +25,7 @@ type EndpointOption func() (batchUploader, error)
func WithAgentEndpoint(agentEndpoint string) func() (batchUploader, error) { func WithAgentEndpoint(agentEndpoint string) func() (batchUploader, error) {
return func() (batchUploader, error) { return func() (batchUploader, error) {
if agentEndpoint == "" { if agentEndpoint == "" {
return nil, errors.New("agentEndpoint must not be empty.") return nil, errors.New("agentEndpoint must not be empty")
} }
client, err := newAgentClientUDP(agentEndpoint, udpPacketMaxLength) client, err := newAgentClientUDP(agentEndpoint, udpPacketMaxLength)
@ -42,7 +42,7 @@ func WithAgentEndpoint(agentEndpoint string) func() (batchUploader, error) {
func WithCollectorEndpoint(collectorEndpoint string, options ...CollectorEndpointOption) func() (batchUploader, error) { func WithCollectorEndpoint(collectorEndpoint string, options ...CollectorEndpointOption) func() (batchUploader, error) {
return func() (batchUploader, error) { return func() (batchUploader, error) {
if collectorEndpoint == "" { if collectorEndpoint == "" {
return nil, errors.New("collectorEndpoint must not be empty.") return nil, errors.New("collectorEndpoint must not be empty")
} }
o := &CollectorEndpointOptions{} o := &CollectorEndpointOptions{}

View File

@ -30,9 +30,9 @@ type MockTracer struct {
// Sampled specifies if the new span should be sampled or not. // Sampled specifies if the new span should be sampled or not.
Sampled bool Sampled bool
// StartSpanId is used to initialize spanId. It is incremented by one // StartSpanID is used to initialize spanId. It is incremented by one
// every time a new span is created. // every time a new span is created.
StartSpanId *uint64 StartSpanID *uint64
} }
var _ apitrace.Tracer = (*MockTracer)(nil) var _ apitrace.Tracer = (*MockTracer)(nil)
@ -81,7 +81,7 @@ func (mt *MockTracer) Start(ctx context.Context, name string, o ...apitrace.Span
} else { } else {
sc = opts.Reference.SpanContext sc = opts.Reference.SpanContext
} }
sc.SpanID = atomic.AddUint64(mt.StartSpanId, 1) sc.SpanID = atomic.AddUint64(mt.StartSpanID, 1)
span = &MockSpan{ span = &MockSpan{
sc: sc, sc: sc,
tracer: mt, tracer: mt,

View File

@ -31,7 +31,7 @@ var (
HostKey = key.New("http.host") HostKey = key.New("http.host")
URLKey = key.New("http.url") URLKey = key.New("http.url")
propagator = propagation.HttpTraceContextPropagator() propagator = propagation.HTTPTraceContextPropagator{}
) )
// Returns the Attributes, Context Entries, and SpanContext that were encoded by Inject. // Returns the Attributes, Context Entries, and SpanContext that were encoded by Inject.

View File

@ -117,7 +117,7 @@ func NewHandler(handler http.Handler, operation string, opts ...HandlerOption) h
h := HTTPHandler{handler: handler} h := HTTPHandler{handler: handler}
defaultOpts := []HandlerOption{ defaultOpts := []HandlerOption{
WithTracer(trace.GlobalTracer()), WithTracer(trace.GlobalTracer()),
WithPropagator(prop.HttpTraceContextPropagator()), WithPropagator(prop.HTTPTraceContextPropagator{}),
} }
for _, opt := range append(defaultOpts, opts...) { for _, opt := range append(defaultOpts, opts...) {

View File

@ -29,7 +29,7 @@ func TestBasics(t *testing.T) {
rr := httptest.NewRecorder() rr := httptest.NewRecorder()
var id uint64 var id uint64
tracer := mocktrace.MockTracer{StartSpanId: &id} tracer := mocktrace.MockTracer{StartSpanID: &id}
h := NewHandler( h := NewHandler(
http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {

View File

@ -27,7 +27,7 @@ var _ apipropagation.BinaryFormatPropagator = binaryPropagator{}
// BinaryPropagator creates a new propagator. The propagator implements // BinaryPropagator creates a new propagator. The propagator implements
// ToBytes and FromBytes method to transform SpanContext to/from byte array. // ToBytes and FromBytes method to transform SpanContext to/from byte array.
func BinaryPropagator() binaryPropagator { func BinaryPropagator() apipropagation.BinaryFormatPropagator {
return binaryPropagator{} return binaryPropagator{}
} }

View File

@ -36,16 +36,7 @@ const (
B3ParentSpanIDHeader = "X-B3-ParentSpanId" B3ParentSpanIDHeader = "X-B3-ParentSpanId"
) )
type httpB3Propagator struct { // HTTPB3Propagator that facilitates core.SpanContext
singleHeader bool
}
var _ apipropagation.TextFormatPropagator = httpB3Propagator{}
var hexStr32ByteRegex = regexp.MustCompile("^[a-f0-9]{32}$")
var hexStr16ByteRegex = regexp.MustCompile("^[a-f0-9]{16}$")
// HttpB3Propagator creates a new text format propagator that facilitates core.SpanContext
// propagation using B3 Headers. // propagation using B3 Headers.
// This propagator supports both version of B3 headers, // This propagator supports both version of B3 headers,
// 1. Single Header : // 1. Single Header :
@ -57,18 +48,21 @@ var hexStr16ByteRegex = regexp.MustCompile("^[a-f0-9]{16}$")
// X-B3-Sampled: {SamplingState} // X-B3-Sampled: {SamplingState}
// X-B3-Flags: {DebugFlag} // X-B3-Flags: {DebugFlag}
// //
// If singleHeader is set to true then X-B3 header is used to inject and extract. Otherwise, // If SingleHeader is set to true then X-B3 header is used to inject and extract. Otherwise,
// separate headers are used to inject and extract. // separate headers are used to inject and extract.
func HttpB3Propagator(singleHeader bool) httpB3Propagator { type HTTPB3Propagator struct {
// [TODO](rghetia): should it automatically look for both versions? which one to pick if SingleHeader bool
// both are present? What if one is valid and other one is not.
return httpB3Propagator{singleHeader}
} }
func (b3 httpB3Propagator) Inject(ctx context.Context, supplier apipropagation.Supplier) { var _ apipropagation.TextFormatPropagator = HTTPB3Propagator{}
var hexStr32ByteRegex = regexp.MustCompile("^[a-f0-9]{32}$")
var hexStr16ByteRegex = regexp.MustCompile("^[a-f0-9]{16}$")
func (b3 HTTPB3Propagator) Inject(ctx context.Context, supplier apipropagation.Supplier) {
sc := trace.CurrentSpan(ctx).SpanContext() sc := trace.CurrentSpan(ctx).SpanContext()
if sc.IsValid() { if sc.IsValid() {
if b3.singleHeader { if b3.SingleHeader {
sampled := sc.TraceFlags & core.TraceFlagsSampled sampled := sc.TraceFlags & core.TraceFlagsSampled
supplier.Set(B3SingleHeader, supplier.Set(B3SingleHeader,
fmt.Sprintf("%.16x%.16x-%.16x-%.1d", sc.TraceID.High, sc.TraceID.Low, sc.SpanID, sampled)) fmt.Sprintf("%.16x%.16x-%.16x-%.1d", sc.TraceID.High, sc.TraceID.Low, sc.SpanID, sampled))
@ -90,21 +84,21 @@ func (b3 httpB3Propagator) Inject(ctx context.Context, supplier apipropagation.S
} }
// Extract retrieves B3 Headers from the supplier // Extract retrieves B3 Headers from the supplier
func (b3 httpB3Propagator) Extract(ctx context.Context, supplier apipropagation.Supplier) core.SpanContext { func (b3 HTTPB3Propagator) Extract(ctx context.Context, supplier apipropagation.Supplier) core.SpanContext {
if b3.singleHeader { if b3.SingleHeader {
return b3.extractSingleHeader(supplier) return b3.extractSingleHeader(supplier)
} }
return b3.extract(supplier) return b3.extract(supplier)
} }
func (b3 httpB3Propagator) GetAllKeys() []string { func (b3 HTTPB3Propagator) GetAllKeys() []string {
if b3.singleHeader { if b3.SingleHeader {
return []string{B3SingleHeader} return []string{B3SingleHeader}
} }
return []string{B3TraceIDHeader, B3SpanIDHeader, B3SampledHeader} return []string{B3TraceIDHeader, B3SpanIDHeader, B3SampledHeader}
} }
func (b3 httpB3Propagator) extract(supplier apipropagation.Supplier) core.SpanContext { func (b3 HTTPB3Propagator) extract(supplier apipropagation.Supplier) core.SpanContext {
tid, ok := b3.extractTraceID(supplier.Get(B3TraceIDHeader)) tid, ok := b3.extractTraceID(supplier.Get(B3TraceIDHeader))
if !ok { if !ok {
return core.EmptySpanContext() return core.EmptySpanContext()
@ -139,7 +133,7 @@ func (b3 httpB3Propagator) extract(supplier apipropagation.Supplier) core.SpanCo
return sc return sc
} }
func (b3 httpB3Propagator) extractSingleHeader(supplier apipropagation.Supplier) core.SpanContext { func (b3 HTTPB3Propagator) extractSingleHeader(supplier apipropagation.Supplier) core.SpanContext {
h := supplier.Get(B3SingleHeader) h := supplier.Get(B3SingleHeader)
if h == "" || h == "0" { if h == "" || h == "0" {
core.EmptySpanContext() core.EmptySpanContext()
@ -153,30 +147,30 @@ func (b3 httpB3Propagator) extractSingleHeader(supplier apipropagation.Supplier)
if l < 2 { if l < 2 {
return core.EmptySpanContext() return core.EmptySpanContext()
} else { }
var ok bool
sc.TraceID, ok = b3.extractTraceID(parts[0]) var ok bool
sc.TraceID, ok = b3.extractTraceID(parts[0])
if !ok {
return core.EmptySpanContext()
}
sc.SpanID, ok = b3.extractSpanID(parts[1])
if !ok {
return core.EmptySpanContext()
}
if l > 2 {
sc.TraceFlags, ok = b3.extractSampledState(parts[2])
if !ok { if !ok {
return core.EmptySpanContext() return core.EmptySpanContext()
} }
}
sc.SpanID, ok = b3.extractSpanID(parts[1]) if l == 4 {
_, ok = b3.extractSpanID(parts[3])
if !ok { if !ok {
return core.EmptySpanContext() return core.EmptySpanContext()
} }
if l > 2 {
sc.TraceFlags, ok = b3.extractSampledState(parts[2])
if !ok {
return core.EmptySpanContext()
}
}
if l == 4 {
_, ok = b3.extractSpanID(parts[3])
if !ok {
return core.EmptySpanContext()
}
}
} }
if !sc.IsValid() { if !sc.IsValid() {
@ -187,12 +181,12 @@ func (b3 httpB3Propagator) extractSingleHeader(supplier apipropagation.Supplier)
} }
// extractTraceID parses the value of the X-B3-TraceId b3Header. // extractTraceID parses the value of the X-B3-TraceId b3Header.
func (b3 httpB3Propagator) extractTraceID(tid string) (traceID core.TraceID, ok bool) { func (b3 HTTPB3Propagator) extractTraceID(tid string) (traceID core.TraceID, ok bool) {
if hexStr32ByteRegex.MatchString(tid) { if hexStr32ByteRegex.MatchString(tid) {
traceID.High, _ = strconv.ParseUint(tid[0:(16)], 16, 64) traceID.High, _ = strconv.ParseUint(tid[0:(16)], 16, 64)
traceID.Low, _ = strconv.ParseUint(tid[(16):32], 16, 64) traceID.Low, _ = strconv.ParseUint(tid[(16):32], 16, 64)
ok = true ok = true
} else if b3.singleHeader && hexStr16ByteRegex.MatchString(tid) { } else if b3.SingleHeader && hexStr16ByteRegex.MatchString(tid) {
traceID.Low, _ = strconv.ParseUint(tid[:16], 16, 64) traceID.Low, _ = strconv.ParseUint(tid[:16], 16, 64)
ok = true ok = true
} }
@ -200,7 +194,7 @@ func (b3 httpB3Propagator) extractTraceID(tid string) (traceID core.TraceID, ok
} }
// extractSpanID parses the value of the X-B3-SpanId or X-B3-ParentSpanId headers. // extractSpanID parses the value of the X-B3-SpanId or X-B3-ParentSpanId headers.
func (b3 httpB3Propagator) extractSpanID(sid string) (spanID uint64, ok bool) { func (b3 HTTPB3Propagator) extractSpanID(sid string) (spanID uint64, ok bool) {
if hexStr16ByteRegex.MatchString(sid) { if hexStr16ByteRegex.MatchString(sid) {
spanID, _ = strconv.ParseUint(sid, 16, 64) spanID, _ = strconv.ParseUint(sid, 16, 64)
ok = true ok = true
@ -209,18 +203,18 @@ func (b3 httpB3Propagator) extractSpanID(sid string) (spanID uint64, ok bool) {
} }
// extractSampledState parses the value of the X-B3-Sampled b3Header. // extractSampledState parses the value of the X-B3-Sampled b3Header.
func (b3 httpB3Propagator) extractSampledState(sampled string) (flag byte, ok bool) { func (b3 HTTPB3Propagator) extractSampledState(sampled string) (flag byte, ok bool) {
switch sampled { switch sampled {
case "", "0": case "", "0":
return 0, true return 0, true
case "1": case "1":
return core.TraceFlagsSampled, true return core.TraceFlagsSampled, true
case "true": case "true":
if !b3.singleHeader { if !b3.SingleHeader {
return core.TraceFlagsSampled, true return core.TraceFlagsSampled, true
} }
case "d": case "d":
if b3.singleHeader { if b3.SingleHeader {
return core.TraceFlagsSampled, true return core.TraceFlagsSampled, true
} }
} }
@ -228,7 +222,7 @@ func (b3 httpB3Propagator) extractSampledState(sampled string) (flag byte, ok bo
} }
// extracDebugFlag parses the value of the X-B3-Sampled b3Header. // extracDebugFlag parses the value of the X-B3-Sampled b3Header.
func (b3 httpB3Propagator) extracDebugFlag(debug string) (flag byte, ok bool) { func (b3 HTTPB3Propagator) extracDebugFlag(debug string) (flag byte, ok bool) {
switch debug { switch debug {
case "", "0": case "", "0":
return 0, true return 0, true

View File

@ -54,7 +54,7 @@ func BenchmarkExtractB3(b *testing.B) {
trace.SetGlobalTracer(&mocktrace.MockTracer{}) trace.SetGlobalTracer(&mocktrace.MockTracer{})
for _, tg := range testGroup { for _, tg := range testGroup {
propagator := propagation.HttpB3Propagator(tg.singleHeader) propagator := propagation.HTTPB3Propagator{tg.singleHeader}
for _, tt := range tg.tests { for _, tt := range tg.tests {
traceBenchmark(tg.name+"/"+tt.name, b, func(b *testing.B) { traceBenchmark(tg.name+"/"+tt.name, b, func(b *testing.B) {
ctx := context.Background() ctx := context.Background()
@ -93,13 +93,13 @@ func BenchmarkInjectB3(b *testing.B) {
mockTracer := &mocktrace.MockTracer{ mockTracer := &mocktrace.MockTracer{
Sampled: false, Sampled: false,
StartSpanId: &id, StartSpanID: &id,
} }
trace.SetGlobalTracer(mockTracer) trace.SetGlobalTracer(mockTracer)
for _, tg := range testGroup { for _, tg := range testGroup {
id = 0 id = 0
propagator := propagation.HttpB3Propagator(tg.singleHeader) propagator := propagation.HTTPB3Propagator{tg.singleHeader}
for _, tt := range tg.tests { for _, tt := range tg.tests {
traceBenchmark(tg.name+"/"+tt.name, b, func(b *testing.B) { traceBenchmark(tg.name+"/"+tt.name, b, func(b *testing.B) {
req, _ := http.NewRequest("GET", "http://example.com", nil) req, _ := http.NewRequest("GET", "http://example.com", nil)

View File

@ -56,7 +56,7 @@ func TestExtractB3(t *testing.T) {
trace.SetGlobalTracer(&mocktrace.MockTracer{}) trace.SetGlobalTracer(&mocktrace.MockTracer{})
for _, tg := range testGroup { for _, tg := range testGroup {
propagator := propagation.HttpB3Propagator(tg.singleHeader) propagator := propagation.HTTPB3Propagator{tg.singleHeader}
for _, tt := range tg.tests { for _, tt := range tg.tests {
t.Run(tt.name, func(t *testing.T) { t.Run(tt.name, func(t *testing.T) {
req, _ := http.NewRequest("GET", "http://example.com", nil) req, _ := http.NewRequest("GET", "http://example.com", nil)
@ -95,13 +95,13 @@ func TestInjectB3(t *testing.T) {
mockTracer := &mocktrace.MockTracer{ mockTracer := &mocktrace.MockTracer{
Sampled: false, Sampled: false,
StartSpanId: &id, StartSpanID: &id,
} }
trace.SetGlobalTracer(mockTracer) trace.SetGlobalTracer(mockTracer)
for _, tg := range testGroup { for _, tg := range testGroup {
id = 0 id = 0
propagator := propagation.HttpB3Propagator(tg.singleHeader) propagator := propagation.HTTPB3Propagator{tg.singleHeader}
for _, tt := range tg.tests { for _, tt := range tg.tests {
t.Run(tt.name, func(t *testing.T) { t.Run(tt.name, func(t *testing.T) {
req, _ := http.NewRequest("GET", "http://example.com", nil) req, _ := http.NewRequest("GET", "http://example.com", nil)
@ -132,8 +132,8 @@ func TestInjectB3(t *testing.T) {
} }
} }
func TestHttpB3Propagator_GetAllKeys(t *testing.T) { func TestHTTPB3Propagator_GetAllKeys(t *testing.T) {
propagator := propagation.HttpB3Propagator(false) propagator := propagation.HTTPB3Propagator{false}
want := []string{ want := []string{
propagation.B3TraceIDHeader, propagation.B3TraceIDHeader,
propagation.B3SpanIDHeader, propagation.B3SpanIDHeader,
@ -146,7 +146,7 @@ func TestHttpB3Propagator_GetAllKeys(t *testing.T) {
} }
func TestHttpB3PropagatorWithSingleHeader_GetAllKeys(t *testing.T) { func TestHttpB3PropagatorWithSingleHeader_GetAllKeys(t *testing.T) {
propagator := propagation.HttpB3Propagator(true) propagator := propagation.HTTPB3Propagator{true}
want := []string{ want := []string{
propagation.B3SingleHeader, propagation.B3SingleHeader,
} }

View File

@ -34,12 +34,13 @@ const (
TraceparentHeader = "Traceparent" TraceparentHeader = "Traceparent"
) )
type httpTraceContextPropagator struct{} // HTTPTraceContextPropagator propagates SpanContext in W3C TraceContext format.
type HTTPTraceContextPropagator struct{}
var _ apipropagation.TextFormatPropagator = httpTraceContextPropagator{} var _ apipropagation.TextFormatPropagator = HTTPTraceContextPropagator{}
var traceCtxRegExp = regexp.MustCompile("^[0-9a-f]{2}-[a-f0-9]{32}-[a-f0-9]{16}-[a-f0-9]{2}-?") var traceCtxRegExp = regexp.MustCompile("^[0-9a-f]{2}-[a-f0-9]{32}-[a-f0-9]{16}-[a-f0-9]{2}-?")
func (hp httpTraceContextPropagator) Inject(ctx context.Context, supplier apipropagation.Supplier) { func (hp HTTPTraceContextPropagator) Inject(ctx context.Context, supplier apipropagation.Supplier) {
sc := trace.CurrentSpan(ctx).SpanContext() sc := trace.CurrentSpan(ctx).SpanContext()
if sc.IsValid() { if sc.IsValid() {
h := fmt.Sprintf("%.2x-%.16x%.16x-%.16x-%.2x", h := fmt.Sprintf("%.2x-%.16x%.16x-%.16x-%.2x",
@ -52,7 +53,7 @@ func (hp httpTraceContextPropagator) Inject(ctx context.Context, supplier apipro
} }
} }
func (hp httpTraceContextPropagator) Extract(ctx context.Context, supplier apipropagation.Supplier) core.SpanContext { func (hp HTTPTraceContextPropagator) Extract(ctx context.Context, supplier apipropagation.Supplier) core.SpanContext {
h := supplier.Get(TraceparentHeader) h := supplier.Get(TraceparentHeader)
if h == "" { if h == "" {
return core.EmptySpanContext() return core.EmptySpanContext()
@ -127,12 +128,6 @@ func (hp httpTraceContextPropagator) Extract(ctx context.Context, supplier apipr
return sc return sc
} }
func (hp httpTraceContextPropagator) GetAllKeys() []string { func (hp HTTPTraceContextPropagator) GetAllKeys() []string {
return []string{TraceparentHeader} return []string{TraceparentHeader}
} }
// HttpTraceContextPropagator creates a new text format propagator that propagates SpanContext
// in W3C TraceContext format.
func HttpTraceContextPropagator() apipropagation.TextFormatPropagator {
return httpTraceContextPropagator{}
}

View File

@ -11,7 +11,7 @@ import (
) )
func BenchmarkInject(b *testing.B) { func BenchmarkInject(b *testing.B) {
t := httpTraceContextPropagator{} var t HTTPTraceContextPropagator
injectSubBenchmarks(b, func(ctx context.Context, b *testing.B) { injectSubBenchmarks(b, func(ctx context.Context, b *testing.B) {
req, _ := http.NewRequest("GET", "http://example.com", nil) req, _ := http.NewRequest("GET", "http://example.com", nil)
@ -31,7 +31,7 @@ func injectSubBenchmarks(b *testing.B, fn func(context.Context, *testing.B)) {
) )
mockTracer := &mocktrace.MockTracer{ mockTracer := &mocktrace.MockTracer{
Sampled: false, Sampled: false,
StartSpanId: &id, StartSpanID: &id,
} }
b.ReportAllocs() b.ReportAllocs()
sc := core.SpanContext{ sc := core.SpanContext{
@ -53,7 +53,7 @@ func injectSubBenchmarks(b *testing.B, fn func(context.Context, *testing.B)) {
func BenchmarkExtract(b *testing.B) { func BenchmarkExtract(b *testing.B) {
extractSubBenchmarks(b, func(b *testing.B, req *http.Request) { extractSubBenchmarks(b, func(b *testing.B, req *http.Request) {
propagator := HttpTraceContextPropagator() var propagator HTTPTraceContextPropagator
ctx := context.Background() ctx := context.Background()
b.ResetTimer() b.ResetTimer()
for i := 0; i < b.N; i++ { for i := 0; i < b.N; i++ {

View File

@ -35,7 +35,7 @@ var (
func TestExtractValidTraceContextFromHTTPReq(t *testing.T) { func TestExtractValidTraceContextFromHTTPReq(t *testing.T) {
trace.SetGlobalTracer(&mocktrace.MockTracer{}) trace.SetGlobalTracer(&mocktrace.MockTracer{})
propagator := propagation.HttpTraceContextPropagator() var propagator propagation.HTTPTraceContextPropagator
tests := []struct { tests := []struct {
name string name string
header string header string
@ -129,7 +129,7 @@ func TestExtractValidTraceContextFromHTTPReq(t *testing.T) {
func TestExtractInvalidTraceContextFromHTTPReq(t *testing.T) { func TestExtractInvalidTraceContextFromHTTPReq(t *testing.T) {
trace.SetGlobalTracer(&mocktrace.MockTracer{}) trace.SetGlobalTracer(&mocktrace.MockTracer{})
propagator := propagation.HttpTraceContextPropagator() var propagator propagation.HTTPTraceContextPropagator
wantSc := core.EmptySpanContext() wantSc := core.EmptySpanContext()
tests := []struct { tests := []struct {
name string name string
@ -219,9 +219,9 @@ func TestInjectTraceContextToHTTPReq(t *testing.T) {
var id uint64 var id uint64
mockTracer := &mocktrace.MockTracer{ mockTracer := &mocktrace.MockTracer{
Sampled: false, Sampled: false,
StartSpanId: &id, StartSpanID: &id,
} }
propagator := propagation.HttpTraceContextPropagator() var propagator propagation.HTTPTraceContextPropagator
tests := []struct { tests := []struct {
name string name string
sc core.SpanContext sc core.SpanContext
@ -277,7 +277,7 @@ func TestInjectTraceContextToHTTPReq(t *testing.T) {
} }
func TestHttpTraceContextPropagator_GetAllKeys(t *testing.T) { func TestHttpTraceContextPropagator_GetAllKeys(t *testing.T) {
propagator := propagation.HttpTraceContextPropagator() var propagator propagation.HTTPTraceContextPropagator
want := []string{"Traceparent"} want := []string{"Traceparent"}
got := propagator.GetAllKeys() got := propagator.GetAllKeys()
if diff := cmp.Diff(got, want); diff != "" { if diff := cmp.Diff(got, want); diff != "" {

View File

@ -24,9 +24,9 @@ import (
) )
const ( const (
defaultMaxQueueSize = 2048 defaultMaxQueueSize = 2048
defaultScheduledDelayMillis = time.Duration(5000 * time.Millisecond) defaultScheduledDelay = time.Duration(5000 * time.Millisecond)
defaultMaxExportBatchSize = 512 defaultMaxExportBatchSize = 512
) )
var ( var (
@ -86,7 +86,7 @@ func NewBatchSpanProcessor(e export.SpanBatcher, opts ...BatchSpanProcessorOptio
} }
o := BatchSpanProcessorOptions{ o := BatchSpanProcessorOptions{
ScheduledDelayMillis: defaultScheduledDelayMillis, ScheduledDelayMillis: defaultScheduledDelay,
MaxQueueSize: defaultMaxQueueSize, MaxQueueSize: defaultMaxQueueSize,
MaxExportBatchSize: defaultMaxExportBatchSize, MaxExportBatchSize: defaultMaxExportBatchSize,
} }