2020-07-30 15:48:19 -07:00
|
|
|
// Copyright The OpenTelemetry Authors
|
|
|
|
//
|
|
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
|
|
// you may not use this file except in compliance with the License.
|
|
|
|
// You may obtain a copy of the License at
|
|
|
|
//
|
|
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
//
|
|
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
|
|
// See the License for the specific language governing permissions and
|
|
|
|
// limitations under the License.
|
|
|
|
|
|
|
|
// tracetest is a testing helper package for the SDK. User can configure no-op or in-memory exporters to verify
|
|
|
|
// different SDK behaviors or custom instrumentation.
|
|
|
|
package tracetest // import "go.opentelemetry.io/otel/sdk/export/trace/tracetest"
|
|
|
|
|
|
|
|
import (
|
|
|
|
"context"
|
|
|
|
"sync"
|
|
|
|
|
|
|
|
"go.opentelemetry.io/otel/sdk/export/trace"
|
|
|
|
)
|
|
|
|
|
2020-09-09 10:19:03 -07:00
|
|
|
var _ trace.SpanExporter = (*NoopExporter)(nil)
|
2020-07-30 15:48:19 -07:00
|
|
|
|
|
|
|
// NewNoopExporter returns a new no-op exporter.
|
|
|
|
func NewNoopExporter() *NoopExporter {
|
|
|
|
return new(NoopExporter)
|
|
|
|
}
|
|
|
|
|
2020-09-09 10:19:03 -07:00
|
|
|
// NoopExporter is an exporter that drops all received SpanData and performs
|
|
|
|
// no action.
|
2020-07-30 15:48:19 -07:00
|
|
|
type NoopExporter struct{}
|
|
|
|
|
2020-09-09 10:19:03 -07:00
|
|
|
// ExportSpans handles export of SpanData by dropping it.
|
|
|
|
func (nsb *NoopExporter) ExportSpans(context.Context, []*trace.SpanData) error { return nil }
|
2020-07-30 15:48:19 -07:00
|
|
|
|
2020-09-09 10:19:03 -07:00
|
|
|
// Shutdown stops the exporter by doing nothing.
|
|
|
|
func (nsb *NoopExporter) Shutdown(context.Context) error { return nil }
|
2020-07-30 15:48:19 -07:00
|
|
|
|
2020-09-09 10:19:03 -07:00
|
|
|
var _ trace.SpanExporter = (*InMemoryExporter)(nil)
|
2020-07-30 15:48:19 -07:00
|
|
|
|
2020-09-09 10:19:03 -07:00
|
|
|
// NewInMemoryExporter returns a new InMemoryExporter.
|
2020-07-30 15:48:19 -07:00
|
|
|
func NewInMemoryExporter() *InMemoryExporter {
|
|
|
|
return new(InMemoryExporter)
|
|
|
|
}
|
|
|
|
|
2020-09-09 10:19:03 -07:00
|
|
|
// InMemoryExporter is an exporter that stores all received spans in-memory.
|
2020-07-30 15:48:19 -07:00
|
|
|
type InMemoryExporter struct {
|
|
|
|
mu sync.Mutex
|
|
|
|
sds []*trace.SpanData
|
|
|
|
}
|
|
|
|
|
2020-09-09 10:19:03 -07:00
|
|
|
// ExportSpans handles export of SpanData by storing it in memory.
|
|
|
|
func (imsb *InMemoryExporter) ExportSpans(_ context.Context, sds []*trace.SpanData) error {
|
2020-07-30 15:48:19 -07:00
|
|
|
imsb.mu.Lock()
|
|
|
|
defer imsb.mu.Unlock()
|
|
|
|
imsb.sds = append(imsb.sds, sds...)
|
2020-09-09 10:19:03 -07:00
|
|
|
return nil
|
2020-07-30 15:48:19 -07:00
|
|
|
}
|
|
|
|
|
2020-09-09 10:19:03 -07:00
|
|
|
// Shutdown stops the exporter by clearing SpanData held in memory.
|
|
|
|
func (imsb *InMemoryExporter) Shutdown(context.Context) error {
|
|
|
|
imsb.Reset()
|
|
|
|
return nil
|
2020-07-30 15:48:19 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
// Reset the current in-memory storage.
|
|
|
|
func (imsb *InMemoryExporter) Reset() {
|
|
|
|
imsb.mu.Lock()
|
|
|
|
defer imsb.mu.Unlock()
|
|
|
|
imsb.sds = nil
|
|
|
|
}
|
|
|
|
|
|
|
|
// GetSpans returns the current in-memory stored spans.
|
|
|
|
func (imsb *InMemoryExporter) GetSpans() []*trace.SpanData {
|
|
|
|
imsb.mu.Lock()
|
|
|
|
defer imsb.mu.Unlock()
|
|
|
|
ret := make([]*trace.SpanData, len(imsb.sds))
|
|
|
|
copy(ret, imsb.sds)
|
|
|
|
return ret
|
|
|
|
}
|