print better error when losing connection

2020-09-02 15:57:42 -07:00 · 2020-09-02 15:57:42 -07:00 · 09194aa613
commit 09194aa613
parent e58327d4ed
1 changed files with 4 additions and 1 deletions
--- a/cmd/lotus-shed/mempool-stats.go
+++ b/cmd/lotus-shed/mempool-stats.go
@ -106,7 +106,10 @@ var mpoolStatsCmd = &cli.Command{
 		tick := time.Tick(time.Second)
 		for {
 			select {
-			case u := <-updates:
+			case u, ok := <-updates:
+				if !ok {
+					return fmt.Errorf("connection with lotus node broke")
+				}
 				switch u.Type {
 				case lapi.MpoolAdd:
 					tracker[u.Message.Cid()] = &msgInfo{