From 3b66dc1101e1e8e13b4ec810cdea72ec7c504b2f Mon Sep 17 00:00:00 2001
From: Colin Alworth <colinalworth@deephaven.io>
Date: Mon, 10 Jan 2022 14:01:16 -0600
Subject: [PATCH] Record sent bytes for when we do flowcontrol

---
 .../java/io/grpc/servlet/web/websocket/WebsocketStreamImpl.java  | 1 +
 1 file changed, 1 insertion(+)

diff --git a/grpc-java/grpc-servlet-websocket-jakarta/src/main/java/io/grpc/servlet/web/websocket/WebsocketStreamImpl.java b/grpc-java/grpc-servlet-websocket-jakarta/src/main/java/io/grpc/servlet/web/websocket/WebsocketStreamImpl.java
index a4bdfef00ef..459dbdb2018 100644
--- a/grpc-java/grpc-servlet-websocket-jakarta/src/main/java/io/grpc/servlet/web/websocket/WebsocketStreamImpl.java
+++ b/grpc-java/grpc-servlet-websocket-jakarta/src/main/java/io/grpc/servlet/web/websocket/WebsocketStreamImpl.java
@@ -201,6 +201,7 @@ public void writeFrame(@Nullable WritableBuffer frame, boolean flush, int numMes
                             ByteBuffer.wrap(((ByteArrayWritableBuffer) frame).bytes, 0, frame.readableBytes());
 
                     websocketSession.getBasicRemote().sendBinary(payload);
+                    transportState.runOnTransportThread(() -> transportState.onSentBytes(numBytes));
                 }
 
             } catch (IOException e) {