update core

core/buffer_pool.go

@@ -4,20 +4,32 @@ import (
 	"sync"
 )
 
-const defaultBufferSize = 2 * 1024
+var pool *sync.Pool
 
-var bufPool = sync.Pool{New: func() interface{} { return make([]byte, defaultBufferSize) }}
+const BufSize = 2 * 1024
 
-func newBytes(size int) []byte {
-	if size <= defaultBufferSize {
-		return bufPool.Get().([]byte)
+func SetBufferPool(p *sync.Pool) {
+	pool = p
+}
+
+func NewBytes(size int) []byte {
+	if size <= BufSize {
+		return pool.Get().([]byte)
 	} else {
 		return make([]byte, size)
 	}
 }
 
-func freeBytes(b []byte) {
-	if len(b) >= defaultBufferSize {
-		bufPool.Put(b)
+func FreeBytes(b []byte) {
+	if len(b) >= BufSize {
+		pool.Put(b)
 	}
 }
+
+func init() {
+	SetBufferPool(&sync.Pool{
+		New: func() interface{} {
+			return make([]byte, BufSize)
+		},
+	})
+}