1 package de.juplo.kafka.chat.backend.persistence;
3 import com.fasterxml.jackson.databind.ObjectMapper;
4 import com.fasterxml.jackson.databind.SerializationFeature;
5 import com.fasterxml.jackson.datatype.jsr310.JavaTimeModule;
6 import de.juplo.kafka.chat.backend.domain.ShardingStrategy;
7 import de.juplo.kafka.chat.backend.domain.ChatHomeService;
8 import de.juplo.kafka.chat.backend.domain.ChatRoomFactory;
9 import de.juplo.kafka.chat.backend.persistence.inmemory.InMemoryChatRoomFactory;
10 import de.juplo.kafka.chat.backend.persistence.storage.files.FilesStorageStrategy;
11 import de.juplo.kafka.chat.backend.persistence.inmemory.InMemoryChatHomeService;
12 import de.juplo.kafka.chat.backend.persistence.inmemory.InMemoryChatRoomService;
13 import lombok.extern.slf4j.Slf4j;
14 import org.junit.jupiter.api.BeforeEach;
16 import java.io.IOException;
17 import java.nio.file.Files;
18 import java.nio.file.Path;
19 import java.nio.file.Paths;
20 import java.time.Clock;
21 import java.util.function.Supplier;
25 public class InMemoryWithFilesStorageIT extends AbstractStorageStrategyIT
27 final static Path path = Paths.get("target","files");
30 final ObjectMapper mapper;
31 final FilesStorageStrategy storageStrategy;
34 public InMemoryWithFilesStorageIT()
36 clock = Clock.systemDefaultZone();
37 mapper = new ObjectMapper();
38 mapper.registerModule(new JavaTimeModule());
39 mapper.disable(SerializationFeature.WRITE_DATES_AS_TIMESTAMPS);
40 storageStrategy = new FilesStorageStrategy(
45 messageFlux -> new InMemoryChatRoomService(messageFlux),
51 protected StorageStrategy getStorageStrategy()
53 return storageStrategy;
57 protected Supplier<ChatHomeService> getChatHomeServiceSupplier()
59 return () -> new InMemoryChatHomeService(
62 getStorageStrategy().read());
66 protected ChatRoomFactory getChatRoomFactory()
68 ShardingStrategy strategy = chatRoomId -> 0;
69 return new InMemoryChatRoomFactory(strategy, clock, 8);
73 void reset() throws Exception
75 if (Files.exists(path))
83 if (!file.equals(path))
85 log.debug("Deleting file {}", file);
91 throw new RuntimeException(e);
94 log.debug("Deleting data-directory {}", path);