Add service random

2015-12-09 12:44:38 +00:00 · 2015-12-09 12:44:38 +00:00 · a2c862dd21
commit a2c862dd21
parent 29ee9e2753
1 changed files with 9 additions and 1 deletions
--- a/registry/random_selector.go
+++ b/registry/random_selector.go
@ -48,7 +48,15 @@ func (r *randomSelector) Select(service string, opts ...SelectOption) (SelectNex
 	}

 	return func() (*Node, error) {
-		return nodes[rand.Int()%len(nodes)], nil
+		i := rand.Int()
+		j := i % len(services)
+
+		if len(services[j].Nodes) == 0 {
+			return nil, ErrNotFound
+		}
+
+		k := i % len(services[j].Nodes)
+		return services[j].Nodes[k], nil
 	}, nil
 }