forked from krakend/krakend-opencensus
/
xray.go
54 lines (47 loc) · 1.64 KB
/
xray.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
package xray
import (
"context"
"errors"
"time"
ocAws "contrib.go.opencensus.io/exporter/aws"
"github.com/aws/aws-sdk-go/aws"
"github.com/aws/aws-sdk-go/aws/credentials"
"github.com/aws/aws-sdk-go/aws/session"
"github.com/aws/aws-sdk-go/service/xray"
opencensus "github.com/davron112/krakend-opencensus/v2"
)
func init() {
opencensus.RegisterExporterFactories(func(ctx context.Context, cfg opencensus.Config) (interface{}, error) {
return Exporter(ctx, cfg)
})
}
func Exporter(_ context.Context, cfg opencensus.Config) (*ocAws.Exporter, error) {
if cfg.Exporters.Xray == nil {
return nil, errors.New("xray exporter disabled")
}
if cfg.Exporters.Xray.Version == "" {
cfg.Exporters.Xray.Version = "KrakenD-opencensus"
}
if !cfg.Exporters.Xray.UseEnv {
mySession := setupAWSSession(cfg.Exporters.Xray.AccessKey, cfg.Exporters.Xray.SecretKey, cfg.Exporters.Xray.Region)
return ocAws.NewExporter(
ocAws.WithAPI(xray.New(mySession, aws.NewConfig().WithRegion(cfg.Exporters.Xray.Region))),
ocAws.WithRegion(cfg.Exporters.Xray.Region),
ocAws.WithInterval(time.Duration(cfg.ReportingPeriod)),
ocAws.WithBufferSize(cfg.SampleRate),
ocAws.WithVersion(cfg.Exporters.Xray.Version),
)
}
return ocAws.NewExporter(
ocAws.WithRegion(cfg.Exporters.Xray.Region),
ocAws.WithInterval(time.Duration(cfg.ReportingPeriod)),
ocAws.WithBufferSize(cfg.SampleRate),
ocAws.WithVersion(cfg.Exporters.Xray.Version),
)
}
func setupAWSSession(id, secret, region string) *session.Session {
return session.Must(session.NewSession(&aws.Config{
Credentials: credentials.NewStaticCredentials(id, secret, ""),
Region: aws.String(region),
}))
}