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

优化网络内存池

Browse Source
This commit is contained in:
boyce
2020-12-23 17:15:58 +08:00
parent a34546ff68
commit 73be7dcedf
2 changed files with 3 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
network/slicepool.go
View File

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

5
network/tcp_msg.go
View File

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