/
grpc_resolver.go
51 lines (42 loc) · 1.42 KB
/
grpc_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
package main
import "google.golang.org/grpc/resolver"
//TODO
// expose ports which can add an address to the address list and then call UpdateState (aka dynamically adding new server locations to already running clients)
// Following is an example name resolver implementation. Read the name
// resolution example to learn more about it.
const (
exampleScheme = "example"
exampleServiceName = "www.example.com"
)
var addrs = []string{"192.168.2.75:50051"}
type ExampleResolverBuilder struct{}
func (*ExampleResolverBuilder) Build(target resolver.Target, cc resolver.ClientConn, opts resolver.BuildOption) (resolver.Resolver, error) {
r := &exampleResolver{
target: target,
cc: cc,
addrsStore: map[string][]string{
exampleServiceName: addrs,
},
}
r.start()
return r, nil
}
func (*ExampleResolverBuilder) Scheme() string { return exampleScheme }
type exampleResolver struct {
target resolver.Target
cc resolver.ClientConn
addrsStore map[string][]string
}
func (r *exampleResolver) start() {
addrStrs := r.addrsStore[r.target.Endpoint]
addrs := make([]resolver.Address, len(addrStrs))
for i, s := range addrStrs {
addrs[i] = resolver.Address{Addr: s}
}
r.cc.UpdateState(resolver.State{Addresses: addrs})
}
func (*exampleResolver) ResolveNow(o resolver.ResolveNowOption) {}
func (*exampleResolver) Close() {}
func init() {
resolver.Register(&ExampleResolverBuilder{})
}