62 lines
1.3 KiB
Go
62 lines
1.3 KiB
Go
|
package main
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"fmt"
|
||
|
|
"io"
|
||
|
|
|
||
|
|
opentracing "github.com/opentracing/opentracing-go"
|
||
|
|
jconfig "github.com/uber/jaeger-client-go/config"
|
||
|
|
"go.unistack.org/micro/v3/logger"
|
||
|
|
)
|
||
|
|
|
||
|
|
func NewTracer(cfg *TracerConfig) (opentracing.Tracer, io.Closer, error) {
|
||
|
|
jcfg := &jconfig.Configuration{
|
||
|
|
ServiceName: cfg.Service,
|
||
|
|
Sampler: &jconfig.SamplerConfig{
|
||
|
|
Type: "remote",
|
||
|
|
Param: 1,
|
||
|
|
},
|
||
|
|
Reporter: &jconfig.ReporterConfig{
|
||
|
|
LogSpans: false,
|
||
|
|
QueueSize: 100,
|
||
|
|
},
|
||
|
|
}
|
||
|
|
|
||
|
|
if len(cfg.AgentHost) > 0 {
|
||
|
|
jcfg.Sampler.SamplingServerURL = fmt.Sprintf("http://%s:5778/sampling", cfg.AgentHost)
|
||
|
|
}
|
||
|
|
|
||
|
|
if len(cfg.AgentHost) > 0 {
|
||
|
|
agentPort := "6831"
|
||
|
|
if len(cfg.AgentPort) > 0 {
|
||
|
|
agentPort = cfg.AgentPort
|
||
|
|
}
|
||
|
|
jcfg.Reporter.LocalAgentHostPort = fmt.Sprintf("%s:%s", cfg.AgentHost, agentPort)
|
||
|
|
} else if len(cfg.Collector) > 0 {
|
||
|
|
jcfg.Reporter.CollectorEndpoint = cfg.Collector
|
||
|
|
}
|
||
|
|
|
||
|
|
for k, v := range cfg.Metadata {
|
||
|
|
jcfg.Tags = append(jcfg.Tags, opentracing.Tag{Key: k, Value: v})
|
||
|
|
}
|
||
|
|
|
||
|
|
tracer, closer, err := jcfg.NewTracer(jconfig.Logger(&log{}))
|
||
|
|
if err != nil {
|
||
|
|
return nil, nil, err
|
||
|
|
}
|
||
|
|
|
||
|
|
opentracing.SetGlobalTracer(tracer)
|
||
|
|
return tracer, closer, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
type log struct{}
|
||
|
|
|
||
|
|
func (l *log) Error(msg string) {
|
||
|
|
logger.Error(context.Background(), msg)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (l *log) Infof(msg string, args ...interface{}) {
|
||
|
|
logger.Infof(context.Background(), msg, args...)
|
||
|
|
}
|