io.camunda.exporter.cache.ExporterEntityCacheImpl Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of camunda-exporter Show documentation

Camunda Exporter

There is a newer version: 8.7.0-alpha2

/*
 * Copyright Camunda Services GmbH and/or licensed to Camunda Services GmbH under
 * one or more contributor license agreements. See the NOTICE file distributed
 * with this work for additional information regarding copyright ownership.
 * Licensed under the Camunda License 1.0. You may not use this file
 * except in compliance with the Camunda License 1.0.
 */
package io.camunda.exporter.cache;

import com.github.benmanes.caffeine.cache.CacheLoader;
import com.github.benmanes.caffeine.cache.Caffeine;
import com.github.benmanes.caffeine.cache.LoadingCache;
import java.util.Optional;

public class ExporterEntityCacheImpl implements ExporterEntityCache {

  private final LoadingCache cache;

  public ExporterEntityCacheImpl(
      final long maxSize,
      final CacheLoader cacheLoader,
      final ExporterCacheMetrics exporterCacheMetrics) {
    cache =
        Caffeine.newBuilder()
            .maximumSize(maxSize)
            .recordStats(() -> exporterCacheMetrics)
            .build(
                k -> {
                  try {
                    return cacheLoader.load(k);
                  } catch (final Exception e) {
                    throw new CacheLoaderFailedException(e);
                  }
                });
  }

  @Override
  public Optional get(final K entityKey) {
    return Optional.ofNullable(cache.get(entityKey));
  }

  @Override
  public void put(final K entityKey, final T entity) {
    cache.put(entityKey, entity);
  }

  @Override
  public void remove(final K entityKey) {
    cache.invalidate(entityKey);
  }

  @Override
  public void clear() {
    cache.invalidateAll();
  }
}