forked from cortexproject/cortex
/
client.go
44 lines (39 loc) · 1.24 KB
/
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
package client
import (
"github.com/grpc-ecosystem/grpc-opentracing/go/otgrpc"
"github.com/mwitkow/go-grpc-middleware"
"github.com/opentracing/opentracing-go"
"google.golang.org/grpc"
_ "google.golang.org/grpc/encoding/gzip" // get gzip compressor registered
"github.com/weaveworks/common/middleware"
)
type closableIngesterClient struct {
IngesterClient
conn *grpc.ClientConn
}
// MakeIngesterClient makes a new IngesterClient
func MakeIngesterClient(addr string, withCompression bool) (IngesterClient, error) {
opts := []grpc.DialOption{
grpc.WithInsecure(),
grpc.WithUnaryInterceptor(grpc_middleware.ChainUnaryClient(
otgrpc.OpenTracingClientInterceptor(opentracing.GlobalTracer()),
middleware.ClientUserHeaderInterceptor,
)),
// We have seen 20MB returns from queries - add a bit of headroom
grpc.WithDefaultCallOptions(grpc.MaxCallRecvMsgSize(64 * 1024 * 1024)),
}
if withCompression {
opts = append(opts, grpc.WithDefaultCallOptions(grpc.UseCompressor("gzip")))
}
conn, err := grpc.Dial(addr, opts...)
if err != nil {
return nil, err
}
return &closableIngesterClient{
IngesterClient: NewIngesterClient(conn),
conn: conn,
}, nil
}
func (c *closableIngesterClient) Close() error {
return c.conn.Close()
}