public class ChatRoomChannel implements Runnable
{
private final String topic;
- private final Consumer<String, ChatRoomTo> consumer;
- private final Producer<String, ChatRoomTo> producer;
- private final ChatRoomFactory chatRoomFactory;
+ private final Consumer<Integer, ChatRoomTo> consumer;
+ private final Producer<Integer, ChatRoomTo> producer;
+ private final ShardingStrategy shardingStrategy;
+ private final ChatMessageChannel chatMessageChannel;
private boolean running;
{
try
{
- ConsumerRecords<String, ChatRoomTo> records = consumer.poll(Duration.ofMinutes(5));
+ ConsumerRecords<Integer, ChatRoomTo> records = consumer.poll(Duration.ofMinutes(5));
log.info("Fetched {} messages", records.count());
- for (ConsumerRecord<String, ChatRoomTo> record : records)
+ for (ConsumerRecord<Integer, ChatRoomTo> record : records)
{
UUID id = record.value().getId();
String name = record.value().getName();
}
}
+ void createChatRoom()
+ {
+ log.info("Creating ChatRoom with buffer-size {}", bufferSize);
+ KafkaChatRoomService service = new KafkaChatRoomService(chatMessageChannel, id);
+ int shard = shardingStrategy.selectShard(id);
+ ChatRoom chatRoom = new ChatRoom(id, name, shard, clock, service, bufferSize);
+ chatMessageChannel.putChatRoom(chatRoom);
+ }
+
Mono<ChatRoomInfo> sendCreateChatRoomRequest(
UUID chatRoomId,
String name)
{
- ChatRoomTo chatRoomTo = ChatRoomTo.of(chatRoomId, name);
+ int shard = this.shardingStrategy.selectShard(chatRoomId);
+ ChatRoomTo chatRoomTo = ChatRoomTo.of(chatRoomId, name, shard);
return Mono.create(sink ->
{
- ProducerRecord<String, ChatRoomTo> record =
+ ProducerRecord<Integer, ChatRoomTo> record =
new ProducerRecord<>(
topic,
- chatRoomId.toString(),
+ shard,
chatRoomTo);
producer.send(record, ((metadata, exception) ->