From b96af6e9731e5cbda2186dbe9e24899a9c803dee Mon Sep 17 00:00:00 2001
From: kercylan98 <kercylan@gmail.com>
Date: Tue, 23 May 2023 15:03:08 +0800
Subject: [PATCH] =?UTF-8?q?=E6=94=AF=E6=8C=81=E6=8E=A7=E5=88=B6=E5=8F=B0?=
 =?UTF-8?q?=E5=91=BD=E4=BB=A4?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 server/event.go | 33 +++++++++++++++------------------
 1 file changed, 15 insertions(+), 18 deletions(-)

diff --git a/server/event.go b/server/event.go
index f5dfb13..c5e5481 100644
--- a/server/event.go
+++ b/server/event.go
@@ -48,21 +48,7 @@ func (slf *event) RegConsoleCommandEvent(command string, handle ConsoleCommandEv
 			for {
 				var input string
 				_, _ = fmt.Scanln(&input)
-				handles, exist := slf.consoleCommandEventHandles[input]
-				if !exist {
-					switch input {
-					case "exit", "quit", "close", "shutdown", "EXIT", "QUIT", "CLOSE", "SHUTDOWN":
-						log.Info("Console", zap.String("Receive", input), zap.String("Action", "Shutdown"))
-						slf.Server.Shutdown(nil)
-						return
-					}
-					log.Warn("Server", zap.String("Command", "unregistered"))
-				} else {
-					for _, handle := range handles {
-						handle(slf.Server)
-					}
-				}
-
+				slf.OnConsoleCommandEvent(input)
 			}
 		}()
 	})
@@ -82,9 +68,20 @@ func (slf *event) RegStartBeforeEvent(handle StartBeforeEventHandle) {
 	log.Info("Server", zap.String("RegEvent", runtimes.CurrentRunningFuncName()), zap.String("handle", reflect.TypeOf(handle).String()))
 }
 
-func (slf *event) OnStartBeforeEvent() {
-	for _, handle := range slf.startBeforeEventHandles {
-		handle(slf.Server)
+func (slf *event) OnStartBeforeEvent(command string) {
+	handles, exist := slf.consoleCommandEventHandles[command]
+	if !exist {
+		switch command {
+		case "exit", "quit", "close", "shutdown", "EXIT", "QUIT", "CLOSE", "SHUTDOWN":
+			log.Info("Console", zap.String("Receive", command), zap.String("Action", "Shutdown"))
+			slf.Server.Shutdown(nil)
+			return
+		}
+		log.Warn("Server", zap.String("Command", "unregistered"))
+	} else {
+		for _, handle := range handles {
+			handle(slf.Server)
+		}
 	}
 }