fix buffer pool

core/buffer_pool.go

@@ -0,0 +1,23 @@
+package core
+
+import (
+	"sync"
+)
+
+const defaultBufferSize = 2 * 1024
+
+var bufPool = sync.Pool{New: func() interface{} { return make([]byte, defaultBufferSize) }}
+
+func newBytes(size int) []byte {
+	if size <= defaultBufferSize {
+		return bufPool.Get().([]byte)
+	} else {
+		return make([]byte, size)
+	}
+}
+
+func freeBytes(b []byte) {
+	if len(b) >= defaultBufferSize {
+		bufPool.Put(b)
+	}
+}

core/output_export.go

@@ -19,10 +19,10 @@ func output(p *C.struct_pbuf) C.err_t {
 		buf := (*[1 << 30]byte)(unsafe.Pointer(p.payload))[:totlen:totlen]
 		OutputFn(buf[:totlen])
 	} else {
-		buf := NewBytes(totlen)
+		buf := newBytes(totlen)
 		C.pbuf_copy_partial(p, unsafe.Pointer(&buf[0]), p.tot_len, 0) // data copy here!
 		OutputFn(buf[:totlen])
-		FreeBytes(buf)
+		freeBytes(buf)
 	}
 	return C.ERR_OK
 }

core/tcp_callback_export.go

@@ -81,8 +81,8 @@ func tcpRecvFn(arg unsafe.Pointer, tpcb *C.struct_tcp_pcb, p *C.struct_pbuf, err
 	if p.tot_len == p.len {
 		buf = (*[1 << 30]byte)(unsafe.Pointer(p.payload))[:totlen:totlen]
 	} else {
-		buf = NewBytes(totlen)
-		defer FreeBytes(buf)
+		buf = newBytes(totlen)
+		defer freeBytes(buf)
 		C.pbuf_copy_partial(p, unsafe.Pointer(&buf[0]), p.tot_len, 0)
 	}
 

core/udp_callback_export.go

@@ -53,8 +53,8 @@ func udpRecvFn(arg unsafe.Pointer, pcb *C.struct_udp_pcb, p *C.struct_pbuf, addr
 	if p.tot_len == p.len {
 		buf = (*[1 << 30]byte)(unsafe.Pointer(p.payload))[:totlen:totlen]
 	} else {
-		buf = NewBytes(totlen)
-		defer FreeBytes(buf)
+		buf = newBytes(totlen)
+		defer freeBytes(buf)
 		C.pbuf_copy_partial(p, unsafe.Pointer(&buf[0]), p.tot_len, 0)
 	}