Record sent bytes for when we do flowcontrol

deephaven · Jan 10, 2022 · 3b66dc1 · 3b66dc1
1 parent e72b839
commit 3b66dc1
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/...et-websocket-jakarta/src/main/java/io/grpc/servlet/web/websocket/WebsocketStreamImpl.java b/...et-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) {