File tree Expand file tree Collapse file tree 1 file changed +6
-1
lines changed Expand file tree Collapse file tree 1 file changed +6
-1
lines changed Original file line number Diff line number Diff line change @@ -1371,8 +1371,11 @@ void TWriteSessionImpl::SendImpl() {
1371
1371
auto * writeRequest = clientMessage.mutable_write_request ();
1372
1372
1373
1373
ui32 prevCodec = 0 ;
1374
+
1375
+ ui64 currentSize = 0 ;
1376
+
1374
1377
// Send blocks while we can without messages reordering.
1375
- while (IsReadyToSendNextImpl () && clientMessage. ByteSizeLong () < GetMaxGrpcMessageSize ()) {
1378
+ while (IsReadyToSendNextImpl () && currentSize < GetMaxGrpcMessageSize ()) {
1376
1379
const auto & block = PackedMessagesToSend.top ();
1377
1380
Y_ABORT_UNLESS (block.Valid );
1378
1381
if (writeRequest->messages_size () > 0 && prevCodec != block.CodecID ) {
@@ -1420,6 +1423,8 @@ void TWriteSessionImpl::SendImpl() {
1420
1423
moveBlock.Move (block);
1421
1424
SentPackedMessage.emplace (std::move (moveBlock));
1422
1425
PackedMessagesToSend.pop ();
1426
+
1427
+ currentSize += writeRequest->ByteSizeLong ();
1423
1428
}
1424
1429
UpdateTokenIfNeededImpl ();
1425
1430
LOG_LAZY (DbDriverState->Log ,
You can’t perform that action at this time.
0 commit comments