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) +}