import org.springframework.web.reactive.function.client.WebClientResponseException;
import reactor.core.publisher.Flux;
import reactor.core.publisher.Mono;
+import reactor.core.scheduler.Schedulers;
import reactor.util.retry.Retry;
import java.nio.charset.Charset;
import java.time.Duration;
import java.util.Iterator;
+import java.util.LinkedList;
+import java.util.List;
import java.util.concurrent.ThreadLocalRandom;
@Slf4j
-public class TestWriter implements Runnable
+public class TestWriter
{
- @Override
- public void run()
+ public Mono<Void> run()
{
- Flux
+ return Flux
.fromIterable((Iterable<Integer>) () -> new Iterator<>()
{
private int i = 0;
return i++;
}
})
+ .delayElements(Duration.ofMillis(ThreadLocalRandom.current().nextLong(500, 1500)))
.map(i -> "Message #" + i)
.flatMap(message -> sendMessage(chatRoom, message)
- .delayElement(Duration.ofMillis(ThreadLocalRandom.current().nextLong(500, 1500)))
- .retryWhen(Retry.fixedDelay(10, Duration.ofSeconds(1))))
- .doOnNext(message -> log.info(
- "{} sent a message to {}: {}",
- user,
- chatRoom,
- message))
+ .retryWhen(Retry.backoff(10, Duration.ofSeconds(1))))
+ .doOnNext(message ->
+ {
+ sentMessages.add(message);
+ log.info(
+ "{} sent a message to {}: {}",
+ user,
+ chatRoom,
+ message);
+ })
.doOnError(throwable ->
{
WebClientResponseException e = (WebClientResponseException)throwable.getCause();
user,
e.getResponseBodyAsString(Charset.defaultCharset()));
})
- .then()
- .block();
+ .limitRate(1)
+ .takeUntil(message -> !running)
+ .doOnComplete(() -> log.info("TestWriter {} is done", user))
+ .parallel(1)
+ .runOn(Schedulers.parallel())
+ .then();
}
private Mono<MessageTo> sendMessage(
private final WebClient webClient;
- private final ChatRoomInfoTo chatRoom;
- private final User user;
+
+ final ChatRoomInfoTo chatRoom;
+ final User user;
+ final List<MessageTo> sentMessages = new LinkedList<>();
volatile boolean running = true;