fix bug #799 (#848)

* fix bug #799 I have a test. Just close etcd, after 10 seconds, it will dead loop crazy, cpu100%. Need add a little guard code. * Update client.go
go-kit · Mar 12, 2019 · a7fe87b · a7fe87b
1 parent 90f6ad6
commit a7fe87b
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/sd/etcdv3/client.go b/sd/etcdv3/client.go
@@ -209,7 +209,11 @@ func (c *client) Register(s Service) error {
 	go func() {
 		for {
 			select {
-			case <-c.hbch:
+			case r := <-c.hbch:
+				// avoid dead loop when channel was closed
+				if r == nil {
+					return
+				}
 			case <-c.ctx.Done():
 				return
 			}