refine buffer allocation

common/buf/buffer_pool.go

@@ -4,39 +4,29 @@ import (
 	"sync"
 )
 
-// Pool provides functionality to generate and recycle buffers on demand.
-type Pool struct {
-	allocator *sync.Pool
-}
-
-// NewPool creates a SyncPool with given buffer size.
-func NewPool(bufferSize uint32) *Pool {
-	pool := &Pool{
-		allocator: &sync.Pool{
-			New: func() interface{} { return make([]byte, bufferSize) },
-		},
-	}
-	return pool
-}
-
-// Allocate either returns a unused buffer from the pool, or generates a new one from system.
-func (p *Pool) Allocate() *Buffer {
-	return &Buffer{
-		v:    p.allocator.Get().([]byte),
-		pool: p,
-	}
-}
-
-// // Free recycles the given buffer.
-func (p *Pool) Free(buffer *Buffer) {
-	if buffer.v != nil {
-		p.allocator.Put(buffer.v)
-	}
-}
-
 const (
 	// Size of a regular buffer.
 	Size = 2 * 1024
 )
 
-var mediumPool = NewPool(Size)
+func createAllocFunc(size uint32) func() interface{} {
+	return func() interface{} {
+		return make([]byte, size)
+	}
+}
+
+var pool2k = &sync.Pool{
+	New: createAllocFunc(2 * 1024),
+}
+
+var pool8k = &sync.Pool{
+	New: createAllocFunc(8 * 1024),
+}
+
+var pool64k = &sync.Pool{
+	New: createAllocFunc(64 * 1024),
+}
+
+var pool128k = &sync.Pool{
+	New: createAllocFunc(128 * 1024),
+}