/
xresolver.go
79 lines (65 loc) · 1.57 KB
/
xresolver.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
72
73
74
75
76
77
78
79
package xresolver
import (
"strings"
"google.golang.org/grpc/resolver"
"github.com/ydb-platform/ydb-go-sdk/v3/internal/stack"
"github.com/ydb-platform/ydb-go-sdk/v3/internal/xerrors"
"github.com/ydb-platform/ydb-go-sdk/v3/trace"
)
type dnsBuilder struct {
resolver.Builder
scheme string
trace *trace.Driver
}
type clientConn struct {
resolver.ClientConn
target resolver.Target
trace *trace.Driver
}
func (c *clientConn) Endpoint() string {
endpoint := c.target.URL.Path
if endpoint == "" {
endpoint = c.target.URL.Opaque
}
return strings.TrimPrefix(endpoint, "/")
}
func (c *clientConn) UpdateState(state resolver.State) (err error) {
onDone := trace.DriverOnResolve(c.trace,
stack.FunctionID("github.com/ydb-platform/ydb-go-sdk/3/internal/xresolver.(*clientConn).UpdateState"),
c.Endpoint(), func() (addrs []string) {
for i := range state.Addresses {
addrs = append(addrs, state.Addresses[i].Addr)
}
return
}(),
)
defer func() {
onDone(err)
}()
err = c.ClientConn.UpdateState(state)
if err != nil {
return xerrors.WithStackTrace(err)
}
return nil
}
func (d *dnsBuilder) Build(
target resolver.Target, //nolint:gocritic
cc resolver.ClientConn,
opts resolver.BuildOptions,
) (resolver.Resolver, error) {
return d.Builder.Build(target, &clientConn{
ClientConn: cc,
target: target,
trace: d.trace,
}, opts)
}
func (d *dnsBuilder) Scheme() string {
return d.scheme
}
func New(scheme string, trace *trace.Driver) resolver.Builder {
return &dnsBuilder{
Builder: resolver.Get("dns"),
scheme: scheme,
trace: trace,
}
}