From f1488fe7d35bc2331a67277ba336a214565e71f3 Mon Sep 17 00:00:00 2001 From: kercylan98 Date: Tue, 9 May 2023 18:17:47 +0800 Subject: [PATCH] =?UTF-8?q?=E6=B6=88=E6=81=AF=E6=B1=A0=E9=BB=98=E8=AE=A4?= =?UTF-8?q?=E5=A4=A7=E5=B0=8F=E8=B0=83=E6=95=B4?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- server/options.go | 5 ++++- server/server.go | 2 +- 2 files changed, 5 insertions(+), 2 deletions(-) diff --git a/server/options.go b/server/options.go index 83ac09a..371d5fa 100644 --- a/server/options.go +++ b/server/options.go @@ -48,10 +48,13 @@ func WithWebsocketMessageType(messageTypes ...int) Option { } // WithMessageBufferSize 通过特定的消息缓冲池大小运行服务器 -// - 默认大小为 1024 +// - 默认大小为 4096 * 1024 // - 消息数量超出这个值的时候，消息处理将会造成更大的开销（频繁创建新的结构体），同时服务器将输出警告内容 func WithMessageBufferSize(size int) Option { return func(srv *Server) { + if size <= 0 { + return + } srv.messagePoolSize = size } } diff --git a/server/server.go b/server/server.go index 9b179a6..336129c 100644 --- a/server/server.go +++ b/server/server.go @@ -87,7 +87,7 @@ func (slf *Server) Run(addr string) error { var connectionInitHandle = func(callback func()) { slf.initMessageChannel = true if slf.messagePoolSize <= 0 { - slf.messagePoolSize = 1024 + slf.messagePoolSize = 4096 * 1024 } slf.messagePool = synchronization.NewPool[*message](slf.messagePoolSize, func() *message {