mirror of
https://github.com/mattermost/mattermost.git
synced 2025-02-25 18:55:24 -06:00
182c29b456
* initial implementation of opentracing * app layer * Revert Makefile * . * cleanup * . * . * . * . * . * . * . * . * . * . * . * [ci] * autogenerate interface * . * missed vendor files * updated interfaces * updated store layers * lint fixes * . * finishing layer generators and nested spans * added errors and b3 support * code review * . * . * fixed build error due to misplased flag.Parse() * code review addressed
89 lines
2.4 KiB
Go
89 lines
2.4 KiB
Go
// Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.
|
|
// See LICENSE.txt for license information.
|
|
|
|
package tracing
|
|
|
|
import (
|
|
"io"
|
|
"time"
|
|
|
|
"github.com/mattermost/mattermost-server/v5/mlog"
|
|
opentracing "github.com/opentracing/opentracing-go"
|
|
"github.com/uber/jaeger-lib/metrics"
|
|
|
|
"context"
|
|
|
|
"github.com/uber/jaeger-client-go"
|
|
jaegercfg "github.com/uber/jaeger-client-go/config"
|
|
"github.com/uber/jaeger-client-go/zipkin"
|
|
)
|
|
|
|
// Tracer is a wrapper around Jaeger OpenTracing client, used to properly de-initialize jaeger on exit
|
|
type Tracer struct {
|
|
closer io.Closer
|
|
}
|
|
|
|
type LogrusAdapter struct {
|
|
}
|
|
|
|
// Error - logrus adapter for span errors
|
|
func (LogrusAdapter) Error(msg string) {
|
|
mlog.Error(msg)
|
|
}
|
|
|
|
// Infof - logrus adapter for span info logging
|
|
func (LogrusAdapter) Infof(msg string, args ...interface{}) {
|
|
// we ignore Info messages from opentracing
|
|
}
|
|
|
|
// New instantiates Jaeger opentracing client with default options
|
|
// To override the defaults use environment variables listed here: https://github.com/jaegertracing/jaeger-client-go/blob/master/config/config.go
|
|
func New() (*Tracer, error) {
|
|
cfg := jaegercfg.Configuration{
|
|
Sampler: &jaegercfg.SamplerConfig{
|
|
Type: jaeger.SamplerTypeConst,
|
|
Param: 1,
|
|
},
|
|
Reporter: &jaegercfg.ReporterConfig{
|
|
LogSpans: true,
|
|
},
|
|
}
|
|
|
|
zipkinPropagator := zipkin.NewZipkinB3HTTPHeaderPropagator()
|
|
|
|
closer, err := cfg.InitGlobalTracer(
|
|
"mattermost",
|
|
jaegercfg.Logger(LogrusAdapter{}),
|
|
jaegercfg.Metrics(metrics.NullFactory),
|
|
jaegercfg.Tag("serverStartTime", time.Now().UTC().Format(time.RFC3339)),
|
|
jaegercfg.Injector(opentracing.HTTPHeaders, zipkinPropagator),
|
|
jaegercfg.Extractor(opentracing.HTTPHeaders, zipkinPropagator),
|
|
jaegercfg.ZipkinSharedRPCSpan(true),
|
|
)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
mlog.Info("Opentracing initialzed")
|
|
return &Tracer{
|
|
closer: closer,
|
|
}, nil
|
|
}
|
|
|
|
func (t *Tracer) Close() error {
|
|
return t.closer.Close()
|
|
}
|
|
|
|
func StartRootSpanByContext(ctx context.Context, operationName string) (opentracing.Span, context.Context) {
|
|
return opentracing.StartSpanFromContext(ctx, operationName)
|
|
}
|
|
|
|
func StartSpanWithParentByContext(ctx context.Context, operationName string) (opentracing.Span, context.Context) {
|
|
parentSpan := opentracing.SpanFromContext(ctx)
|
|
|
|
if parentSpan == nil {
|
|
return StartRootSpanByContext(ctx, operationName)
|
|
}
|
|
|
|
return opentracing.StartSpanFromContext(ctx, operationName, opentracing.ChildOf(parentSpan.Context()))
|
|
}
|