perf: set 1s limit for graceful shutdown to avoid waiting for a stuck…

… redis (#572) Signed-off-by: Rueian <rueiancsie@gmail.com>
redis · Jun 17, 2024 · 4d59a58 · 4d59a58
1 parent 463b6c3
commit 4d59a58
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/pipe.go b/pipe.go
@@ -1480,7 +1480,10 @@ func (p *pipe) Close() {
 			p.background()
 		}
 		if block == 1 && (stopping1 || stopping2) { // make sure there is no block cmd
-			<-p.queue.PutOne(cmds.PingCmd)
+			select {
+			case <-p.queue.PutOne(cmds.PingCmd):
+			case <-time.After(time.Second):
+			}
 		}
 	}
 	atomic.AddInt32(&p.waits, -1)