mirror of
https://github.com/go-kratos/kratos.git
synced 2025-02-03 13:11:42 +02:00
36 lines
803 B
Go
36 lines
803 B
Go
package direct
|
|
|
|
import (
|
|
"strings"
|
|
|
|
"google.golang.org/grpc/resolver"
|
|
)
|
|
|
|
func init() {
|
|
resolver.Register(NewBuilder())
|
|
}
|
|
|
|
type directBuilder struct{}
|
|
|
|
// NewBuilder creates a directBuilder which is used to factory direct resolvers.
|
|
// example:
|
|
// direct://<authority>/127.0.0.1:9000,127.0.0.2:9000
|
|
func NewBuilder() resolver.Builder {
|
|
return &directBuilder{}
|
|
}
|
|
|
|
func (d *directBuilder) Build(target resolver.Target, cc resolver.ClientConn, opts resolver.BuildOptions) (resolver.Resolver, error) {
|
|
var addrs []resolver.Address
|
|
for _, addr := range strings.Split(target.Endpoint, ",") {
|
|
addrs = append(addrs, resolver.Address{Addr: addr})
|
|
}
|
|
cc.UpdateState(resolver.State{
|
|
Addresses: addrs,
|
|
})
|
|
return newDirectResolver(), nil
|
|
}
|
|
|
|
func (d *directBuilder) Scheme() string {
|
|
return "direct"
|
|
}
|