From bac76884d9a34046723457d3e441185893d82812 Mon Sep 17 00:00:00 2001
From: kercylan98 <kercylan@gmail.com>
Date: Sat, 6 May 2023 11:45:39 +0800
Subject: [PATCH] =?UTF-8?q?=E6=9C=8D=E5=8A=A1=E5=99=A8=E5=A4=9A=E6=A0=B8?=
 =?UTF-8?q?=E6=A8=A1=E5=BC=8F?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 server/options.go | 18 ++++++++++++++++++
 server/server.go  | 10 ++++++++--
 2 files changed, 26 insertions(+), 2 deletions(-)

diff --git a/server/options.go b/server/options.go
index 055a0e2..dfe75f6 100644
--- a/server/options.go
+++ b/server/options.go
@@ -1,5 +1,10 @@
 package server
 
+import (
+	"go.uber.org/zap"
+	"minotaur/utils/log"
+)
+
 type Option func(srv *Server)
 
 // WithProd 通过生产模式运行服务器
@@ -17,3 +22,16 @@ func WithMessageBufferSize(size int) Option {
 		srv.messagePoolSize = size
 	}
 }
+
+// WithMultiCore 通过特定核心数量运行服务器，默认为单核
+//   - count > 1 的情况下，将会有对应数量的 goroutine 来处理消息
+//   - 注意：HTTP和GRPC网络模式下不会生效
+func WithMultiCore(count int) Option {
+	return func(srv *Server) {
+		srv.core = count
+		if srv.core < 1 {
+			log.Warn("WithMultiCore", zap.Int("count", count), zap.String("tips", "wrong core count configuration, corrected to 1, currently in single-core mode"))
+			srv.core = 1
+		}
+	}
+}
diff --git a/server/server.go b/server/server.go
index 6f568ce..477a83c 100644
--- a/server/server.go
+++ b/server/server.go
@@ -26,6 +26,7 @@ func New(network Network, options ...Option) *Server {
 		event:   &event{},
 		network: network,
 		options: options,
+		core:    1,
 	}
 	server.event.Server = server
 
@@ -61,6 +62,7 @@ type Server struct {
 	initMessageChannel bool                                // 消息管道是否已经初始化
 	multiple           bool                                // 是否为多服务器模式下运行
 	prod               bool                                // 是否为生产模式
+	core               int                                 // 消息处理核心数
 }
 
 // Run 使用特定地址运行服务器
@@ -103,8 +105,12 @@ func (slf *Server) Run(addr string) error {
 		if callback != nil {
 			go callback()
 		}
-		for message := range slf.messageChannel {
-			slf.dispatchMessage(message)
+		for i := 0; i < slf.core; i++ {
+			go func() {
+				for message := range slf.messageChannel {
+					slf.dispatchMessage(message)
+				}
+			}()
 		}
 	}