paulmataruso/komari
1
0
Fork 0
mirror of https://github.com/komari-monitor/komari.git synced 2025-10-23 03:31:56 +00:00
Code Issues Packages Projects Releases Wiki Activity

refactor: 移除不必要的 nezhaBoot 变量，更新 StopNezhaCompat 以强制停止服务器

Browse Source
This commit is contained in:
Akizon77
2025-09-13 15:52:57 +08:00
parent a06a5c9e72
commit 7138c22d02
1 changed files with 2 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
cmd/nezha_compat.go
View File

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