unistack-org/micro
4
0
Fork 3
Code Issues 17 Pull Requests 1 Actions Packages Projects Releases 258 Wiki Activity

Support listing full service info in etcd

Browse Source
This commit is contained in:
Asim Aslam 2019-10-07 16:11:52 +01:00
parent fa6590f999
commit e9dcff49e0
1 changed files with 13 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
registry/etcd/etcd.go
View File

@ -349,7 +349,7 @@ func (e *etcdRegistry) GetService(name string) ([]*registry.Service, error) {
func (e *etcdRegistry) ListServices() ([]*registry.Service, error) { func (e *etcdRegistry) ListServices() ([]*registry.Service, error) {
var services []*registry.Service var services []*registry.Service
nameSet := make(map[string]struct{}) versions := make(map[string]*registry.Service)
ctx, cancel := context.WithTimeout(context.Background(), e.options.Timeout) ctx, cancel := context.WithTimeout(context.Background(), e.options.Timeout)
defer cancel() defer cancel()
@ -364,13 +364,20 @@ func (e *etcdRegistry) ListServices() ([]*registry.Service, error) {
} }
for _, n := range rsp.Kvs { for _, n := range rsp.Kvs {
if sn := decode(n.Value); sn != nil { sn := decode(n.Value)
nameSet[sn.Name] = struct{}{} if sn == nil {
continue
} }
v, ok := versions[sn.Name+sn.Version]
if !ok {
versions[sn.Name+sn.Version] = sn
continue
} }
for k := range nameSet { // append to service:version nodes
service := &registry.Service{} v.Nodes = append(v.Nodes, sn.Nodes...)
service.Name = k }
for _, service := range versions {
services = append(services, service) services = append(services, service)
} }