juplo.de Git - demos/kafka/chat/blob - src/test/java/de/juplo/kafka/chat/backend/persistence/inmemory/ShardedChatHomeTest.java

   1 package de.juplo.kafka.chat.backend.persistence.inmemory;
   2
   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;
   9
  10 import java.nio.file.Paths;
  11 import java.time.Clock;
  12 import java.util.stream.IntStream;
  13
  14 public class ShardedChatHomeTest extends ChatHomeWithShardsTest
  15 {
  16   @TestConfiguration
  17   static class Configuration
  18   {
  19     @Bean
  20     ShardedChatHome chatHome(
  21         StorageStrategy storageStrategy,
  22         Clock clock)
  23     {
  24       SimpleChatHome[] chatHomes = new SimpleChatHome[NUM_SHARDS];
  25
  26       IntStream
  27           .of(ownedShards())
  28           .forEach(shard -> chatHomes[shard] = new SimpleChatHome(
  29               shard,
  30               storageStrategy,
  31               clock,
  32               bufferSize()));
  33
  34       ShardingStrategy strategy = new KafkaLikeShardingStrategy(NUM_SHARDS);
  35
  36       return new ShardedChatHome(chatHomes, strategy);
  37     }
  38
  39     @Bean
  40     public FilesStorageStrategy storageStrategy(Clock clock)
  41     {
  42       return new FilesStorageStrategy(
  43           Paths.get("target", "test-classes", "data", "files"),
  44           new KafkaLikeShardingStrategy(NUM_SHARDS),
  45           new ObjectMapper());
  46     }
  47
  48     @Bean
  49     Clock clock()
  50     {
  51       return Clock.systemDefaultZone();
  52     }
  53
  54     int[] ownedShards()
  55     {
  56       return new int[] { OWNED_SHARD };
  57     }
  58
  59     int bufferSize()
  60     {
  61       return 8;
  62     }
  63   }
  64 }