Ma/origin
1
0
Fork 0
mirror of https://github.com/duanhf2012/origin.git synced 2026-02-03 22:45:13 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

tcp消息内存优化

Browse Source
This commit is contained in:
lifeiyi
2020-12-14 17:56:31 +08:00
parent a5666a6766
commit 9e11b045e2
2 changed files with 14 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
network/slicepool.go
View File

@@ -5,7 +5,8 @@ import (
)
type INetMempool interface {
MakeByteSlice(size int) []byte
MakeReadByteSlice(size int) []byte
MakeWriteByteSlice(size int) []byte
ReleaseByteSlice(byteBuff []byte) bool
}
@@ -80,6 +81,14 @@ func (areaPool *memAreaPool) MakeByteSlice(size int) []byte{
return nil
}
func (areaPool *memAreaPool) MakeReadByteSlice(size int) []byte{
return areaPool.MakeByteSlice(size)
}
func (areaPool *memAreaPool) MakeWriteByteSlice(size int) []byte{
return areaPool.MakeByteSlice(size)
}
func (areaPool *memAreaPool) ReleaseByteSlice(byteBuff []byte) bool {
for i:=0;i<len(memAreaPoolList);i++{
if cap(byteBuff) <= memAreaPoolList[i].maxAreaValue {

8
network/tcp_msg.go
View File

@@ -101,13 +101,13 @@ func (p *MsgParser) Read(conn *TCPConn) ([]byte, error) {
// data
//msgData := make([]byte, msgLen)
msgData := p.MakeByteSlice(int(msgLen))
if _, err := io.ReadFull(conn, msgData); err != nil {
msgData := p.MakeReadByteSlice(int(msgLen))
if _, err := io.ReadFull(conn, msgData[:msgLen]); err != nil {
p.ReleaseByteSlice(msgData)
return nil, err
}
return msgData, nil
return msgData[:msgLen], nil
}
// goroutine safe
@@ -126,7 +126,7 @@ func (p *MsgParser) Write(conn *TCPConn, args ...[]byte) error {
}
//msg := make([]byte, uint32(p.lenMsgLen)+msgLen)
msg := p.MakeByteSlice(p.lenMsgLen+int(msgLen))
msg := p.MakeWriteByteSlice(p.lenMsgLen+int(msgLen))
// write len
switch p.lenMsgLen {
case 1: