-
Notifications
You must be signed in to change notification settings - Fork 0
/
tracer.go
58 lines (50 loc) · 1.16 KB
/
tracer.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
package tracer
import (
"fmt"
"github.com/opentracing/opentracing-go"
"github.com/uber/jaeger-client-go/config"
"io"
)
// singleton
var bt *backTracer
func InitDefaultTracer(serviceName string) error {
return InitTracer(TRACER_TYPE_JAEGER, serviceName)
}
func InitTracer(tracerType string, serviceName string) error {
// only init once
if bt != nil {
return nil
}
switch tracerType {
case TRACER_TYPE_JAEGER:
return initJaeger(serviceName)
default:
return fmt.Errorf("unsupported tracer type:%s", tracerType)
}
}
func CloseTracer() error {
if bt != nil {
return bt.closer.Close()
}
return nil
}
type backTracer struct {
tracer opentracing.Tracer
closer io.Closer
}
func initJaeger(serviceName string) error {
fun := "initJaeger-->"
defaultConfig := defaultConfigurator.GetConfig(serviceName)
if cfg, ok := defaultConfig.TracerConfig.(config.Configuration); !ok {
return fmt.Errorf("wrong tracer config:%v", cfg)
} else {
tracer, closer, err := cfg.NewTracer()
if err != nil {
return err
}
bt = &backTracer{tracer, closer}
opentracing.SetGlobalTracer(bt.tracer)
fmt.Printf("%s succeed for %s\n", fun, serviceName)
return nil
}
}