juplo.de Git - demos/kafka/chat/blob - src/test/java/de/juplo/kafka/chat/backend/implementation/inmemory/ShardedChatHomeServiceTest.java

   1 package de.juplo.kafka.chat.backend.implementation.inmemory;
   2
   3 import com.fasterxml.jackson.databind.ObjectMapper;
   4 import de.juplo.kafka.chat.backend.domain.ChatHomeServiceWithShardsTest;
   5 import de.juplo.kafka.chat.backend.implementation.ShardingStrategy;
   6 import de.juplo.kafka.chat.backend.implementation.StorageStrategy;
   7 import de.juplo.kafka.chat.backend.storage.files.FilesStorageStrategy;
   8 import org.springframework.boot.test.context.TestConfiguration;
   9 import org.springframework.context.annotation.Bean;
  10
  11 import java.net.URI;
  12 import java.nio.file.Paths;
  13 import java.time.Clock;
  14 import java.util.stream.IntStream;
  15
  16 public class ShardedChatHomeServiceTest extends ChatHomeServiceWithShardsTest
  17 {
  18   @TestConfiguration
  19   static class Configuration
  20   {
  21     @Bean
  22     ShardedChatHomeService chatHome(
  23         StorageStrategy storageStrategy,
  24         Clock clock)
  25     {
  26       SimpleChatHomeService[] chatHomes = new SimpleChatHomeService[NUM_SHARDS];
  27
  28       IntStream
  29           .of(ownedShards())
  30           .forEach(shard -> chatHomes[shard] = new SimpleChatHomeService(
  31               shard,
  32               storageStrategy,
  33               clock,
  34               bufferSize()));
  35
  36       ShardingStrategy strategy = new KafkaLikeShardingStrategy(NUM_SHARDS);
  37
  38       return new ShardedChatHomeService(
  39           chatHomes,
  40           IntStream
  41               .range(0, NUM_SHARDS)
  42               .mapToObj(shard -> "http://instance-0")
  43               .map(uriString -> URI.create(uriString))
  44               .toArray(size -> new URI[size]),
  45           strategy);
  46     }
  47
  48     @Bean
  49     public FilesStorageStrategy storageStrategy(Clock clock)
  50     {
  51       return new FilesStorageStrategy(
  52           Paths.get("target", "test-classes", "data", "files"),
  53           new KafkaLikeShardingStrategy(NUM_SHARDS),
  54           new ObjectMapper());
  55     }
  56
  57     @Bean
  58     Clock clock()
  59     {
  60       return Clock.systemDefaultZone();
  61     }
  62
  63     int[] ownedShards()
  64     {
  65       return new int[] { OWNED_SHARD };
  66     }
  67
  68     int bufferSize()
  69     {
  70       return 8;
  71     }
  72   }
  73 }