From 11b214b445a53099aed175ef3ef386263b05fbda Mon Sep 17 00:00:00 2001 From: boyce Date: Thu, 24 Dec 2020 11:19:55 +0800 Subject: [PATCH] =?UTF-8?q?=E6=96=B0=E5=A2=9E=E7=BD=91=E7=BB=9C=E5=B1=82bu?= =?UTF-8?q?ff=E5=86=85=E5=AD=98=E6=B1=A0=E5=9B=9E=E6=94=B6=E6=8E=A5?= =?UTF-8?q?=E5=8F=A3?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- network/tcp_server.go | 4 ++++ sysservice/tcpservice/tcpservice.go | 8 ++++++++ 2 files changed, 12 insertions(+) diff --git a/network/tcp_server.go b/network/tcp_server.go index 20e3944..14bf138 100644 --- a/network/tcp_server.go +++ b/network/tcp_server.go @@ -68,6 +68,10 @@ func (server *TCPServer) SetNetMempool(mempool INetMempool){ server.msgParser.INetMempool = mempool } +func (server *TCPServer) GetNetMempool() INetMempool{ + return server.msgParser.INetMempool +} + func (server *TCPServer) run() { server.wgLn.Add(1) defer server.wgLn.Done() diff --git a/sysservice/tcpservice/tcpservice.go b/sysservice/tcpservice/tcpservice.go index 60440db..aebc4ec 100644 --- a/sysservice/tcpservice/tcpservice.go +++ b/sysservice/tcpservice/tcpservice.go @@ -275,3 +275,11 @@ func (tcpService *TcpService) GetConnNum() int { func (server *TcpService) SetNetMempool(mempool network.INetMempool){ server.tcpServer.SetNetMempool(mempool) } + +func (server *TcpService) GetNetMempool() network.INetMempool{ + return server.tcpServer.GetNetMempool() +} + +func (server *TcpService) ReleaseNetMem(byteBuff []byte) { + server.tcpServer.GetNetMempool().ReleaseByteSlice(byteBuff) +}