github
/
nps
mirror of https://github.com/ehang-io/nps
1
0
Fork 0
Code Issues Releases Wiki Activity

reduce memory allocate

pull/265/head
ffdfgdfg 2019-10-08 21:41:25 +08:00
parent d9f9dc6acb
commit 4c8d7b0738
2 changed files with 9 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
lib/common/const.go
View File

@ -48,5 +48,5 @@ const (
MUX_CONN_CLOSE MUX_CONN_CLOSE
MUX_PING_RETURN MUX_PING_RETURN
MUX_PING int32 = -1 MUX_PING int32 = -1
MAXIMUM_SEGMENT_SIZE = 4096 - 16 - 32 - 32 - 8 MAXIMUM_SEGMENT_SIZE = PoolSizeWindow
) )

16
lib/common/pool.go
View File

@ -9,7 +9,8 @@ const PoolSize = 64 * 1024
const PoolSizeSmall = 100 const PoolSizeSmall = 100
const PoolSizeUdp = 1472 const PoolSizeUdp = 1472
const PoolSizeCopy = 32 << 10 const PoolSizeCopy = 32 << 10
const PoolSizeWindow = 1<<16 - 1 const PoolSizeBuffer = 4096
const PoolSizeWindow = PoolSizeBuffer - 16 - 32 - 32 - 8
var BufPool = sync.Pool{ var BufPool = sync.Pool{
New: func() interface{} { New: func() interface{} {
@ -92,18 +93,18 @@ type windowBufferPool struct {
func (Self *windowBufferPool) New() { func (Self *windowBufferPool) New() {
Self.pool = sync.Pool{ Self.pool = sync.Pool{
New: func() interface{} { New: func() interface{} {
return make([]byte, 0, PoolSizeWindow) return make([]byte, PoolSizeWindow, PoolSizeWindow)
}, },
} }
} }
func (Self *windowBufferPool) Get() (buf []byte) { func (Self *windowBufferPool) Get() (buf []byte) {
buf = Self.pool.Get().([]byte) buf = Self.pool.Get().([]byte)
return buf[:0] return buf[:PoolSizeWindow]
} }
func (Self *windowBufferPool) Put(x []byte) { func (Self *windowBufferPool) Put(x []byte) {
if cap(x) == PoolSizeWindow { if len(x) == PoolSizeWindow {
Self.pool.Put(x[:PoolSizeWindow]) // make buf to full Self.pool.Put(x[:PoolSizeWindow]) // make buf to full
} else { } else {
x = nil x = nil
@ -117,7 +118,7 @@ type bufferPool struct {
func (Self *bufferPool) New() { func (Self *bufferPool) New() {
Self.pool = sync.Pool{ Self.pool = sync.Pool{
New: func() interface{} { New: func() interface{} {
return new(bytes.Buffer) return bytes.NewBuffer(make([]byte, 0, PoolSizeBuffer))
}, },
} }
} }
@ -146,13 +147,12 @@ func (Self *muxPackagerPool) New() {
func (Self *muxPackagerPool) Get() *MuxPackager { func (Self *muxPackagerPool) Get() *MuxPackager {
pack := Self.pool.Get().(*MuxPackager) pack := Self.pool.Get().(*MuxPackager)
buf := CopyBuff.Get() pack.Content = WindowBuff.Get()
pack.Content = buf
return pack return pack
} }
func (Self *muxPackagerPool) Put(pack *MuxPackager) { func (Self *muxPackagerPool) Put(pack *MuxPackager) {
CopyBuff.Put(pack.Content) WindowBuff.Put(pack.Content)
Self.pool.Put(pack) Self.pool.Put(pack)
} }