@@ -537,13 +537,15 @@ private class OrderedRunGenerator implements Iterable<Run> {
537
537
private final DBPort port ;
538
538
private final WriteConcern writeConcern ;
539
539
private final DBEncoder encoder ;
540
+ private final int maxBatchWriteSize ;
540
541
541
542
public OrderedRunGenerator (final List <WriteRequest > writeRequests , final WriteConcern writeConcern , final DBEncoder encoder ,
542
543
final DBPort port ) {
543
544
this .writeRequests = writeRequests ;
544
545
this .port = port ;
545
546
this .writeConcern = writeConcern .continueOnError (false );
546
547
this .encoder = encoder ;
548
+ this .maxBatchWriteSize = db .getConnector ().getServerDescription (port .getAddress ()).getMaxWriteBatchSize ();
547
549
}
548
550
549
551
@ Override
@@ -570,8 +572,7 @@ public Run next() {
570
572
private int getStartIndexOfNextRun () {
571
573
WriteRequest .Type type = writeRequests .get (curIndex ).getType ();
572
574
for (int i = curIndex ; i < writeRequests .size (); i ++) {
573
- if (i == curIndex + db .getConnector ().getServerDescription (port .getAddress ()).getMaxWriteBatchSize ()
574
- || writeRequests .get (i ).getType () != type ) {
575
+ if (i == curIndex + maxBatchWriteSize || writeRequests .get (i ).getType () != type ) {
575
576
return i ;
576
577
}
577
578
}
@@ -592,13 +593,15 @@ private class UnorderedRunGenerator implements Iterable<Run> {
592
593
private final DBPort port ;
593
594
private final WriteConcern writeConcern ;
594
595
private final DBEncoder encoder ;
596
+ private final int maxBatchWriteSize ;
595
597
596
598
public UnorderedRunGenerator (final List <WriteRequest > writeRequests , final WriteConcern writeConcern ,
597
599
final DBEncoder encoder , final DBPort port ) {
598
600
this .writeRequests = writeRequests ;
599
601
this .port = port ;
600
602
this .writeConcern = writeConcern .continueOnError (true );
601
603
this .encoder = encoder ;
604
+ this .maxBatchWriteSize = db .getConnector ().getServerDescription (port .getAddress ()).getMaxWriteBatchSize ();
602
605
}
603
606
604
607
@ Override
@@ -629,7 +632,7 @@ public Run next() {
629
632
}
630
633
run .add (writeRequest , curIndex );
631
634
curIndex ++;
632
- if (run .size () > db . getConnector (). getServerDescription ( port . getAddress ()). getMaxWriteBatchSize () ) {
635
+ if (run .size () > maxBatchWriteSize ) {
633
636
return runs .remove (run .type );
634
637
}
635
638
}
0 commit comments