From 6eb3e8ee30680259be315c50b9ba14938f255f1a Mon Sep 17 00:00:00 2001
From: Armon Dadgar <armon.dadgar@gmail.com>
Date: Wed, 16 Jul 2014 15:11:45 -0700
Subject: [PATCH] agent: Fixing issue with excessive failed node filtering

---
 command/agent/dns.go      |  2 ++
 command/agent/dns_test.go | 42 ++++++++++++++++++++++++++++++++++++++-
 2 files changed, 43 insertions(+), 1 deletion(-)

diff --git a/command/agent/dns.go b/command/agent/dns.go
index 697f314fcb..4a0bd14f0c 100644
--- a/command/agent/dns.go
+++ b/command/agent/dns.go
@@ -471,6 +471,7 @@ RPC:
 // health checks to prevent routing to unhealthy nodes
 func (d *DNSServer) filterServiceNodes(nodes structs.CheckServiceNodes) structs.CheckServiceNodes {
 	n := len(nodes)
+OUTER:
 	for i := 0; i < n; i++ {
 		node := nodes[i]
 		for _, check := range node.Checks {
@@ -480,6 +481,7 @@ func (d *DNSServer) filterServiceNodes(nodes structs.CheckServiceNodes) structs.
 				nodes[i], nodes[n-1] = nodes[n-1], structs.CheckServiceNode{}
 				n--
 				i--
+				continue OUTER
 			}
 		}
 	}
diff --git a/command/agent/dns_test.go b/command/agent/dns_test.go
index 3a2804a4c5..cf1ae791f7 100644
--- a/command/agent/dns_test.go
+++ b/command/agent/dns_test.go
@@ -651,6 +651,40 @@ func TestDNS_ServiceLookup_FilterCritical(t *testing.T) {
 		t.Fatalf("err: %v", err)
 	}
 
+	args3 := &structs.RegisterRequest{
+		Datacenter: "dc1",
+		Node:       "bar",
+		Address:    "127.0.0.2",
+		Service: &structs.NodeService{
+			Service: "db",
+			Tags:    []string{"master"},
+			Port:    12345,
+		},
+		Check: &structs.HealthCheck{
+			CheckID:   "db",
+			Name:      "db",
+			ServiceID: "db",
+			Status:    structs.HealthCritical,
+		},
+	}
+	if err := srv.agent.RPC("Catalog.Register", args3, &out); err != nil {
+		t.Fatalf("err: %v", err)
+	}
+
+	args4 := &structs.RegisterRequest{
+		Datacenter: "dc1",
+		Node:       "baz",
+		Address:    "127.0.0.3",
+		Service: &structs.NodeService{
+			Service: "db",
+			Tags:    []string{"master"},
+			Port:    12345,
+		},
+	}
+	if err := srv.agent.RPC("Catalog.Register", args4, &out); err != nil {
+		t.Fatalf("err: %v", err)
+	}
+
 	m := new(dns.Msg)
 	m.SetQuestion("db.service.consul.", dns.TypeANY)
 
@@ -662,9 +696,15 @@ func TestDNS_ServiceLookup_FilterCritical(t *testing.T) {
 	}
 
 	// Should get no answer since we are failing!
-	if len(in.Answer) != 0 {
+	if len(in.Answer) != 1 {
 		t.Fatalf("Bad: %#v", in)
 	}
+
+	resp := in.Answer[0]
+	aRec := resp.(*dns.A)
+	if aRec.A.String() != "127.0.0.3" {
+		t.Fatalf("Bad: %#v", in.Answer[0])
+	}
 }
 
 func TestDNS_ServiceLookup_Randomize(t *testing.T) {