34140ff3e0
Previously, public referred to gRPC services that are both exposed on the dedicated gRPC port and have their definitions in the proto-public directory (so were considered usable by 3rd parties). Whereas private referred to services on the multiplexed server port that are only usable by agents and other servers. Now, we're splitting these definitions, such that external/internal refers to the port and public/private refers to whether they can be used by 3rd parties. This is necessary because the peering replication API needs to be exposed on the dedicated port, but is not (yet) suitable for use by 3rd parties.
55 lines
1.4 KiB
Go
55 lines
1.4 KiB
Go
package resolver
|
|
|
|
import (
|
|
"fmt"
|
|
"sync"
|
|
|
|
"google.golang.org/grpc/resolver"
|
|
)
|
|
|
|
// registry of ServerResolverBuilder. This type exists because grpc requires that
|
|
// resolvers are registered globally before any requests are made. This is
|
|
// incompatible with our resolver implementation and testing strategy, which
|
|
// requires a different Resolver for each test.
|
|
type registry struct {
|
|
lock sync.RWMutex
|
|
byAuthority map[string]*ServerResolverBuilder
|
|
}
|
|
|
|
func (r *registry) Build(target resolver.Target, cc resolver.ClientConn, opts resolver.BuildOptions) (resolver.Resolver, error) {
|
|
r.lock.RLock()
|
|
defer r.lock.RUnlock()
|
|
res, ok := r.byAuthority[target.Authority]
|
|
if !ok {
|
|
return nil, fmt.Errorf("no resolver registered for %v", target.Authority)
|
|
}
|
|
return res.Build(target, cc, opts)
|
|
}
|
|
|
|
func (r *registry) Scheme() string {
|
|
return "consul"
|
|
}
|
|
|
|
var _ resolver.Builder = (*registry)(nil)
|
|
|
|
var reg = ®istry{byAuthority: make(map[string]*ServerResolverBuilder)}
|
|
|
|
func init() {
|
|
resolver.Register(reg)
|
|
}
|
|
|
|
// Register a ServerResolverBuilder with the global registry.
|
|
func Register(res *ServerResolverBuilder) {
|
|
reg.lock.Lock()
|
|
defer reg.lock.Unlock()
|
|
reg.byAuthority[res.Authority()] = res
|
|
}
|
|
|
|
// Deregister the ServerResolverBuilder associated with the authority. Only used
|
|
// for testing.
|
|
func Deregister(authority string) {
|
|
reg.lock.Lock()
|
|
defer reg.lock.Unlock()
|
|
delete(reg.byAuthority, authority)
|
|
}
|