X-Git-Url: http://juplo.de/gitweb/?a=blobdiff_plain;f=src%2Ftest%2Fjava%2Fde%2Fjuplo%2Fkafka%2Fchat%2Fbackend%2Fpersistence%2Finmemory%2FShardedChatHomeTest.java;h=d36bf2201b2772506b42713ba2b2bbb89ad95b2f;hb=3cb0ac8b376573c1229ee525d83611b8af7a7a2b;hp=ff82986a9403556d29aba0a7ac55a98ff2c31f51;hpb=bb1f9a4b980e35790cef4892063ab6186f2b7084;p=demos%2Fkafka%2Fchat diff --git a/src/test/java/de/juplo/kafka/chat/backend/persistence/inmemory/ShardedChatHomeTest.java b/src/test/java/de/juplo/kafka/chat/backend/persistence/inmemory/ShardedChatHomeTest.java index ff82986a..d36bf220 100644 --- a/src/test/java/de/juplo/kafka/chat/backend/persistence/inmemory/ShardedChatHomeTest.java +++ b/src/test/java/de/juplo/kafka/chat/backend/persistence/inmemory/ShardedChatHomeTest.java @@ -19,52 +19,46 @@ public class ShardedChatHomeTest extends ChatHomeWithShardsTestBase { @Bean ShardedChatHome chatHome( - Integer numShards, - int[] ownedShards, InMemoryChatHomeService chatHomeService) { - SimpleChatHome[] chatHomes = new SimpleChatHome[numShards]; + SimpleChatHome[] chatHomes = new SimpleChatHome[numShards()]; IntStream - .of(ownedShards) + .of(ownedShards()) .forEach(shard -> chatHomes[shard] = new SimpleChatHome(chatHomeService, shard)); - ShardingStrategy strategy = new KafkaLikeShardingStrategy(numShards); + ShardingStrategy strategy = new KafkaLikeShardingStrategy(numShards()); return new ShardedChatHome(chatHomes, strategy); } @Bean InMemoryChatHomeService chatHomeService( - Integer numShards, - int[] ownedShards, StorageStrategy storageStrategy) { return new InMemoryChatHomeService( - numShards, - ownedShards, + numShards(), + ownedShards(), storageStrategy.read()); } @Bean - public FilesStorageStrategy storageStrategy(Integer numShards) + public FilesStorageStrategy storageStrategy() { return new FilesStorageStrategy( Paths.get("target", "test-classes", "data", "files"), Clock.systemDefaultZone(), 8, - new KafkaLikeShardingStrategy(numShards), + new KafkaLikeShardingStrategy(numShards()), messageFlux -> new InMemoryChatRoomService(messageFlux), new ObjectMapper()); } - @Bean Integer numShards() { return 10; } - @Bean int[] ownedShards() { return new int[] { 2 };