diff --git a/server/options.go b/server/options.go index 389bd6c..055a0e2 100644 --- a/server/options.go +++ b/server/options.go @@ -8,3 +8,12 @@ func WithProd() Option { srv.prod = true } } + +// WithMessageBufferSize 通过特定的消息缓冲池大小运行服务器 +// - 默认大小为 1024 +// - 消息数量超出这个值的时候,消息处理将会造成更大的开销(频繁创建新的结构体),同时服务器将输出警告内容 +func WithMessageBufferSize(size int) Option { + return func(srv *Server) { + srv.messagePoolSize = size + } +} diff --git a/server/server.go b/server/server.go index d19ccd1..2e6737c 100644 --- a/server/server.go +++ b/server/server.go @@ -45,6 +45,7 @@ type Server struct { grpcServer *grpc.Server // GRPC模式下的服务器 gServer *gNet // TCP或UDP模式下的服务器 messagePool *synchronization.Pool[*message] // 消息池 + messagePoolSize int // 消息池大小 messageChannel chan *message // 消息管道 initMessageChannel bool // 消息管道是否已经初始化 multiple bool // 是否为多服务器模式下运行 @@ -73,7 +74,10 @@ func (slf *Server) Run(addr string) error { var connectionInitHandle = func(callback func()) { slf.connections = synchronization.NewMap[string, *Conn]() slf.initMessageChannel = true - slf.messagePool = synchronization.NewPool[*message](1024, + if slf.messagePoolSize <= 0 { + slf.messagePoolSize = 1024 + } + slf.messagePool = synchronization.NewPool[*message](slf.messagePoolSize, func() *message { return &message{} }, func(data *message) {