209 lines
5.0 KiB
Diff
209 lines
5.0 KiB
Diff
|
diff --git a/pkg/cmd/grafana-server/server.go b/pkg/cmd/grafana-server/server.go
|
||
|
index 85d6db43e..6de99b58b 100644
|
||
|
--- a/pkg/cmd/grafana-server/server.go
|
||
|
+++ b/pkg/cmd/grafana-server/server.go
|
||
|
@@ -23,7 +23,6 @@ import (
|
||
|
_ "github.com/grafana/grafana/pkg/infra/metrics"
|
||
|
_ "github.com/grafana/grafana/pkg/infra/remotecache"
|
||
|
_ "github.com/grafana/grafana/pkg/infra/serverlock"
|
||
|
- _ "github.com/grafana/grafana/pkg/infra/tracing"
|
||
|
_ "github.com/grafana/grafana/pkg/infra/usagestats"
|
||
|
"github.com/grafana/grafana/pkg/login"
|
||
|
"github.com/grafana/grafana/pkg/login/social"
|
||
|
diff --git a/pkg/infra/tracing/tracing.go b/pkg/infra/tracing/tracing.go
|
||
|
deleted file mode 100644
|
||
|
index d1511a533..000000000
|
||
|
--- a/pkg/infra/tracing/tracing.go
|
||
|
+++ /dev/null
|
||
|
@@ -1,148 +0,0 @@
|
||
|
-package tracing
|
||
|
-
|
||
|
-import (
|
||
|
- "context"
|
||
|
- "fmt"
|
||
|
- "io"
|
||
|
- "strings"
|
||
|
-
|
||
|
- "github.com/grafana/grafana/pkg/infra/log"
|
||
|
- "github.com/grafana/grafana/pkg/registry"
|
||
|
- "github.com/grafana/grafana/pkg/setting"
|
||
|
-
|
||
|
- opentracing "github.com/opentracing/opentracing-go"
|
||
|
- jaegercfg "github.com/uber/jaeger-client-go/config"
|
||
|
- "github.com/uber/jaeger-client-go/zipkin"
|
||
|
-)
|
||
|
-
|
||
|
-func init() {
|
||
|
- registry.RegisterService(&TracingService{})
|
||
|
-}
|
||
|
-
|
||
|
-type TracingService struct {
|
||
|
- enabled bool
|
||
|
- address string
|
||
|
- customTags map[string]string
|
||
|
- samplerType string
|
||
|
- samplerParam float64
|
||
|
- log log.Logger
|
||
|
- closer io.Closer
|
||
|
- zipkinPropagation bool
|
||
|
- disableSharedZipkinSpans bool
|
||
|
-
|
||
|
- Cfg *setting.Cfg `inject:""`
|
||
|
-}
|
||
|
-
|
||
|
-func (ts *TracingService) Init() error {
|
||
|
- ts.log = log.New("tracing")
|
||
|
- ts.parseSettings()
|
||
|
-
|
||
|
- if ts.enabled {
|
||
|
- ts.initGlobalTracer()
|
||
|
- }
|
||
|
-
|
||
|
- return nil
|
||
|
-}
|
||
|
-
|
||
|
-func (ts *TracingService) parseSettings() {
|
||
|
- var section, err = ts.Cfg.Raw.GetSection("tracing.jaeger")
|
||
|
- if err != nil {
|
||
|
- return
|
||
|
- }
|
||
|
-
|
||
|
- ts.address = section.Key("address").MustString("")
|
||
|
- if ts.address != "" {
|
||
|
- ts.enabled = true
|
||
|
- }
|
||
|
-
|
||
|
- ts.customTags = splitTagSettings(section.Key("always_included_tag").MustString(""))
|
||
|
- ts.samplerType = section.Key("sampler_type").MustString("")
|
||
|
- ts.samplerParam = section.Key("sampler_param").MustFloat64(1)
|
||
|
- ts.zipkinPropagation = section.Key("zipkin_propagation").MustBool(false)
|
||
|
- ts.disableSharedZipkinSpans = section.Key("disable_shared_zipkin_spans").MustBool(false)
|
||
|
-}
|
||
|
-
|
||
|
-func (ts *TracingService) initGlobalTracer() error {
|
||
|
- cfg := jaegercfg.Configuration{
|
||
|
- ServiceName: "grafana",
|
||
|
- Disabled: !ts.enabled,
|
||
|
- Sampler: &jaegercfg.SamplerConfig{
|
||
|
- Type: ts.samplerType,
|
||
|
- Param: ts.samplerParam,
|
||
|
- },
|
||
|
- Reporter: &jaegercfg.ReporterConfig{
|
||
|
- LogSpans: false,
|
||
|
- LocalAgentHostPort: ts.address,
|
||
|
- },
|
||
|
- }
|
||
|
-
|
||
|
- jLogger := &jaegerLogWrapper{logger: log.New("jaeger")}
|
||
|
-
|
||
|
- options := []jaegercfg.Option{}
|
||
|
- options = append(options, jaegercfg.Logger(jLogger))
|
||
|
-
|
||
|
- for tag, value := range ts.customTags {
|
||
|
- options = append(options, jaegercfg.Tag(tag, value))
|
||
|
- }
|
||
|
-
|
||
|
- if ts.zipkinPropagation {
|
||
|
- zipkinPropagator := zipkin.NewZipkinB3HTTPHeaderPropagator()
|
||
|
- options = append(options,
|
||
|
- jaegercfg.Injector(opentracing.HTTPHeaders, zipkinPropagator),
|
||
|
- jaegercfg.Extractor(opentracing.HTTPHeaders, zipkinPropagator),
|
||
|
- )
|
||
|
-
|
||
|
- if !ts.disableSharedZipkinSpans {
|
||
|
- options = append(options, jaegercfg.ZipkinSharedRPCSpan(true))
|
||
|
- }
|
||
|
- }
|
||
|
-
|
||
|
- tracer, closer, err := cfg.NewTracer(options...)
|
||
|
- if err != nil {
|
||
|
- return err
|
||
|
- }
|
||
|
-
|
||
|
- opentracing.InitGlobalTracer(tracer)
|
||
|
-
|
||
|
- ts.closer = closer
|
||
|
-
|
||
|
- return nil
|
||
|
-}
|
||
|
-
|
||
|
-func (ts *TracingService) Run(ctx context.Context) error {
|
||
|
- <-ctx.Done()
|
||
|
-
|
||
|
- if ts.closer != nil {
|
||
|
- ts.log.Info("Closing tracing")
|
||
|
- ts.closer.Close()
|
||
|
- }
|
||
|
-
|
||
|
- return nil
|
||
|
-}
|
||
|
-
|
||
|
-func splitTagSettings(input string) map[string]string {
|
||
|
- res := map[string]string{}
|
||
|
-
|
||
|
- tags := strings.Split(input, ",")
|
||
|
- for _, v := range tags {
|
||
|
- kv := strings.Split(v, ":")
|
||
|
- if len(kv) > 1 {
|
||
|
- res[kv[0]] = kv[1]
|
||
|
- }
|
||
|
- }
|
||
|
-
|
||
|
- return res
|
||
|
-}
|
||
|
-
|
||
|
-type jaegerLogWrapper struct {
|
||
|
- logger log.Logger
|
||
|
-}
|
||
|
-
|
||
|
-func (jlw *jaegerLogWrapper) Error(msg string) {
|
||
|
- jlw.logger.Error(msg)
|
||
|
-}
|
||
|
-
|
||
|
-func (jlw *jaegerLogWrapper) Infof(format string, args ...interface{}) {
|
||
|
- msg := fmt.Sprintf(format, args...)
|
||
|
- jlw.logger.Info(msg)
|
||
|
-}
|
||
|
diff --git a/pkg/infra/tracing/tracing_test.go b/pkg/infra/tracing/tracing_test.go
|
||
|
deleted file mode 100644
|
||
|
index 27e4de777..000000000
|
||
|
--- a/pkg/infra/tracing/tracing_test.go
|
||
|
+++ /dev/null
|
||
|
@@ -1,36 +0,0 @@
|
||
|
-package tracing
|
||
|
-
|
||
|
-import "testing"
|
||
|
-
|
||
|
-func TestGroupSplit(t *testing.T) {
|
||
|
- tests := []struct {
|
||
|
- input string
|
||
|
- expected map[string]string
|
||
|
- }{
|
||
|
- {
|
||
|
- input: "tag1:value1,tag2:value2",
|
||
|
- expected: map[string]string{
|
||
|
- "tag1": "value1",
|
||
|
- "tag2": "value2",
|
||
|
- },
|
||
|
- },
|
||
|
- {
|
||
|
- input: "",
|
||
|
- expected: map[string]string{},
|
||
|
- },
|
||
|
- {
|
||
|
- input: "tag1",
|
||
|
- expected: map[string]string{},
|
||
|
- },
|
||
|
- }
|
||
|
-
|
||
|
- for _, test := range tests {
|
||
|
- tags := splitTagSettings(test.input)
|
||
|
- for k, v := range test.expected {
|
||
|
- value, exists := tags[k]
|
||
|
- if !exists || value != v {
|
||
|
- t.Errorf("tags does not match %v ", test)
|
||
|
- }
|
||
|
- }
|
||
|
- }
|
||
|
-}
|