add func: SetBufferPool

lxzan · Jul 25, 2024 · 4cdceff · 4cdceff
1 parent b85d0de
commit 4cdceff
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 4 deletions.
diff --git a/init.go b/init.go
@@ -3,7 +3,17 @@ package gws
 import "github.com/lxzan/gws/internal"
 
 var (
-	framePadding  = frameHeader{}                         // 帧头填充物
-	binaryPool    = internal.NewBufferPool(128, 256*1024) // 缓冲池
-	defaultLogger = new(stdLogger)                        // 默认日志工具
+	framePadding  = frameHeader{}            // 帧头填充物
+	binaryPool    = new(internal.BufferPool) // 内存池
+	defaultLogger = new(stdLogger)           // 默认日志工具
 )
+
+func init() {
+	SetBufferPool(256 * 1024)
+}
+
+// SetBufferPool set up the memory pool, any memory that exceeds maxSize will not be reclaimed.
+// 设置内存池, 超过maxSize将不会被回收.
+func SetBufferPool(maxSize uint32) {
+	binaryPool = internal.NewBufferPool(128, maxSize)
+}
diff --git a/writer.go b/writer.go
@@ -119,7 +119,7 @@ func (c *Conn) genFrame(opcode Opcode, payload internal.Payload, isBroadcast boo
 		return nil, internal.CloseMessageTooLarge
 	}
 
-	var buf = binaryPool.Get(n*11/10 + frameHeaderSize)
+	var buf = binaryPool.Get(n*105/100 + frameHeaderSize)
 	buf.Write(framePadding[0:])
 
 	if c.pd.Enabled && opcode.isDataFrame() && n >= c.pd.Threshold {