package register // import "go.unistack.org/micro/v3/util/register" import ( "go.unistack.org/micro/v3/register" ) func addNodes(old, neu []*register.Node) []*register.Node { nodes := make([]*register.Node, 0, len(neu)) // add all new nodes for _, n := range neu { node := *n nodes = append(nodes, &node) } // look at old nodes for _, o := range old { var exists bool // check against new nodes for _, n := range nodes { // ids match then skip if o.ID == n.ID { exists = true break } } // keep old node if !exists { node := *o nodes = append(nodes, &node) } } return nodes } func delNodes(old, del []*register.Node) []*register.Node { var nodes []*register.Node for _, o := range old { var rem bool for _, n := range del { if o.ID == n.ID { rem = true break } } if !rem { nodes = append(nodes, o) } } return nodes } // CopyService make a copy of service func CopyService(service *register.Service) *register.Service { // copy service s := ®ister.Service{} *s = *service // copy nodes nodes := make([]*register.Node, len(service.Nodes)) for j, node := range service.Nodes { n := ®ister.Node{} *n = *node nodes[j] = n } s.Nodes = nodes // copy endpoints eps := make([]*register.Endpoint, len(service.Endpoints)) for j, ep := range service.Endpoints { e := ®ister.Endpoint{} *e = *ep eps[j] = e } s.Endpoints = eps return s } // Copy makes a copy of services func Copy(current []*register.Service) []*register.Service { services := make([]*register.Service, len(current)) for i, service := range current { services[i] = CopyService(service) } return services } // Merge merges two lists of services and returns a new copy func Merge(olist []*register.Service, nlist []*register.Service) []*register.Service { var srv []*register.Service for _, n := range nlist { var seen bool for _, o := range olist { if o.Version == n.Version { sp := ®ister.Service{} // make copy *sp = *o // set nodes sp.Nodes = addNodes(o.Nodes, n.Nodes) // mark as seen seen = true srv = append(srv, sp) break } else { sp := ®ister.Service{} // make copy *sp = *o srv = append(srv, sp) } } if !seen { srv = append(srv, Copy([]*register.Service{n})...) } } return srv } // Remove removes services and returns a new copy func Remove(old, del []*register.Service) []*register.Service { var services []*register.Service for _, o := range old { srv := ®ister.Service{} *srv = *o var rem bool for _, s := range del { if srv.Version == s.Version { srv.Nodes = delNodes(srv.Nodes, s.Nodes) if len(srv.Nodes) == 0 { rem = true } } } if !rem { services = append(services, srv) } } return services }