pl.allegro.tech.hermes.management.domain.topic.TopicOwnerCache Maven / Gradle / Ivy

Go to download
package pl.allegro.tech.hermes.management.domain.topic;

import com.google.common.collect.ArrayListMultimap;
import com.google.common.collect.Multimap;
import com.google.common.collect.Multimaps;
import com.google.common.util.concurrent.ThreadFactoryBuilder;
import jakarta.annotation.PreDestroy;
import java.util.Collection;
import java.util.concurrent.Executors;
import java.util.concurrent.ScheduledExecutorService;
import java.util.concurrent.TimeUnit;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Component;
import pl.allegro.tech.hermes.api.OwnerId;
import pl.allegro.tech.hermes.api.Topic;
import pl.allegro.tech.hermes.api.TopicName;
import pl.allegro.tech.hermes.domain.topic.TopicRepository;
import pl.allegro.tech.hermes.management.domain.group.GroupService;

@Component
public class TopicOwnerCache {

  private static final Logger logger = LoggerFactory.getLogger(TopicOwnerCache.class);

  private final TopicRepository topicRepository;
  private final GroupService groupService;
  private final ScheduledExecutorService scheduledExecutorService;

  private Multimap cache =
      Multimaps.synchronizedMultimap(ArrayListMultimap.create());

  public TopicOwnerCache(
      TopicRepository topicRepository,
      GroupService groupService,
      @Value("${topicOwnerCache.refreshRateInSeconds}") int refreshRateInSeconds) {
    this.topicRepository = topicRepository;
    this.groupService = groupService;
    scheduledExecutorService =
        Executors.newSingleThreadScheduledExecutor(
            new ThreadFactoryBuilder().setNameFormat("topic-owner-cache-%d").build());
    scheduledExecutorService.scheduleAtFixedRate(
        this::refillCache, 0, refreshRateInSeconds, TimeUnit.SECONDS);
  }

  @PreDestroy
  public void stop() {
    scheduledExecutorService.shutdown();
  }

  public Collection get(OwnerId ownerId) {
    return cache.get(ownerId);
  }

  private void refillCache() {
    try {
      logger.info("Starting filling TopicOwnerCache cache");
      long start = System.currentTimeMillis();
      Multimap cache = ArrayListMultimap.create();
      groupService.listGroupNames().stream()
          .flatMap(groupName -> topicRepository.listTopics(groupName).stream())
          .forEach(topic -> cache.put(topic.getOwner(), topic.getName()));
      this.cache = Multimaps.synchronizedMultimap(cache);
      long end = System.currentTimeMillis();
      logger.info("TopicOwnerCache filled. Took {}ms", end - start);
    } catch (Exception e) {
      logger.error("Error while filling TopicOwnerCache", e);
    }
  }

  void onRemovedTopic(Topic topic) {
    cache.remove(topic.getOwner(), topic.getName());
  }

  void onCreatedTopic(Topic topic) {
    cache.put(topic.getOwner(), topic.getName());
  }

  void onUpdatedTopic(Topic oldTopic, Topic newTopic) {
    cache.remove(oldTopic.getOwner(), oldTopic.getName());
    cache.put(newTopic.getOwner(), newTopic.getName());
  }
}