From ee7ad88fd2da56018e70c61984db853d44915de4 Mon Sep 17 00:00:00 2001
From: Rueian <rueiancsie@gmail.com>
Date: Thu, 13 Jun 2024 23:25:49 +0800
Subject: [PATCH] feat: set 1s limit for graceful shutdown

---
 pipe.go | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/pipe.go b/pipe.go
index 8e8840b4..4091f4e6 100644
--- 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)