From e83bc43dfa771b0ad6727418cc5b0e49d02d2201 Mon Sep 17 00:00:00 2001
From: Asim <asim@chuhnk.me>
Date: Wed, 9 Dec 2015 00:18:06 +0000
Subject: [PATCH] Fix failing test

---
 .../client/node_selector/node_selector.go     |  87 --------------
 examples/client/selector/selector.go          | 108 ++++++++++++++++++
 2 files changed, 108 insertions(+), 87 deletions(-)
 delete mode 100644 examples/client/node_selector/node_selector.go
 create mode 100644 examples/client/selector/selector.go

diff --git a/examples/client/node_selector/node_selector.go b/examples/client/node_selector/node_selector.go
deleted file mode 100644
index 9181dc8c..00000000
--- a/examples/client/node_selector/node_selector.go
+++ /dev/null
@@ -1,87 +0,0 @@
-package main
-
-import (
-	"fmt"
-	"math/rand"
-	"time"
-
-	"github.com/micro/go-micro/client"
-	"github.com/micro/go-micro/cmd"
-	"github.com/micro/go-micro/errors"
-	example "github.com/micro/go-micro/examples/server/proto/example"
-	"github.com/micro/go-micro/registry"
-	"golang.org/x/net/context"
-)
-
-func init() {
-	rand.Seed(time.Now().Unix())
-}
-
-// Built in random hashed node selector
-type nodeSelector struct {
-	r registry.Registry
-}
-
-func (n *nodeSelector) Select(ctx context.Context, req client.Request) (*registry.Node, error) {
-	service, err := n.r.GetService(req.Service())
-	if err != nil {
-		return nil, errors.InternalServerError("go.micro.client", err.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")
-	}
-
-	k := i % len(service[j].Nodes)
-	return service[j].Nodes[k], nil
-}
-
-func (n *nodeSelector) Response(node *registry.Node, err error) {
-	return
-}
-
-func (n *nodeSelector) Reset() {
-	return
-}
-
-// Return a new random node selector
-func RandomSelector(r registry.Registry) client.NodeSelector {
-	return &nodeSelector{r}
-}
-
-func call(i int) {
-	// Create new request to service go.micro.srv.example, method Example.Call
-	req := client.NewRequest("go.micro.srv.example", "Example.Call", &example.Request{
-		Name: "John",
-	})
-
-	rsp := &example.Response{}
-
-	// Call service
-	if err := client.Call(context.Background(), req, rsp); err != nil {
-		fmt.Println("call err: ", err, rsp)
-		return
-	}
-
-	fmt.Println("Call:", i, "rsp:", rsp.Msg)
-}
-
-func main() {
-	cmd.Init()
-
-	client.DefaultClient = client.NewClient(
-		client.Select(RandomSelector),
-	)
-
-	fmt.Println("\n--- Call example ---\n")
-	for i := 0; i < 10; i++ {
-		call(i)
-	}
-}
diff --git a/examples/client/selector/selector.go b/examples/client/selector/selector.go
new file mode 100644
index 00000000..7bb0e220
--- /dev/null
+++ b/examples/client/selector/selector.go
@@ -0,0 +1,108 @@
+package main
+
+import (
+	"fmt"
+	"math/rand"
+	"time"
+
+	"github.com/micro/go-micro/client"
+	"github.com/micro/go-micro/cmd"
+	example "github.com/micro/go-micro/examples/server/proto/example"
+	"github.com/micro/go-micro/registry"
+	"golang.org/x/net/context"
+)
+
+func init() {
+	rand.Seed(time.Now().Unix())
+}
+
+// Built in random hashed node selector
+type firstNodeSelector struct {
+	opts registry.SelectorOptions
+}
+
+func (n *firstNodeSelector) Select(service string, opts ...registry.SelectOption) (registry.SelectNext, error) {
+	services, err := n.opts.Registry.GetService(service)
+	if err != nil {
+		return nil, err
+	}
+
+	if len(services) == 0 {
+		return nil, registry.ErrNotFound
+	}
+
+	var sopts registry.SelectOptions
+	for _, opt := range opts {
+		opt(&sopts)
+	}
+
+	for _, filter := range sopts.Filters {
+		services = filter(services)
+	}
+
+	if len(services) == 0 {
+		return nil, registry.ErrNotFound
+	}
+
+	if len(services[0].Nodes) == 0 {
+		return nil, registry.ErrNotFound
+	}
+
+	return func() (*registry.Node, error) {
+		return services[0].Nodes[0], nil
+	}, nil
+}
+
+func (n *firstNodeSelector) Mark(service string, node *registry.Node, err error) {
+	return
+}
+
+func (n *firstNodeSelector) Reset(service string) {
+	return
+}
+
+func (n *firstNodeSelector) Close() error {
+	return nil
+}
+
+// Return a new first node selector
+func FirstNodeSelector(opts ...registry.SelectorOption) registry.Selector {
+	var sopts registry.SelectorOptions
+	for _, opt := range opts {
+		opt(&sopts)
+	}
+	if sopts.Registry == nil {
+		sopts.Registry = registry.DefaultRegistry
+	}
+	return &firstNodeSelector{sopts}
+}
+
+func call(i int) {
+	// Create new request to service go.micro.srv.example, method Example.Call
+	req := client.NewRequest("go.micro.srv.example", "Example.Call", &example.Request{
+		Name: "John",
+	})
+
+	rsp := &example.Response{}
+
+	// Call service
+	if err := client.Call(context.Background(), req, rsp); err != nil {
+		fmt.Println("call err: ", err, rsp)
+		return
+	}
+
+	fmt.Println("Call:", i, "rsp:", rsp.Msg)
+}
+
+func main() {
+	cmd.Init()
+
+	client.DefaultClient = client.NewClient(
+		client.Selector(FirstNodeSelector()),
+	)
+
+	fmt.Println("\n--- Call example ---\n")
+	for i := 0; i < 10; i++ {
+		call(i)
+	}
+}