private final UUID id;
@Getter
private final String name;
- private final PersistenceStrategy persistence;
- private final Sinks.Many<Message> sink;
+ private final ChatroomService chatroomService;
+ private final int bufferSize;
+ private Sinks.Many<Message> sink;
public Chatroom(
UUID id,
String name,
- PersistenceStrategy persistence,
+ ChatroomService chatroomService,
int bufferSize)
{
this.id = id;
this.name = name;
- this.persistence = persistence;
- this.sink = Sinks.many().multicast().onBackpressureBuffer(bufferSize);
+ this.chatroomService = chatroomService;
+ this.bufferSize = bufferSize;
+ this.sink = createSink();
}
String user,
String text)
{
- return persistence
+ return chatroomService
.persistMessage(Message.MessageKey.of(user, id), timestamp, text)
- .doOnNext(message -> sink.tryEmitNext(message).orThrow());
+ .doOnNext(message ->
+ {
+ Sinks.EmitResult result = sink.tryEmitNext(message);
+ if (result.isFailure())
+ {
+ log.warn("Emitting of message failed with {} for {}", result.name(), message);
+ }
+ });
}
public Mono<Message> getMessage(String username, Long messageId)
{
- return persistence.getMessage(Message.MessageKey.of(username, messageId));
+ Message.MessageKey key = Message.MessageKey.of(username, messageId);
+ return chatroomService.getMessage(key);
}
- public Flux<Message> listen()
+ synchronized public Flux<Message> listen()
{
- return sink.asFlux();
+ return sink
+ .asFlux()
+ .doOnCancel(() -> sink = createSink()); // Sink hast to be recreated on auto-cancel!
}
public Flux<Message> getMessages()
public Flux<Message> getMessages(long first, long last)
{
- return persistence.getMessages(first, last);
+ return chatroomService.getMessages(first, last);
+ }
+
+ private Sinks.Many<Message> createSink()
+ {
+ return Sinks
+ .many()
+ .multicast()
+ .onBackpressureBuffer(bufferSize);
}
}