micro/resolver/registry/registry.go

35 lines
820 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 (
"context"
"github.com/unistack-org/micro/v3/registry"
"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) {
services, err := r.Registry.GetService(context.TODO(), name)
2019-07-28 14:14:40 +03:00
if err != nil {
return nil, err
}
records := make([]*resolver.Record, 0, len(services))
2019-07-28 14:14:40 +03:00
for _, service := range services {
for _, node := range service.Nodes {
records = append(records, &resolver.Record{
Address: node.Address,
})
}
}
return records, nil
}