From 9656d80c2b9c66eb8b01780578dc3cfe155a0e77 Mon Sep 17 00:00:00 2001 From: oxedions Date: Thu, 7 Apr 2022 09:48:07 +0200 Subject: [PATCH] Feat: ensure gotty shutdown if main command exit --- server/handlers.go | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/server/handlers.go b/server/handlers.go index ba9771d..79854ab 100644 --- a/server/handlers.go +++ b/server/handlers.go @@ -46,7 +46,7 @@ func (server *Server) generateHandleWS(ctx context.Context, cancel context.Cance closeReason, r.RemoteAddr, num, server.options.MaxConnection, ) - if server.options.Once { + if (server.options.Once) || (closeReason == "local command") { cancel() } }()