File tree Expand file tree Collapse file tree 1 file changed +16
-3
lines changed
src/test/java/com/rabbitmq/stream/impl Expand file tree Collapse file tree 1 file changed +16
-3
lines changed Original file line number Diff line number Diff line change @@ -207,9 +207,22 @@ static void publishAndWaitForConfirms(
207
207
cf .get (new Client .ClientParameters ().publishConfirmListener (publishConfirmListener ));
208
208
209
209
client .declarePublisher (b (1 ), null , stream );
210
- for (int i = 1 ; i <= publishCount ; i ++) {
211
- Message message = messageFactory .apply (client .messageBuilder ());
212
- client .publish (b (1 ), Collections .singletonList (message ));
210
+ int batchSize = 100 ;
211
+ if (publishCount > batchSize ) {
212
+ List <Message > messages = new ArrayList <>(batchSize );
213
+ for (int i = 1 ; i <= publishCount ; i ++) {
214
+ Message message = messageFactory .apply (client .messageBuilder ());
215
+ messages .add (message );
216
+ if (i % batchSize == 0 || i == publishCount ) {
217
+ client .publish (b (1 ), messages );
218
+ messages .clear ();
219
+ }
220
+ }
221
+ } else {
222
+ for (int i = 1 ; i <= publishCount ; i ++) {
223
+ Message message = messageFactory .apply (client .messageBuilder ());
224
+ client .publish (b (1 ), Collections .singletonList (message ));
225
+ }
213
226
}
214
227
215
228
latchAssert (latchConfirm ).completes (timeout );
You can’t perform that action at this time.
0 commit comments