unblack list when err is nil

2015-12-09 02:42:02 +00:00 · 2015-12-09 02:42:02 +00:00 · 46905347bb
commit 46905347bb
parent f59787a906
1 changed files with 5 additions and 0 deletions
--- a/registry/black_list_selector.go
+++ b/registry/black_list_selector.go
@ -107,6 +107,11 @@ func (r *blackListSelector) Select(service string, opts ...SelectOption) (Select
 func (r *blackListSelector) Mark(service string, node *Node, err error) {
 	r.Lock()
 	defer r.Unlock()
+	if err == nil {
+		delete(r.bl, node.Id)
+		return
+	}
+
 	r.bl[node.Id] = blackListNode{
 		age:     time.Now().Add(time.Duration(r.ttl) * time.Second),
 		id:      node.Id,