/
resolver.go
53 lines (45 loc) · 1.32 KB
/
resolver.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
package grpc
import (
"github.com/resource-aware-jds/resource-aware-jds/pkg/cert"
"google.golang.org/grpc/resolver"
)
func ProvideRAJDSGRPCResolver() RAJDSGRPCResolver {
res := RAJDSGRPCResolver{
AddressMapper: make(map[string][]string),
}
resolver.Register(&res)
return res
}
type RAJDSGRPCResolver struct {
AddressMapper map[string][]string
}
func (r *RAJDSGRPCResolver) AddHost(name string, value string) {
r.AddressMapper[name] = []string{value}
}
func (r *RAJDSGRPCResolver) Build(target resolver.Target, cc resolver.ClientConn, opts resolver.BuildOptions) (resolver.Resolver, error) {
res := &customResolver{
target: target,
cc: cc,
addrsStore: r.AddressMapper,
}
res.start()
return res, nil
}
func (r *RAJDSGRPCResolver) Scheme() string {
return cert.DefaultScheme
}
type customResolver struct {
target resolver.Target
cc resolver.ClientConn
addrsStore map[string][]string
}
func (r *customResolver) start() {
addrStrs := r.addrsStore[r.target.URL.Hostname()]
addrs := make([]resolver.Address, len(addrStrs))
for i, s := range addrStrs {
addrs[i] = resolver.Address{Addr: s}
}
r.cc.UpdateState(resolver.State{Addresses: addrs}) //nolint:errcheck
}
func (*customResolver) ResolveNow(o resolver.ResolveNowOptions) {}
func (*customResolver) Close() {}