rename util

Signed-off-by: Vasiliy Tolstov <v.tolstov@unistack.org>

util/register/util.go

@@ -0,0 +1,148 @@
+package register
+
+import (
+	"github.com/unistack-org/micro/v3/register"
+)
+
+func addNodes(old, neu []*register.Node) []*register.Node {
+	nodes := make([]*register.Node, len(neu))
+	// add all new nodes
+	for i, n := range neu {
+		node := *n
+		nodes[i] = &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 := &register.Service{}
+	*s = *service
+
+	// copy nodes
+	nodes := make([]*register.Node, len(service.Nodes))
+	for j, node := range service.Nodes {
+		n := &register.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 := &register.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 := &register.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 := &register.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 := &register.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
+}