import org.springframework.http.HttpStatus;
import org.springframework.http.MediaType;
import org.springframework.web.reactive.function.client.WebClient;
-import reactor.core.publisher.Flux;
+import org.springframework.web.reactive.function.client.WebClientResponseException;
import reactor.core.publisher.Mono;
import reactor.util.retry.Retry;
+import java.nio.charset.Charset;
import java.time.Duration;
+import java.util.concurrent.ThreadLocalRandom;
@Slf4j
-public class TestClient
+public class TestClient implements Runnable
{
+ @Override
public void run()
{
- Flux
- .range(1, 100)
- .flatMap(i -> Flux
- .fromArray(chatRooms)
- .map(chatRoom -> sendMessage(chatRoom, "Message #" + i))
- .flatMap(result -> result
- .map(MessageTo::toString)
- .retryWhen(Retry.fixedDelay(10, Duration.ofSeconds(1)))))
- .subscribe(result -> log.info("{}", result));
+ for (int i = 0; running; i++)
+ {
+ String message = "Message #" + i;
+ for (ChatRoomInfoTo chatRoom : chatRooms)
+ {
+ sendMessage(chatRoom, message)
+ .retryWhen(Retry.fixedDelay(10, Duration.ofSeconds(1)))
+ .map(MessageTo::toString)
+ .onErrorResume(throwable ->
+ {
+ WebClientResponseException e = (WebClientResponseException)throwable.getCause();
+ return Mono.just(e.getResponseBodyAsString(Charset.defaultCharset()));
+ })
+ .subscribe(result -> log.info(
+ "{} sent a message to {}: {}",
+ user,
+ chatRoom,
+ result));
+ }
+ try
+ {
+ Thread.sleep(ThreadLocalRandom.current().nextLong(700, 1000));
+ }
+ catch (Exception e)
+ {
+ throw new RuntimeException(e);
+ }
+ }
}
private Mono<MessageTo> sendMessage(
private final ChatRoomInfoTo[] chatRooms;
private final User user;
+ volatile boolean running = true;
+
TestClient(Integer port, ChatRoomInfoTo[] chatRooms, String username)
{