forked from lileio/lile
/
fromenv.go
71 lines (58 loc) · 1.72 KB
/
fromenv.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
59
60
61
62
63
64
65
66
67
68
69
70
71
// package fromenv provides utilities to create lile options from environment
// variables. fromenv will error with fatal if it cannot resolve or errors
package fromenv
import (
"fmt"
"os"
"github.com/lileio/pubsub"
"github.com/lileio/pubsub/google"
opentracing "github.com/opentracing/opentracing-go"
zipkin "github.com/openzipkin/zipkin-go-opentracing"
"github.com/sirupsen/logrus"
)
func Tracer(name string) opentracing.Tracer {
var collector zipkin.Collector
var err error
zipkin_host := os.Getenv("ZIPKIN_SERVICE_HOST")
if zipkin_host != "" {
addr := fmt.Sprintf("http://%s:%s/api/v1/spans",
os.Getenv("ZIPKIN_SERVICE_HOST"),
os.Getenv("ZIPKIN_SERVICE_PORT"))
collector, err = zipkin.NewHTTPCollector(addr)
if err != nil {
logrus.Fatalf("unable to create Zipkin HTTP collector: %+v", err)
}
logrus.Infof("Using Zipkin HTTP tracer: %s", addr)
}
if collector == nil {
logrus.Infof("Using Zipkin Global tracer")
return opentracing.GlobalTracer()
}
// create recorder.
recorder := zipkin.NewRecorder(collector, false, "", name)
// create tracer.
tracer, err := zipkin.NewTracer(
recorder,
zipkin.ClientServerSameSpan(true),
)
if err != nil {
logrus.Fatalf("unable to create Zipkin tracer: %+v", err)
}
// explicitly set our tracer to be the default tracer.
opentracing.InitGlobalTracer(tracer)
return tracer
}
func PubSubProvider() pubsub.Provider {
gpid := os.Getenv("GOOGLE_PUBSUB_PROJECT_ID")
if gpid != "" {
gc, err := google.NewGoogleCloud(gpid)
if err != nil {
logrus.Fatalf("fronenv: Google Cloud pubsub err: %s", err)
return nil
}
logrus.Infof("Using Google Cloud pubsub: %s", gpid)
return gc
}
logrus.Warn("Using noop pubsub provider")
return pubsub.NoopProvider{}
}