use cache reduce malloc

model/cache_pool.go

@@ -0,0 +1,66 @@
+package model
+
+import (
+	"fmt"
+	"reflect"
+	"unsafe"
+
+	log "github.com/sirupsen/logrus"
+)
+
+
+type sliceBufferPool struct {
+	queue chan []byte
+	bufferSize int
+	name string
+}
+
+func NewSliceBufferPool(name string, bufferSize int) (sbp *sliceBufferPool) {
+	return &sliceBufferPool{
+		queue: make(chan []byte, 512),
+		bufferSize: bufferSize,
+		name: name,
+	}
+}
+
+func (sbp *sliceBufferPool) Enqueue(buffer []byte)  {
+	defer func() {
+		log.Debugf("after enqueue from %s, there is %d elements", sbp.name, len(sbp.queue))
+	}()
+
+	if cap(buffer) < 1 {
+		return
+	}
+
+	sbp.queue <- buffer
+}
+
+func (sbp *sliceBufferPool) DequeueWithInit(initSize int) (buffer []byte)  {
+	if initSize >= sbp.bufferSize {
+		panic(fmt.Sprintf("package size bigger than max buffer size need deal:%d",
+			sbp.bufferSize))
+	}
+
+	defer func() {
+		// reset cache byte
+		pbytes := (*reflect.SliceHeader)(unsafe.Pointer(&buffer))
+		pbytes.Len = initSize
+	}()
+
+	buffer = sbp.Dequeue()
+	return
+}
+
+func (sbp *sliceBufferPool) Dequeue() (buffer []byte)  {
+	defer func() {
+		log.Debugf("after dequeue from %s, there is %d elements", sbp.name, len(sbp.queue))
+	}()
+
+	select {
+	case buffer = <- sbp.queue:
+		return
+	default:
+		buffer = make([]byte, 0, sbp.bufferSize)
+		return
+	}
+}