From 7138c22d0253a24a38ba8ebe30a448520a35fd51 Mon Sep 17 00:00:00 2001
From: Akizon77 <i@akz.moe>
Date: Sat, 13 Sep 2025 15:52:57 +0800
Subject: [PATCH] =?UTF-8?q?refactor:=20=E7=A7=BB=E9=99=A4=E4=B8=8D?=
 =?UTF-8?q?=E5=BF=85=E8=A6=81=E7=9A=84=20nezhaBoot=20=E5=8F=98=E9=87=8F?=
 =?UTF-8?q?=EF=BC=8C=E6=9B=B4=E6=96=B0=20StopNezhaCompat=20=E4=BB=A5?=
 =?UTF-8?q?=E5=BC=BA=E5=88=B6=E5=81=9C=E6=AD=A2=E6=9C=8D=E5=8A=A1=E5=99=A8?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 cmd/nezha_compat.go | 7 ++-----
 1 file changed, 2 insertions(+), 5 deletions(-)

diff --git a/cmd/nezha_compat.go b/cmd/nezha_compat.go
index 39bc6df..dd00c69 100644
--- a/cmd/nezha_compat.go
+++ b/cmd/nezha_compat.go
@@ -68,7 +68,6 @@ var (
 	nezhaSrv   *grpc.Server
 	nezhaLis   net.Listener
 	nezhaOnceM sync.Mutex
-	nezhaBoot  uint64
 )
 
 // StartNezhaCompat starts the Nezha compatible gRPC server asynchronously.
@@ -101,7 +100,6 @@ func StartNezhaCompat(addr string) error {
 	proto.RegisterNezhaServiceServer(gs, sImpl)
 	nezhaSrv = gs
 	nezhaLis = lis
-	nezhaBoot = boot
 	go func() {
 		if err := gs.Serve(lis); err != nil {
 			log.Printf("Nezha compat gRPC server stopped: %v", err)
@@ -118,15 +116,14 @@ func StopNezhaCompat() error {
 	if nezhaSrv == nil {
 		return errors.New("nezha compat server not running")
 	}
-	// GracefulStop allows inflight RPCs to finish.
-	nezhaSrv.GracefulStop()
+	// 强制立即断开所有连接与流，不等待在途 RPC 完成。
+	nezhaSrv.Stop()
 	// Listener close (Serve already returns after GracefulStop, but close to be explicit)
 	if nezhaLis != nil {
 		_ = nezhaLis.Close()
 	}
 	nezhaSrv = nil
 	nezhaLis = nil
-	nezhaBoot = 0
 	log.Printf("Nezha compat gRPC stopped")
 	return nil
 }