支持控制台命令

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]
+				slf.OnConsoleCommandEvent(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)
-					}
-				}
-
 			}
 		}()
 	})
@@ -82,11 +68,22 @@ 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() {
+func (slf *event) OnStartBeforeEvent(command string) {
-	for _, handle := range slf.startBeforeEventHandles {
+	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)
 		}
 	}
+}
 
 // RegStartFinishEvent 在服务器启动完成时将立刻执行被注册的事件处理函数
 func (slf *event) RegStartFinishEvent(handle StartFinishEventHandle) {