micro/resolver/registry/registry.go

40 lines
880 B
Go
Raw Normal View History

2019-07-28 22:00:09 +03:00
// Package registry resolves names using the go-micro registry
2019-07-28 14:14:40 +03:00
package registry
import (
"github.com/unistack-org/micro/v3/registry"
"github.com/unistack-org/micro/v3/registry/mdns"
"github.com/unistack-org/micro/v3/resolver"
2019-07-28 14:14:40 +03:00
)
// Resolver is a registry network resolver
2019-07-28 14:14:40 +03:00
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
2019-07-28 22:00:09 +03:00
func (r *Resolver) Resolve(name string) ([]*resolver.Record, error) {
2019-07-28 14:14:40 +03:00
reg := r.Registry
if reg == nil {
reg = mdns.NewRegistry()
2019-07-28 14:14:40 +03:00
}
2019-07-28 22:00:09 +03:00
services, err := reg.GetService(name)
2019-07-28 14:14:40 +03:00
if err != nil {
return nil, err
}
var records []*resolver.Record
for _, service := range services {
for _, node := range service.Nodes {
records = append(records, &resolver.Record{
Address: node.Address,
})
}
}
return records, nil
}