HBASE-28017 Set request and response size metrics in NettyRpcDuplexHandler (#5473)

rmdmattingly · Ray Mattingly · bbeaudreault · commit 078119a06804 · 2023-10-24T10:28:07.000-04:00
Co-authored-by: Ray Mattingly &lt;rmattingly@hubspot.com&gt;
Signed-off-by: Duo Zhang &lt;zhangduo@apache.org&gt;
Signed-off-by: Bryan Beaudreault &lt;bbeaudreault@apache.org&gt;
diff --git a/hbase-client/src/main/java/org/apache/hadoop/hbase/ipc/NettyRpcDuplexHandler.java b/hbase-client/src/main/java/org/apache/hadoop/hbase/ipc/NettyRpcDuplexHandler.java
@@ -110,6 +110,7 @@ private void writeRequest(ChannelHandlerContext ctx, Call call, ChannelPromise p
       } else {
         ctx.write(buf, promise);
       }
+      call.callStats.setRequestSizeBytes(totalSize);
     }
   }
 
@@ -193,6 +194,7 @@ private void readResponse(ChannelHandlerContext ctx, ByteBuf buf) throws IOExcep
       }
       return;
     }
+    call.callStats.setResponseSizeBytes(totalSize);
     if (remoteExc != null) {
       call.setException(remoteExc);
       return;

Original file line number	Diff line number	Diff line change
`@@ -110,6 +110,7 @@ private void writeRequest(ChannelHandlerContext ctx, Call call, ChannelPromise p`
`110`	`110`	`} else {`
`111`	`111`	`ctx.write(buf, promise);`
`112`	`112`	`}`
	`113`	`+ call.callStats.setRequestSizeBytes(totalSize);`
`113`	`114`	`}`
`114`	`115`	`}`
`115`	`116`
`@@ -193,6 +194,7 @@ private void readResponse(ChannelHandlerContext ctx, ByteBuf buf) throws IOExcep`
`193`	`194`	`}`
`194`	`195`	`return;`
`195`	`196`	`}`
	`197`	`+ call.callStats.setResponseSizeBytes(totalSize);`
`196`	`198`	`if (remoteExc != null) {`
`197`	`199`	`call.setException(remoteExc);`
`198`	`200`	`return;`