traefik/pkg/tracing/jaeger/jaeger.go

113 lines
4.6 KiB
Go
Raw Normal View History

2018-01-10 16:48:04 +00:00
package jaeger
import (
2018-08-01 11:52:03 +00:00
"fmt"
2018-01-10 16:48:04 +00:00
"io"
2019-08-03 01:58:23 +00:00
"github.com/containous/traefik/v2/pkg/log"
2018-01-10 16:48:04 +00:00
"github.com/opentracing/opentracing-go"
jaegercli "github.com/uber/jaeger-client-go"
2018-01-10 16:48:04 +00:00
jaegercfg "github.com/uber/jaeger-client-go/config"
2018-08-01 11:52:03 +00:00
"github.com/uber/jaeger-client-go/zipkin"
2018-01-10 16:48:04 +00:00
jaegermet "github.com/uber/jaeger-lib/metrics"
)
2019-07-15 12:52:04 +00:00
// Name sets the name of this tracer.
2018-01-10 16:48:04 +00:00
const Name = "jaeger"
2019-07-15 12:52:04 +00:00
// Config provides configuration settings for a jaeger tracer.
2018-01-10 16:48:04 +00:00
type Config struct {
2019-07-15 12:52:04 +00:00
SamplingServerURL string `description:"Set the sampling server url." json:"samplingServerURL,omitempty" toml:"samplingServerURL,omitempty" yaml:"samplingServerURL,omitempty"`
SamplingType string `description:"Set the sampling type." json:"samplingType,omitempty" toml:"samplingType,omitempty" yaml:"samplingType,omitempty" export:"true"`
SamplingParam float64 `description:"Set the sampling parameter." json:"samplingParam,omitempty" toml:"samplingParam,omitempty" yaml:"samplingParam,omitempty" export:"true"`
LocalAgentHostPort string `description:"Set jaeger-agent's host:port that the reporter will used." json:"localAgentHostPort,omitempty" toml:"localAgentHostPort,omitempty" yaml:"localAgentHostPort,omitempty"`
Gen128Bit bool `description:"Generate 128 bit span IDs." json:"gen128Bit,omitempty" toml:"gen128Bit,omitempty" yaml:"gen128Bit,omitempty" export:"true"`
Propagation string `description:"Which propagation format to use (jaeger/b3)." json:"propagation,omitempty" toml:"propagation,omitempty" yaml:"propagation,omitempty" export:"true"`
TraceContextHeaderName string `description:"Set the header to use for the trace-id." json:"traceContextHeaderName,omitempty" toml:"traceContextHeaderName,omitempty" yaml:"traceContextHeaderName,omitempty" export:"true"`
Collector *Collector `description:"Define the collector information" json:"collector,omitempty" toml:"collector,omitempty" yaml:"collector,omitempty" export:"true"`
}
// SetDefaults sets the default values.
func (c *Config) SetDefaults() {
c.SamplingServerURL = "http://localhost:5778/sampling"
c.SamplingType = "const"
c.SamplingParam = 1.0
c.LocalAgentHostPort = "127.0.0.1:6831"
c.Propagation = "jaeger"
c.Gen128Bit = false
c.TraceContextHeaderName = jaegercli.TraceContextHeaderName
2018-01-10 16:48:04 +00:00
}
2019-07-15 12:52:04 +00:00
// Collector provides configuration settings for jaeger collector.
type Collector struct {
Endpoint string `description:"Instructs reporter to send spans to jaeger-collector at this URL." json:"endpoint,omitempty" toml:"endpoint,omitempty" yaml:"endpoint,omitempty" export:"true"`
User string `description:"User for basic http authentication when sending spans to jaeger-collector." json:"user,omitempty" toml:"user,omitempty" yaml:"user,omitempty"`
Password string `description:"Password for basic http authentication when sending spans to jaeger-collector." json:"password,omitempty" toml:"password,omitempty" yaml:"password,omitempty"`
}
// SetDefaults sets the default values.
func (c *Collector) SetDefaults() {
c.Endpoint = ""
c.User = ""
c.Password = ""
}
2020-05-11 10:06:07 +00:00
// Setup sets up the tracer.
2018-01-10 16:48:04 +00:00
func (c *Config) Setup(componentName string) (opentracing.Tracer, io.Closer, error) {
2019-07-15 12:52:04 +00:00
reporter := &jaegercfg.ReporterConfig{
LogSpans: true,
LocalAgentHostPort: c.LocalAgentHostPort,
}
if c.Collector != nil {
reporter.CollectorEndpoint = c.Collector.Endpoint
reporter.User = c.Collector.User
reporter.Password = c.Collector.Password
}
2018-01-10 16:48:04 +00:00
jcfg := jaegercfg.Configuration{
Sampler: &jaegercfg.SamplerConfig{
SamplingServerURL: c.SamplingServerURL,
Type: c.SamplingType,
Param: c.SamplingParam,
},
2019-07-15 12:52:04 +00:00
Reporter: reporter,
2019-12-12 16:48:05 +00:00
Headers: &jaegercli.HeadersConfig{
TraceContextHeaderName: c.TraceContextHeaderName,
},
2018-01-10 16:48:04 +00:00
}
jMetricsFactory := jaegermet.NullFactory
2018-08-01 11:52:03 +00:00
opts := []jaegercfg.Option{
2018-11-14 09:18:03 +00:00
jaegercfg.Logger(newJaegerLogger()),
2018-08-01 11:52:03 +00:00
jaegercfg.Metrics(jMetricsFactory),
jaegercfg.Gen128Bit(c.Gen128Bit),
}
switch c.Propagation {
case "b3":
p := zipkin.NewZipkinB3HTTPHeaderPropagator()
opts = append(opts,
jaegercfg.Injector(opentracing.HTTPHeaders, p),
jaegercfg.Extractor(opentracing.HTTPHeaders, p),
)
case "jaeger", "":
default:
return nil, nil, fmt.Errorf("unknown propagation format: %s", c.Propagation)
}
2018-01-10 16:48:04 +00:00
// Initialize tracer with a logger and a metrics factory
closer, err := jcfg.InitGlobalTracer(
componentName,
2018-08-01 11:52:03 +00:00
opts...,
2018-01-10 16:48:04 +00:00
)
if err != nil {
2018-11-14 09:18:03 +00:00
log.WithoutContext().Warnf("Could not initialize jaeger tracer: %s", err.Error())
2018-01-10 16:48:04 +00:00
return nil, nil, err
}
2018-11-14 09:18:03 +00:00
log.WithoutContext().Debug("Jaeger tracer configured")
2018-01-10 16:48:04 +00:00
return opentracing.GlobalTracer(), closer, nil
}