extract byte buffer as interface

Author: mobus

examples/netepoll/main.go

@@ -8,6 +8,7 @@ import (
 	"syscall"
 	"time"
 
+	"github.com/oxtoacart/bpool"
 	"github.com/rcrowley/go-metrics"
 	"github.com/sunvim/utils/netpoll"
 )
@@ -23,9 +24,7 @@ func main() {
 	}()
 
 	var handler = &netpoll.DataHandler{
-		NoShared:   true,
-		NoCopy:     true,
-		BufferSize: 1024,
+		Pool: bpool.NewBytePool(1024, 12*1024),
 		HandlerFunc: func(req []byte) (res []byte) {
 			res = req
 			return

netpoll/handler.go

@@ -7,8 +7,6 @@ import (
 	"errors"
 	"net"
 	"sync"
-
-	"github.com/oxtoacart/bpool"
 )
 
 // ErrHandlerFunc is the error when the HandlerFunc is nil
@@ -70,9 +68,16 @@ func (h *ConnHandler) Serve(ctx Context) error {
 	return h.serve(ctx)
 }
 
+type BytePool interface {
+	Get() []byte
+	Put(b []byte)
+	NumPooled() int
+	Width() int
+}
+
 // DataHandler implements the Handler interface.
 type DataHandler struct {
-	Pool    *bpool.BytePool
+	Pool    BytePool
 	upgrade func(net.Conn) (net.Conn, error)
 	// HandlerFunc is the data Serve function.
 	HandlerFunc func(req []byte) (res []byte)
@@ -83,7 +88,7 @@ type context struct {
 	writing sync.Mutex
 	upgrade bool
 	conn    net.Conn
-	pool    *bpool.BytePool
+	pool    BytePool
 	buffer  []byte
 }