1 package de.juplo.kafka.chat.backend.persistence.inmemory;
3 import com.fasterxml.jackson.databind.ObjectMapper;
4 import de.juplo.kafka.chat.backend.domain.ChatHomeWithShardsTest;
5 import de.juplo.kafka.chat.backend.persistence.StorageStrategy;
6 import de.juplo.kafka.chat.backend.persistence.storage.files.FilesStorageStrategy;
7 import org.springframework.boot.test.context.TestConfiguration;
8 import org.springframework.context.annotation.Bean;
10 import java.nio.file.Paths;
11 import java.time.Clock;
12 import java.util.stream.IntStream;
14 public class ShardedChatHomeTest extends ChatHomeWithShardsTest
17 static class Configuration
20 ShardedChatHome chatHome(
21 StorageStrategy storageStrategy,
24 SimpleChatHome[] chatHomes = new SimpleChatHome[NUM_SHARDS];
28 .forEach(shard -> chatHomes[shard] = new SimpleChatHome(
30 storageStrategy.read(),
34 ShardingStrategy strategy = new KafkaLikeShardingStrategy(NUM_SHARDS);
36 return new ShardedChatHome(chatHomes, strategy);
40 public FilesStorageStrategy storageStrategy(Clock clock)
42 return new FilesStorageStrategy(
43 Paths.get("target", "test-classes", "data", "files"),
46 new KafkaLikeShardingStrategy(NUM_SHARDS),
47 messageFlux -> new InMemoryChatRoomService(messageFlux),
54 return Clock.systemDefaultZone();
59 return new int[] { OWNED_SHARD };