bytepool: enforce equal length and cap

This commit is contained in:
Jimmy Zelinskie 2016-09-05 12:30:03 -04:00
parent c4706022d7
commit 57ee2d0c90
2 changed files with 4 additions and 4 deletions

View file

@ -7,11 +7,11 @@ type BytePool struct {
sync.Pool sync.Pool
} }
// New allocates a new BytePool with slices of the provided capacity. // New allocates a new BytePool with slices of equal length and capacity.
func New(length, capacity int) *BytePool { func New(length int) *BytePool {
var bp BytePool var bp BytePool
bp.Pool.New = func() interface{} { bp.Pool.New = func() interface{} {
return make([]byte, length, capacity) return make([]byte, length, length)
} }
return &bp return &bp
} }

View file

@ -93,7 +93,7 @@ func (t *Frontend) ListenAndServe() error {
} }
defer t.socket.Close() defer t.socket.Close()
pool := bytepool.New(2048, 2048) pool := bytepool.New(2048)
for { for {
// Check to see if we need to shutdown. // Check to see if we need to shutdown.