Update registry so version is surfaced
This commit is contained in:
parent
4e6d1df9ea
commit
0320fc1b55
@ -190,13 +190,14 @@ func (h *httpBroker) Publish(topic string, msg *Message) error {
|
|||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
for _, node := range s.Nodes {
|
for _, service := range s {
|
||||||
r, err := http.Post(fmt.Sprintf("http://%s:%d%s", node.Address, node.Port, DefaultSubPath), "application/json", bytes.NewBuffer(b))
|
for _, node := range service.Nodes {
|
||||||
if err == nil {
|
r, err := http.Post(fmt.Sprintf("http://%s:%d%s", node.Address, node.Port, DefaultSubPath), "application/json", bytes.NewBuffer(b))
|
||||||
r.Body.Close()
|
if err == nil {
|
||||||
|
r.Body.Close()
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
29
client/node_selector.go
Normal file
29
client/node_selector.go
Normal file
@ -0,0 +1,29 @@
|
|||||||
|
package client
|
||||||
|
|
||||||
|
import (
|
||||||
|
"math/rand"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"github.com/myodc/go-micro/errors"
|
||||||
|
"github.com/myodc/go-micro/registry"
|
||||||
|
)
|
||||||
|
|
||||||
|
func init() {
|
||||||
|
rand.Seed(time.Now().UnixNano())
|
||||||
|
}
|
||||||
|
|
||||||
|
func nodeSelector(service []*registry.Service) (*registry.Node, error) {
|
||||||
|
if len(service) == 0 {
|
||||||
|
return nil, errors.NotFound("go.micro.client", "Service not found")
|
||||||
|
}
|
||||||
|
|
||||||
|
i := rand.Int()
|
||||||
|
j := i % len(service)
|
||||||
|
|
||||||
|
if len(service[j].Nodes) == 0 {
|
||||||
|
return nil, errors.NotFound("go.micro.client", "Service not found")
|
||||||
|
}
|
||||||
|
|
||||||
|
n := i % len(service[j].Nodes)
|
||||||
|
return service[j].Nodes[n], nil
|
||||||
|
}
|
46
client/node_selector_test.go
Normal file
46
client/node_selector_test.go
Normal file
@ -0,0 +1,46 @@
|
|||||||
|
package client
|
||||||
|
|
||||||
|
import (
|
||||||
|
"testing"
|
||||||
|
|
||||||
|
"github.com/myodc/go-micro/registry"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestNodeSelector(t *testing.T) {
|
||||||
|
services := []*registry.Service{
|
||||||
|
{
|
||||||
|
Name: "foo",
|
||||||
|
Version: "1.0.0",
|
||||||
|
Nodes: []*registry.Node{
|
||||||
|
{
|
||||||
|
Id: "foo-123",
|
||||||
|
Address: "localhost",
|
||||||
|
Port: 9999,
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
Name: "foo",
|
||||||
|
Version: "1.0.1",
|
||||||
|
Nodes: []*registry.Node{
|
||||||
|
{
|
||||||
|
Id: "foo-321",
|
||||||
|
Address: "localhost",
|
||||||
|
Port: 6666,
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
counts := map[string]int{}
|
||||||
|
|
||||||
|
for i := 0; i < 100; i++ {
|
||||||
|
n, err := nodeSelector(services)
|
||||||
|
if err != nil {
|
||||||
|
t.Errorf("Expected node, got err: %v", err)
|
||||||
|
}
|
||||||
|
counts[n.Id]++
|
||||||
|
}
|
||||||
|
|
||||||
|
t.Logf("Counts %v", counts)
|
||||||
|
}
|
@ -3,10 +3,8 @@ package client
|
|||||||
import (
|
import (
|
||||||
"encoding/json"
|
"encoding/json"
|
||||||
"fmt"
|
"fmt"
|
||||||
"math/rand"
|
|
||||||
"net/http"
|
"net/http"
|
||||||
"sync"
|
"sync"
|
||||||
"time"
|
|
||||||
|
|
||||||
"github.com/myodc/go-micro/broker"
|
"github.com/myodc/go-micro/broker"
|
||||||
c "github.com/myodc/go-micro/context"
|
c "github.com/myodc/go-micro/context"
|
||||||
@ -29,10 +27,6 @@ type rpcClient struct {
|
|||||||
opts options
|
opts options
|
||||||
}
|
}
|
||||||
|
|
||||||
func init() {
|
|
||||||
rand.Seed(time.Now().UnixNano())
|
|
||||||
}
|
|
||||||
|
|
||||||
func newRpcClient(opt ...Option) Client {
|
func newRpcClient(opt ...Option) Client {
|
||||||
var opts options
|
var opts options
|
||||||
|
|
||||||
@ -129,13 +123,11 @@ func (r *rpcClient) Call(ctx context.Context, request Request, response interfac
|
|||||||
return errors.InternalServerError("go.micro.client", err.Error())
|
return errors.InternalServerError("go.micro.client", err.Error())
|
||||||
}
|
}
|
||||||
|
|
||||||
if len(service.Nodes) == 0 {
|
node, err := nodeSelector(service)
|
||||||
return errors.NotFound("go.micro.client", "Service not found")
|
if err != nil {
|
||||||
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
n := rand.Int() % len(service.Nodes)
|
|
||||||
node := service.Nodes[n]
|
|
||||||
|
|
||||||
address := node.Address
|
address := node.Address
|
||||||
if node.Port > 0 {
|
if node.Port > 0 {
|
||||||
address = fmt.Sprintf("%s:%d", address, node.Port)
|
address = fmt.Sprintf("%s:%d", address, node.Port)
|
||||||
@ -154,13 +146,11 @@ func (r *rpcClient) Stream(ctx context.Context, request Request, responseChan in
|
|||||||
return nil, errors.InternalServerError("go.micro.client", err.Error())
|
return nil, errors.InternalServerError("go.micro.client", err.Error())
|
||||||
}
|
}
|
||||||
|
|
||||||
if len(service.Nodes) == 0 {
|
node, err := nodeSelector(service)
|
||||||
return nil, errors.NotFound("go.micro.client", "Service not found")
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
}
|
}
|
||||||
|
|
||||||
n := rand.Int() % len(service.Nodes)
|
|
||||||
node := service.Nodes[n]
|
|
||||||
|
|
||||||
address := node.Address
|
address := node.Address
|
||||||
if node.Port > 0 {
|
if node.Port > 0 {
|
||||||
address = fmt.Sprintf("%s:%d", address, node.Port)
|
address = fmt.Sprintf("%s:%d", address, node.Port)
|
||||||
|
@ -39,6 +39,11 @@ var (
|
|||||||
EnvVar: "MICRO_SERVER_NAME",
|
EnvVar: "MICRO_SERVER_NAME",
|
||||||
Usage: "Name of the server. go.micro.srv.example",
|
Usage: "Name of the server. go.micro.srv.example",
|
||||||
},
|
},
|
||||||
|
cli.StringFlag{
|
||||||
|
Name: "server_version",
|
||||||
|
EnvVar: "MICRO_SERVER_VERSION",
|
||||||
|
Usage: "Version of the server. 1.1.0",
|
||||||
|
},
|
||||||
cli.StringFlag{
|
cli.StringFlag{
|
||||||
Name: "server_id",
|
Name: "server_id",
|
||||||
EnvVar: "MICRO_SERVER_ID",
|
EnvVar: "MICRO_SERVER_ID",
|
||||||
@ -178,6 +183,7 @@ func Setup(c *cli.Context) error {
|
|||||||
|
|
||||||
server.DefaultServer = server.NewServer(
|
server.DefaultServer = server.NewServer(
|
||||||
server.Name(c.String("server_name")),
|
server.Name(c.String("server_name")),
|
||||||
|
server.Version(c.String("server_version")),
|
||||||
server.Id(c.String("server_id")),
|
server.Id(c.String("server_id")),
|
||||||
server.Address(c.String("server_address")),
|
server.Address(c.String("server_address")),
|
||||||
server.Metadata(metadata),
|
server.Metadata(metadata),
|
||||||
|
@ -13,7 +13,7 @@ type consulRegistry struct {
|
|||||||
Client *consul.Client
|
Client *consul.Client
|
||||||
|
|
||||||
mtx sync.RWMutex
|
mtx sync.RWMutex
|
||||||
services map[string]*Service
|
services map[string][]*Service
|
||||||
}
|
}
|
||||||
|
|
||||||
func encodeEndpoints(en []*Endpoint) []string {
|
func encodeEndpoints(en []*Endpoint) []string {
|
||||||
@ -80,7 +80,7 @@ func newConsulRegistry(addrs []string, opts ...Option) Registry {
|
|||||||
cr := &consulRegistry{
|
cr := &consulRegistry{
|
||||||
Address: config.Address,
|
Address: config.Address,
|
||||||
Client: client,
|
Client: client,
|
||||||
services: make(map[string]*Service),
|
services: make(map[string][]*Service),
|
||||||
}
|
}
|
||||||
|
|
||||||
return cr
|
return cr
|
||||||
@ -115,7 +115,7 @@ func (c *consulRegistry) Register(s *Service) error {
|
|||||||
Node: node.Id,
|
Node: node.Id,
|
||||||
Address: node.Address,
|
Address: node.Address,
|
||||||
Service: &consul.AgentService{
|
Service: &consul.AgentService{
|
||||||
ID: node.Id,
|
ID: s.Version,
|
||||||
Service: s.Name,
|
Service: s.Name,
|
||||||
Port: node.Port,
|
Port: node.Port,
|
||||||
Tags: tags,
|
Tags: tags,
|
||||||
@ -125,7 +125,7 @@ func (c *consulRegistry) Register(s *Service) error {
|
|||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
func (c *consulRegistry) GetService(name string) (*Service, error) {
|
func (c *consulRegistry) GetService(name string) ([]*Service, error) {
|
||||||
c.mtx.RLock()
|
c.mtx.RLock()
|
||||||
service, ok := c.services[name]
|
service, ok := c.services[name]
|
||||||
c.mtx.RUnlock()
|
c.mtx.RUnlock()
|
||||||
@ -139,24 +139,47 @@ func (c *consulRegistry) GetService(name string) (*Service, error) {
|
|||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
|
|
||||||
cs := &Service{}
|
serviceMap := map[string]*Service{}
|
||||||
|
|
||||||
for _, s := range rsp {
|
for _, s := range rsp {
|
||||||
if s.ServiceName != name {
|
if s.ServiceName != name {
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
|
|
||||||
cs.Endpoints = decodeEndpoints(s.ServiceTags)
|
id := s.Node
|
||||||
cs.Name = s.ServiceName
|
key := s.ServiceID
|
||||||
cs.Nodes = append(cs.Nodes, &Node{
|
version := s.ServiceID
|
||||||
Id: s.ServiceID,
|
|
||||||
|
// We're adding service version but
|
||||||
|
// don't want to break backwards compatibility
|
||||||
|
if id == version {
|
||||||
|
key = "default"
|
||||||
|
version = ""
|
||||||
|
}
|
||||||
|
|
||||||
|
svc, ok := serviceMap[key]
|
||||||
|
if !ok {
|
||||||
|
svc = &Service{
|
||||||
|
Endpoints: decodeEndpoints(s.ServiceTags),
|
||||||
|
Name: s.ServiceName,
|
||||||
|
Version: version,
|
||||||
|
}
|
||||||
|
serviceMap[key] = svc
|
||||||
|
}
|
||||||
|
|
||||||
|
svc.Nodes = append(svc.Nodes, &Node{
|
||||||
|
Id: id,
|
||||||
Address: s.Address,
|
Address: s.Address,
|
||||||
Port: s.ServicePort,
|
Port: s.ServicePort,
|
||||||
Metadata: decodeMetadata(s.ServiceTags),
|
Metadata: decodeMetadata(s.ServiceTags),
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
return cs, nil
|
var services []*Service
|
||||||
|
for _, service := range serviceMap {
|
||||||
|
services = append(services, service)
|
||||||
|
}
|
||||||
|
return services, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (c *consulRegistry) ListServices() ([]*Service, error) {
|
func (c *consulRegistry) ListServices() ([]*Service, error) {
|
||||||
@ -167,8 +190,8 @@ func (c *consulRegistry) ListServices() ([]*Service, error) {
|
|||||||
var services []*Service
|
var services []*Service
|
||||||
|
|
||||||
if len(serviceMap) > 0 {
|
if len(serviceMap) > 0 {
|
||||||
for _, service := range services {
|
for _, service := range serviceMap {
|
||||||
services = append(services, service)
|
services = append(services, service...)
|
||||||
}
|
}
|
||||||
return services, nil
|
return services, nil
|
||||||
}
|
}
|
||||||
|
@ -39,13 +39,34 @@ func (cw *consulWatcher) serviceHandler(idx uint64, data interface{}) {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
cs := &Service{}
|
serviceMap := map[string]*Service{}
|
||||||
|
serviceName := ""
|
||||||
|
|
||||||
for _, e := range entries {
|
for _, e := range entries {
|
||||||
cs.Endpoints = decodeEndpoints(e.Service.Tags)
|
serviceName = e.Service.Service
|
||||||
cs.Name = e.Service.Service
|
id := e.Node.Node
|
||||||
cs.Nodes = append(cs.Nodes, &Node{
|
key := e.Service.Service + e.Service.ID
|
||||||
Id: e.Service.ID,
|
version := e.Service.ID
|
||||||
|
|
||||||
|
// We're adding service version but
|
||||||
|
// don't want to break backwards compatibility
|
||||||
|
if id == version {
|
||||||
|
key = e.Service.Service + "default"
|
||||||
|
version = ""
|
||||||
|
}
|
||||||
|
|
||||||
|
svc, ok := serviceMap[key]
|
||||||
|
if !ok {
|
||||||
|
svc = &Service{
|
||||||
|
Endpoints: decodeEndpoints(e.Service.Tags),
|
||||||
|
Name: e.Service.Service,
|
||||||
|
Version: version,
|
||||||
|
}
|
||||||
|
serviceMap[key] = svc
|
||||||
|
}
|
||||||
|
|
||||||
|
svc.Nodes = append(svc.Nodes, &Node{
|
||||||
|
Id: id,
|
||||||
Address: e.Node.Address,
|
Address: e.Node.Address,
|
||||||
Port: e.Service.Port,
|
Port: e.Service.Port,
|
||||||
Metadata: decodeMetadata(e.Service.Tags),
|
Metadata: decodeMetadata(e.Service.Tags),
|
||||||
@ -53,7 +74,11 @@ func (cw *consulWatcher) serviceHandler(idx uint64, data interface{}) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
cw.Registry.mtx.Lock()
|
cw.Registry.mtx.Lock()
|
||||||
cw.Registry.services[cs.Name] = cs
|
var services []*Service
|
||||||
|
for _, service := range serviceMap {
|
||||||
|
services = append(services, service)
|
||||||
|
}
|
||||||
|
cw.Registry.services[serviceName] = services
|
||||||
cw.Registry.mtx.Unlock()
|
cw.Registry.mtx.Unlock()
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -20,7 +20,7 @@ type etcdRegistry struct {
|
|||||||
client etcd.KeysAPI
|
client etcd.KeysAPI
|
||||||
|
|
||||||
sync.RWMutex
|
sync.RWMutex
|
||||||
services map[string]*registry.Service
|
services map[string][]*registry.Service
|
||||||
}
|
}
|
||||||
|
|
||||||
func encode(s *registry.Service) string {
|
func encode(s *registry.Service) string {
|
||||||
@ -85,7 +85,7 @@ func (e *etcdRegistry) Register(s *registry.Service) error {
|
|||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (e *etcdRegistry) GetService(name string) (*registry.Service, error) {
|
func (e *etcdRegistry) GetService(name string) ([]*registry.Service, error) {
|
||||||
e.RLock()
|
e.RLock()
|
||||||
service, ok := e.services[name]
|
service, ok := e.services[name]
|
||||||
e.RUnlock()
|
e.RUnlock()
|
||||||
@ -99,23 +99,35 @@ func (e *etcdRegistry) GetService(name string) (*registry.Service, error) {
|
|||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
|
|
||||||
s := ®istry.Service{}
|
serviceMap := map[string]*registry.Service{}
|
||||||
|
|
||||||
for _, n := range rsp.Node.Nodes {
|
for _, n := range rsp.Node.Nodes {
|
||||||
if n.Dir {
|
if n.Dir {
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
sn := decode(n.Value)
|
sn := decode(n.Value)
|
||||||
s.Name = sn.Name
|
|
||||||
s.Version = sn.Version
|
s, ok := serviceMap[sn.Version]
|
||||||
s.Metadata = sn.Metadata
|
if !ok {
|
||||||
s.Endpoints = sn.Endpoints
|
s = ®istry.Service{
|
||||||
|
Name: sn.Name,
|
||||||
|
Version: sn.Version,
|
||||||
|
Metadata: sn.Metadata,
|
||||||
|
Endpoints: sn.Endpoints,
|
||||||
|
}
|
||||||
|
serviceMap[s.Version] = s
|
||||||
|
}
|
||||||
|
|
||||||
for _, node := range sn.Nodes {
|
for _, node := range sn.Nodes {
|
||||||
s.Nodes = append(s.Nodes, node)
|
s.Nodes = append(s.Nodes, node)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
return s, nil
|
var services []*registry.Service
|
||||||
|
for _, service := range serviceMap {
|
||||||
|
services = append(services, service)
|
||||||
|
}
|
||||||
|
return services, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (e *etcdRegistry) ListServices() ([]*registry.Service, error) {
|
func (e *etcdRegistry) ListServices() ([]*registry.Service, error) {
|
||||||
@ -126,8 +138,8 @@ func (e *etcdRegistry) ListServices() ([]*registry.Service, error) {
|
|||||||
var services []*registry.Service
|
var services []*registry.Service
|
||||||
|
|
||||||
if len(serviceMap) > 0 {
|
if len(serviceMap) > 0 {
|
||||||
for _, service := range services {
|
for _, service := range serviceMap {
|
||||||
services = append(services, service)
|
services = append(services, service...)
|
||||||
}
|
}
|
||||||
return services, nil
|
return services, nil
|
||||||
}
|
}
|
||||||
@ -139,15 +151,10 @@ func (e *etcdRegistry) ListServices() ([]*registry.Service, error) {
|
|||||||
|
|
||||||
for _, node := range rsp.Node.Nodes {
|
for _, node := range rsp.Node.Nodes {
|
||||||
service := ®istry.Service{}
|
service := ®istry.Service{}
|
||||||
|
|
||||||
for _, n := range node.Nodes {
|
for _, n := range node.Nodes {
|
||||||
i := decode(n.Value)
|
i := decode(n.Value)
|
||||||
service.Name = i.Name
|
service.Name = i.Name
|
||||||
for _, in := range i.Nodes {
|
|
||||||
service.Nodes = append(service.Nodes, in)
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
services = append(services, service)
|
services = append(services, service)
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -179,7 +186,7 @@ func NewRegistry(addrs []string, opt ...registry.Option) registry.Registry {
|
|||||||
|
|
||||||
e := &etcdRegistry{
|
e := &etcdRegistry{
|
||||||
client: etcd.NewKeysAPI(c),
|
client: etcd.NewKeysAPI(c),
|
||||||
services: make(map[string]*registry.Service),
|
services: make(map[string][]*registry.Service),
|
||||||
}
|
}
|
||||||
|
|
||||||
return e
|
return e
|
||||||
|
@ -11,6 +11,77 @@ type etcdWatcher struct {
|
|||||||
stop chan bool
|
stop chan bool
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func addNodes(old, neu []*registry.Node) []*registry.Node {
|
||||||
|
for _, n := range neu {
|
||||||
|
var seen bool
|
||||||
|
for i, o := range old {
|
||||||
|
if o.Id == n.Id {
|
||||||
|
seen = true
|
||||||
|
old[i] = n
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if !seen {
|
||||||
|
old = append(old, n)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return old
|
||||||
|
}
|
||||||
|
|
||||||
|
func addServices(old, neu []*registry.Service) []*registry.Service {
|
||||||
|
for _, s := range neu {
|
||||||
|
var seen bool
|
||||||
|
for i, o := range old {
|
||||||
|
if o.Version == s.Version {
|
||||||
|
s.Nodes = addNodes(o.Nodes, s.Nodes)
|
||||||
|
seen = true
|
||||||
|
old[i] = s
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if !seen {
|
||||||
|
old = append(old, s)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return old
|
||||||
|
}
|
||||||
|
|
||||||
|
func delNodes(old, del []*registry.Node) []*registry.Node {
|
||||||
|
var nodes []*registry.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
|
||||||
|
}
|
||||||
|
|
||||||
|
func delServices(old, del []*registry.Service) []*registry.Service {
|
||||||
|
var services []*registry.Service
|
||||||
|
for i, o := range old {
|
||||||
|
var rem bool
|
||||||
|
for _, s := range del {
|
||||||
|
if o.Version == s.Version {
|
||||||
|
old[i].Nodes = delNodes(o.Nodes, s.Nodes)
|
||||||
|
if len(old[i].Nodes) == 0 {
|
||||||
|
rem = true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if !rem {
|
||||||
|
services = append(services, o)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return services
|
||||||
|
}
|
||||||
|
|
||||||
func newEtcdWatcher(r *etcdRegistry) (registry.Watcher, error) {
|
func newEtcdWatcher(r *etcdRegistry) (registry.Watcher, error) {
|
||||||
ew := &etcdWatcher{
|
ew := &etcdWatcher{
|
||||||
registry: r,
|
registry: r,
|
||||||
@ -53,7 +124,7 @@ func (e *etcdWatcher) watch(ctx context.Context, w etcd.Watcher) {
|
|||||||
service, ok := e.registry.services[s.Name]
|
service, ok := e.registry.services[s.Name]
|
||||||
if !ok {
|
if !ok {
|
||||||
if rsp.Action == "create" {
|
if rsp.Action == "create" {
|
||||||
e.registry.services[s.Name] = s
|
e.registry.services[s.Name] = []*registry.Service{s}
|
||||||
}
|
}
|
||||||
e.registry.Unlock()
|
e.registry.Unlock()
|
||||||
continue
|
continue
|
||||||
@ -61,22 +132,14 @@ func (e *etcdWatcher) watch(ctx context.Context, w etcd.Watcher) {
|
|||||||
|
|
||||||
switch rsp.Action {
|
switch rsp.Action {
|
||||||
case "delete":
|
case "delete":
|
||||||
var nodes []*registry.Node
|
services := delServices(service, []*registry.Service{s})
|
||||||
for _, node := range service.Nodes {
|
if len(services) > 0 {
|
||||||
var seen bool
|
e.registry.services[s.Name] = services
|
||||||
for _, n := range s.Nodes {
|
} else {
|
||||||
if node.Id == n.Id {
|
delete(e.registry.services, s.Name)
|
||||||
seen = true
|
|
||||||
break
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if !seen {
|
|
||||||
nodes = append(nodes, node)
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
service.Nodes = nodes
|
|
||||||
case "create":
|
case "create":
|
||||||
service.Nodes = append(service.Nodes, s.Nodes...)
|
e.registry.services[s.Name] = addServices(service, []*registry.Service{s})
|
||||||
}
|
}
|
||||||
|
|
||||||
e.registry.Unlock()
|
e.registry.Unlock()
|
||||||
|
@ -35,7 +35,7 @@ type memoryRegistry struct {
|
|||||||
updates chan *update
|
updates chan *update
|
||||||
|
|
||||||
sync.RWMutex
|
sync.RWMutex
|
||||||
services map[string]*registry.Service
|
services map[string][]*registry.Service
|
||||||
}
|
}
|
||||||
|
|
||||||
type update struct {
|
type update struct {
|
||||||
@ -96,7 +96,7 @@ func (d *delegate) LocalState(join bool) []byte {
|
|||||||
}
|
}
|
||||||
|
|
||||||
syncCh := make(chan *registry.Service, 1)
|
syncCh := make(chan *registry.Service, 1)
|
||||||
m := map[string]*registry.Service{}
|
m := map[string][]*registry.Service{}
|
||||||
|
|
||||||
d.updates <- &update{
|
d.updates <- &update{
|
||||||
Action: syncAction,
|
Action: syncAction,
|
||||||
@ -104,7 +104,7 @@ func (d *delegate) LocalState(join bool) []byte {
|
|||||||
}
|
}
|
||||||
|
|
||||||
for s := range syncCh {
|
for s := range syncCh {
|
||||||
m[s.Name] = s
|
m[s.Name] = append(m[s.Name], s)
|
||||||
}
|
}
|
||||||
|
|
||||||
b, _ := json.Marshal(m)
|
b, _ := json.Marshal(m)
|
||||||
@ -119,16 +119,18 @@ func (d *delegate) MergeRemoteState(buf []byte, join bool) {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
var m map[string]*registry.Service
|
var m map[string][]*registry.Service
|
||||||
if err := json.Unmarshal(buf, &m); err != nil {
|
if err := json.Unmarshal(buf, &m); err != nil {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
for _, service := range m {
|
for _, services := range m {
|
||||||
d.updates <- &update{
|
for _, service := range services {
|
||||||
Action: addAction,
|
d.updates <- &update{
|
||||||
Service: service,
|
Action: addAction,
|
||||||
sync: nil,
|
Service: service,
|
||||||
|
sync: nil,
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -138,19 +140,31 @@ func (m *memoryRegistry) run() {
|
|||||||
switch u.Action {
|
switch u.Action {
|
||||||
case addAction:
|
case addAction:
|
||||||
m.Lock()
|
m.Lock()
|
||||||
m.services[u.Service.Name] = u.Service
|
if service, ok := m.services[u.Service.Name]; !ok {
|
||||||
|
m.services[u.Service.Name] = []*registry.Service{u.Service}
|
||||||
|
} else {
|
||||||
|
m.services[u.Service.Name] = addServices(service, []*registry.Service{u.Service})
|
||||||
|
}
|
||||||
m.Unlock()
|
m.Unlock()
|
||||||
case delAction:
|
case delAction:
|
||||||
m.Lock()
|
m.Lock()
|
||||||
delete(m.services, u.Service.Name)
|
if service, ok := m.services[u.Service.Name]; ok {
|
||||||
|
if services := delServices(service, []*registry.Service{u.Service}); len(services) == 0 {
|
||||||
|
delete(m.services, u.Service.Name)
|
||||||
|
} else {
|
||||||
|
m.services[u.Service.Name] = services
|
||||||
|
}
|
||||||
|
}
|
||||||
m.Unlock()
|
m.Unlock()
|
||||||
case syncAction:
|
case syncAction:
|
||||||
if u.sync == nil {
|
if u.sync == nil {
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
m.RLock()
|
m.RLock()
|
||||||
for _, service := range m.services {
|
for _, services := range m.services {
|
||||||
u.sync <- service
|
for _, service := range services {
|
||||||
|
u.sync <- service
|
||||||
|
}
|
||||||
}
|
}
|
||||||
m.RUnlock()
|
m.RUnlock()
|
||||||
close(u.sync)
|
close(u.sync)
|
||||||
@ -160,7 +174,11 @@ func (m *memoryRegistry) run() {
|
|||||||
|
|
||||||
func (m *memoryRegistry) Register(s *registry.Service) error {
|
func (m *memoryRegistry) Register(s *registry.Service) error {
|
||||||
m.Lock()
|
m.Lock()
|
||||||
m.services[s.Name] = s
|
if service, ok := m.services[s.Name]; !ok {
|
||||||
|
m.services[s.Name] = []*registry.Service{s}
|
||||||
|
} else {
|
||||||
|
m.services[s.Name] = addServices(service, []*registry.Service{s})
|
||||||
|
}
|
||||||
m.Unlock()
|
m.Unlock()
|
||||||
|
|
||||||
b, _ := json.Marshal([]*update{
|
b, _ := json.Marshal([]*update{
|
||||||
@ -180,7 +198,13 @@ func (m *memoryRegistry) Register(s *registry.Service) error {
|
|||||||
|
|
||||||
func (m *memoryRegistry) Deregister(s *registry.Service) error {
|
func (m *memoryRegistry) Deregister(s *registry.Service) error {
|
||||||
m.Lock()
|
m.Lock()
|
||||||
delete(m.services, s.Name)
|
if service, ok := m.services[s.Name]; ok {
|
||||||
|
if services := delServices(service, []*registry.Service{s}); len(services) == 0 {
|
||||||
|
delete(m.services, s.Name)
|
||||||
|
} else {
|
||||||
|
m.services[s.Name] = services
|
||||||
|
}
|
||||||
|
}
|
||||||
m.Unlock()
|
m.Unlock()
|
||||||
|
|
||||||
b, _ := json.Marshal([]*update{
|
b, _ := json.Marshal([]*update{
|
||||||
@ -198,7 +222,7 @@ func (m *memoryRegistry) Deregister(s *registry.Service) error {
|
|||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (m *memoryRegistry) GetService(name string) (*registry.Service, error) {
|
func (m *memoryRegistry) GetService(name string) ([]*registry.Service, error) {
|
||||||
m.RLock()
|
m.RLock()
|
||||||
service, ok := m.services[name]
|
service, ok := m.services[name]
|
||||||
m.RUnlock()
|
m.RUnlock()
|
||||||
@ -212,7 +236,7 @@ func (m *memoryRegistry) ListServices() ([]*registry.Service, error) {
|
|||||||
var services []*registry.Service
|
var services []*registry.Service
|
||||||
m.RLock()
|
m.RLock()
|
||||||
for _, service := range m.services {
|
for _, service := range m.services {
|
||||||
services = append(services, service)
|
services = append(services, service...)
|
||||||
}
|
}
|
||||||
m.RUnlock()
|
m.RUnlock()
|
||||||
return services, nil
|
return services, nil
|
||||||
@ -226,6 +250,77 @@ func (w *watcher) Stop() {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func addNodes(old, neu []*registry.Node) []*registry.Node {
|
||||||
|
for _, n := range neu {
|
||||||
|
var seen bool
|
||||||
|
for i, o := range old {
|
||||||
|
if o.Id == n.Id {
|
||||||
|
seen = true
|
||||||
|
old[i] = n
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if !seen {
|
||||||
|
old = append(old, n)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return old
|
||||||
|
}
|
||||||
|
|
||||||
|
func addServices(old, neu []*registry.Service) []*registry.Service {
|
||||||
|
for _, s := range neu {
|
||||||
|
var seen bool
|
||||||
|
for i, o := range old {
|
||||||
|
if o.Version == s.Version {
|
||||||
|
s.Nodes = addNodes(o.Nodes, s.Nodes)
|
||||||
|
seen = true
|
||||||
|
old[i] = s
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if !seen {
|
||||||
|
old = append(old, s)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return old
|
||||||
|
}
|
||||||
|
|
||||||
|
func delNodes(old, del []*registry.Node) []*registry.Node {
|
||||||
|
var nodes []*registry.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
|
||||||
|
}
|
||||||
|
|
||||||
|
func delServices(old, del []*registry.Service) []*registry.Service {
|
||||||
|
var services []*registry.Service
|
||||||
|
for i, o := range old {
|
||||||
|
var rem bool
|
||||||
|
for _, s := range del {
|
||||||
|
if o.Version == s.Version {
|
||||||
|
old[i].Nodes = delNodes(o.Nodes, s.Nodes)
|
||||||
|
if len(old[i].Nodes) == 0 {
|
||||||
|
rem = true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if !rem {
|
||||||
|
services = append(services, o)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return services
|
||||||
|
}
|
||||||
|
|
||||||
func NewRegistry(addrs []string, opt ...registry.Option) registry.Registry {
|
func NewRegistry(addrs []string, opt ...registry.Option) registry.Registry {
|
||||||
cAddrs := []string{}
|
cAddrs := []string{}
|
||||||
hostname, _ := os.Hostname()
|
hostname, _ := os.Hostname()
|
||||||
@ -246,7 +341,7 @@ func NewRegistry(addrs []string, opt ...registry.Option) registry.Registry {
|
|||||||
|
|
||||||
mr := &memoryRegistry{
|
mr := &memoryRegistry{
|
||||||
broadcasts: broadcasts,
|
broadcasts: broadcasts,
|
||||||
services: make(map[string]*registry.Service),
|
services: make(map[string][]*registry.Service),
|
||||||
updates: updates,
|
updates: updates,
|
||||||
}
|
}
|
||||||
|
|
||||||
|
78
registry/memory/memory_test.go
Normal file
78
registry/memory/memory_test.go
Normal file
@ -0,0 +1,78 @@
|
|||||||
|
package memory
|
||||||
|
|
||||||
|
import (
|
||||||
|
"testing"
|
||||||
|
|
||||||
|
"github.com/myodc/go-micro/registry"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestDelServices(t *testing.T) {
|
||||||
|
services := []*registry.Service{
|
||||||
|
{
|
||||||
|
Name: "foo",
|
||||||
|
Version: "1.0.0",
|
||||||
|
Nodes: []*registry.Node{
|
||||||
|
{
|
||||||
|
Id: "foo-123",
|
||||||
|
Address: "localhost",
|
||||||
|
Port: 9999,
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
Name: "foo",
|
||||||
|
Version: "1.0.0",
|
||||||
|
Nodes: []*registry.Node{
|
||||||
|
{
|
||||||
|
Id: "foo-123",
|
||||||
|
Address: "localhost",
|
||||||
|
Port: 6666,
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
servs := delServices([]*registry.Service{services[0]}, []*registry.Service{services[1]})
|
||||||
|
if i := len(servs); i > 0 {
|
||||||
|
t.Errorf("Expected 0 nodes, got %d: %+v", i, servs)
|
||||||
|
}
|
||||||
|
t.Logf("Services %+v", servs)
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestDelNodes(t *testing.T) {
|
||||||
|
services := []*registry.Service{
|
||||||
|
{
|
||||||
|
Name: "foo",
|
||||||
|
Version: "1.0.0",
|
||||||
|
Nodes: []*registry.Node{
|
||||||
|
{
|
||||||
|
Id: "foo-123",
|
||||||
|
Address: "localhost",
|
||||||
|
Port: 9999,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
Id: "foo-321",
|
||||||
|
Address: "localhost",
|
||||||
|
Port: 6666,
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
Name: "foo",
|
||||||
|
Version: "1.0.0",
|
||||||
|
Nodes: []*registry.Node{
|
||||||
|
{
|
||||||
|
Id: "foo-123",
|
||||||
|
Address: "localhost",
|
||||||
|
Port: 6666,
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
nodes := delNodes(services[0].Nodes, services[1].Nodes)
|
||||||
|
if i := len(nodes); i != 1 {
|
||||||
|
t.Errorf("Expected only 1 node, got %d: %+v", i, nodes)
|
||||||
|
}
|
||||||
|
t.Logf("Nodes %+v", nodes)
|
||||||
|
}
|
@ -3,7 +3,7 @@ package registry
|
|||||||
type Registry interface {
|
type Registry interface {
|
||||||
Register(*Service) error
|
Register(*Service) error
|
||||||
Deregister(*Service) error
|
Deregister(*Service) error
|
||||||
GetService(string) (*Service, error)
|
GetService(string) ([]*Service, error)
|
||||||
ListServices() ([]*Service, error)
|
ListServices() ([]*Service, error)
|
||||||
Watch() (Watcher, error)
|
Watch() (Watcher, error)
|
||||||
}
|
}
|
||||||
@ -32,7 +32,7 @@ func Deregister(s *Service) error {
|
|||||||
return DefaultRegistry.Deregister(s)
|
return DefaultRegistry.Deregister(s)
|
||||||
}
|
}
|
||||||
|
|
||||||
func GetService(name string) (*Service, error) {
|
func GetService(name string) ([]*Service, error) {
|
||||||
return DefaultRegistry.GetService(name)
|
return DefaultRegistry.GetService(name)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user