import java.util.Arrays;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
-import java.util.stream.IntStream;
@Testcontainers
private final AbstractHandoverITContainers containers;
- private final ExecutorService executorService = Executors.newFixedThreadPool(NUM_CLIENTS);
+ private final ExecutorService executorService = Executors.newFixedThreadPool(NUM_CLIENTS + 1);
AbstractHandoverIT(AbstractHandoverITContainers containers)
.toStream()
.toArray(size -> new ChatRoomInfoTo[size]);
+ int port = containers.haproxy.getMappedPort(8400);
+
TestWriter[] testWriters = Flux
- .fromStream(IntStream.range(0, NUM_CLIENTS).mapToObj(i -> "user-" + Integer.toString(i)))
+ .range(0, NUM_CLIENTS)
.map(i -> new TestWriter(
- containers.haproxy.getMappedPort(8400),
- chatRooms,
- i))
+ port,
+ chatRooms[i % NUM_CHATROOMS],
+ "user-" + Integer.toString(i)))
.doOnNext(testClient -> executorService.execute(testClient))
.toStream()
.toArray(size -> new TestWriter[size]);
+ TestListener testListener = new TestListener(port, chatRooms);
+ executorService.execute(testListener);
+
Thread.sleep(10000);
+
Arrays
.stream(testWriters)
.forEach(testClient -> testClient.running = false);
-
- Flux
- .fromArray(chatRooms)
- .flatMap(chatRoom ->receiveMessages(chatRoom).take(100))
- .doOnNext(message -> log.info("message: {}", message))
- .then()
- .block();
+ testListener.running = false;
}
Mono<ChatRoomInfoTo> createChatRoom(String name)
});
}
- Flux<ServerSentEvent<String>> receiveMessages(ChatRoomInfoTo chatRoom)
- {
- return webClient
- .get()
- .uri(
- "/{chatRoomId}/listen",
- chatRoom.getId())
- .accept(MediaType.TEXT_EVENT_STREAM)
- .retrieve()
- .bodyToFlux(SSE_TYPE);
- }
-
WebClient webClient;