feat: set 1s limit for graceful shutdown

redis · Jun 13, 2024 · ee7ad88 · ee7ad88
1 parent 16d5f13
commit ee7ad88
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/pipe.go b/pipe.go
@@ -1481,7 +1481,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)