Optimizing internal rpc

Author: dobyte

network/tcp/client_conn.go

@@ -277,31 +277,37 @@ func (c *clientConn) read() {
 
 			switch c.State() {
 			case network.ConnHanged:
+				buf.Release()
 				continue
 			case network.ConnClosed:
+				buf.Release()
 				return
 			default:
 				// ignore
 			}
 
 			// ignore empty packet
 			if buf.Len() == 0 {
+				buf.Release()
 				continue
 			}
 
 			isHeartbeat, err := packet.CheckHeartbeat(buf.Bytes())
 			if err != nil {
 				log.Errorf("check heartbeat message error: %v", err)
+				buf.Release()
 				continue
 			}
 
 			// ignore heartbeat packet
 			if isHeartbeat {
-				continue
-			}
-
-			if c.client.receiveHandler != nil {
-				c.client.receiveHandler(c, buf)
+				buf.Release()
+			} else {
+				if c.client.receiveHandler != nil {
+					c.client.receiveHandler(c, buf)
+				} else {
+					buf.Release()
+				}
 			}
 		}
 	}

network/tcp/server_conn.go

@@ -322,33 +322,40 @@ func (c *serverConn) read() {
 
 			switch c.State() {
 			case network.ConnHanged:
+				buf.Release()
 				continue
 			case network.ConnClosed:
+				buf.Release()
 				return
 			default:
 				// ignore
 			}
 
 			// ignore empty packet
 			if buf.Len() == 0 {
+				buf.Release()
 				continue
 			}
 
 			isHeartbeat, err := packet.CheckHeartbeat(buf.Bytes())
 			if err != nil {
 				log.Errorf("check heartbeat message error: %v", err)
+				buf.Release()
 				continue
 			}
 
 			// ignore heartbeat packet
 			if isHeartbeat {
+				buf.Release()
 				// responsive heartbeat
 				if c.connMgr.server.opts.heartbeatMechanism == RespHeartbeat {
 					c.sendHeartbeat(conn)
 				}
 			} else {
 				if c.connMgr.server.receiveHandler != nil {
 					c.connMgr.server.receiveHandler(c, buf)
+				} else {
+					buf.Release()
 				}
 			}
 		}