45 lines
907 B
Go
45 lines
907 B
Go
// Package registry resolves ids using the go-micro registry
|
|
package registry
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/micro/go-micro/network/resolver"
|
|
"github.com/micro/go-micro/registry"
|
|
)
|
|
|
|
type Resolver struct {
|
|
// Registry is the registry to use otherwise we use the defaul
|
|
Registry registry.Registry
|
|
}
|
|
|
|
// Resolve assumes ID is a domain name e.g micro.mu
|
|
func (r *Resolver) Resolve(id string) ([]*resolver.Record, error) {
|
|
reg := r.Registry
|
|
if reg == nil {
|
|
reg = registry.DefaultRegistry
|
|
}
|
|
|
|
services, err := reg.GetService(id)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
var records []*resolver.Record
|
|
|
|
for _, service := range services {
|
|
for _, node := range service.Nodes {
|
|
addr := node.Address
|
|
// such a hack
|
|
if node.Port > 0 {
|
|
addr = fmt.Sprintf("%s:%d", node.Address, node.Port)
|
|
}
|
|
records = append(records, &resolver.Record{
|
|
Address: addr,
|
|
})
|
|
}
|
|
}
|
|
|
|
return records, nil
|
|
}
|