pl.allegro.tech.hermes.frontend.publishing.preview.MessagePreviewLog Maven / Gradle / Ivy

Go to download
package pl.allegro.tech.hermes.frontend.publishing.preview;

import com.google.common.util.concurrent.AtomicLongMap;
import java.util.ArrayList;
import java.util.List;
import java.util.concurrent.ConcurrentLinkedDeque;
import pl.allegro.tech.hermes.api.Topic;
import pl.allegro.tech.hermes.api.TopicName;
import pl.allegro.tech.hermes.domain.topic.preview.MessagePreview;
import pl.allegro.tech.hermes.domain.topic.preview.TopicsMessagesPreview;
import pl.allegro.tech.hermes.frontend.publishing.message.Message;

public class MessagePreviewLog {

  private final MessagePreviewFactory messagePreviewFactory;

  private final int previewSizePerTopic;

  private final AtomicLongMap limiter = AtomicLongMap.create();

  private final ConcurrentLinkedDeque messages =
      new ConcurrentLinkedDeque<>();

  public MessagePreviewLog(MessagePreviewFactory messagePreviewFactory, int previewSizePerTopic) {
    this.messagePreviewFactory = messagePreviewFactory;
    this.previewSizePerTopic = previewSizePerTopic;
  }

  public void add(Topic topic, Message message) {
    long counter = limiter.getAndIncrement(topic.getName());
    if (counter < previewSizePerTopic) {
      messages.add(
          new MessagePreviewSnapshot(
              topic.getName(),
              messagePreviewFactory.create(message, topic.isSchemaIdAwareSerializationEnabled())));
    }
  }

  public TopicsMessagesPreview snapshotAndClean() {
    List snapshot = new ArrayList<>(messages);
    messages.clear();
    limiter.clear();

    TopicsMessagesPreview preview = new TopicsMessagesPreview();
    for (MessagePreviewSnapshot message : snapshot) {
      preview.add(message.topicName, message.content);
    }
    return preview;
  }

  private static class MessagePreviewSnapshot {

    final TopicName topicName;

    final MessagePreview content;

    MessagePreviewSnapshot(TopicName topicName, MessagePreview content) {
      this.topicName = topicName;
      this.content = content;
    }
  }
}