forked from lightstep/lightstep-tracer-go
-
Notifications
You must be signed in to change notification settings - Fork 0
/
collector_client.go
52 lines (41 loc) · 1.34 KB
/
collector_client.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
package lightstep
import (
"context"
"io"
"net/http"
"github.com/lightstep/lightstep-tracer-common/golang/gogo/collectorpb"
)
var accessTokenHeader = http.CanonicalHeaderKey("Lightstep-Access-Token")
// Connection describes a closable connection. Exposed for testing.
type Connection interface {
io.Closer
}
// ConnectorFactory is for testing purposes.
type ConnectorFactory func() (interface{}, Connection, error)
// collectorResponse encapsulates internal grpc/http responses.
type collectorResponse interface {
GetErrors() []string
Disable() bool
DevMode() bool
}
type reportRequest struct {
protoRequest *collectorpb.ReportRequest
httpRequest *http.Request
}
// collectorClient encapsulates internal grpc/http transports.
type collectorClient interface {
Report(context.Context, reportRequest) (collectorResponse, error)
Translate(context.Context, *reportBuffer) (reportRequest, error)
ConnectClient() (Connection, error)
ShouldReconnect() bool
}
func newCollectorClient(opts Options, reporterID uint64, attributes map[string]string) (collectorClient, error) {
if opts.UseHttp {
return newHTTPCollectorClient(opts, reporterID, attributes)
}
if opts.UseGRPC {
return newGrpcCollectorClient(opts, reporterID, attributes)
}
// No transport specified, defaulting to HTTP
return newHTTPCollectorClient(opts, reporterID, attributes)
}