2020-11-23 16:18:47 +03:00
|
|
|
// Package registry resolves names using the micro registry
|
2019-07-28 12:14:40 +01:00
|
|
|
package registry
|
|
|
|
|
|
|
|
import (
|
2020-11-03 02:02:32 +03:00
|
|
|
"context"
|
|
|
|
|
2020-08-19 17:47:17 +03:00
|
|
|
"github.com/unistack-org/micro/v3/registry"
|
2020-08-20 15:23:41 +03:00
|
|
|
"github.com/unistack-org/micro/v3/resolver"
|
2019-07-28 12:14:40 +01:00
|
|
|
)
|
|
|
|
|
2019-08-20 12:48:51 +01:00
|
|
|
// Resolver is a registry network resolver
|
2019-07-28 12:14:40 +01: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 20:00:09 +01:00
|
|
|
func (r *Resolver) Resolve(name string) ([]*resolver.Record, error) {
|
2020-11-03 02:02:32 +03:00
|
|
|
services, err := r.Registry.GetService(context.TODO(), name)
|
2019-07-28 12:14:40 +01:00
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
|
2020-08-29 17:43:36 +03:00
|
|
|
records := make([]*resolver.Record, 0, len(services))
|
2019-07-28 12:14:40 +01:00
|
|
|
for _, service := range services {
|
|
|
|
for _, node := range service.Nodes {
|
|
|
|
records = append(records, &resolver.Record{
|
|
|
|
Address: node.Address,
|
|
|
|
})
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
return records, nil
|
|
|
|
}
|