mirror of
https://github.com/zeromicro/go-zero.git
synced 2025-01-24 01:30:25 +08:00
22 lines
481 B
Go
22 lines
481 B
Go
package tracetest
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"go.opentelemetry.io/otel"
|
|
"go.opentelemetry.io/otel/sdk/trace"
|
|
"go.opentelemetry.io/otel/sdk/trace/tracetest"
|
|
)
|
|
|
|
// NewInMemoryExporter returns a new InMemoryExporter
|
|
// and sets it as the global for tests.
|
|
func NewInMemoryExporter(t *testing.T) *tracetest.InMemoryExporter {
|
|
me := tracetest.NewInMemoryExporter()
|
|
t.Cleanup(func() {
|
|
me.Reset()
|
|
})
|
|
otel.SetTracerProvider(trace.NewTracerProvider(trace.WithSyncer(me)))
|
|
|
|
return me
|
|
}
|