io.envoyproxy.controlplane.cache.CacheStatusInfoAggregator Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of cache Show documentation
The newest version!
package io.envoyproxy.controlplane.cache;

import java.util.Collection;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.ConcurrentMap;
import java.util.stream.Collectors;
import java.util.stream.Stream;

public class CacheStatusInfoAggregator {
  private final ConcurrentMap>> statuses =
      new ConcurrentHashMap<>();
  private final ConcurrentMap>> deltaStatuses =
      new ConcurrentHashMap<>();

  public Collection groups() {
    return Stream.concat(statuses.keySet().stream(), deltaStatuses.keySet().stream()).collect(Collectors.toSet());
  }

  public void remove(T group) {
    statuses.remove(group);
    deltaStatuses.remove(group);
  }

  /**
   * Returns map of delta status infos for group identifier.
   *
   * @param group group identifier.
   */
  public Map> getDeltaStatus(T group) {
    return deltaStatuses.getOrDefault(group, new ConcurrentHashMap<>());
  }

  /**
   * Returns map of status infos for group identifier.
   *
   * @param group group identifier.
   */
  public Map> getStatus(T group) {
    return statuses.getOrDefault(group, new ConcurrentHashMap<>());
  }

  /**
   * Check if statuses for specific group have any watcher.
   *
   * @param group group identifier.
   * @return true if statuses for specific group have any watcher.
   */
  public boolean hasStatuses(T group) {
    Map> status = getStatus(group);
    Map> deltaStatus = getDeltaStatus(group);
    return status.values().stream().mapToLong(CacheStatusInfo::numWatches).sum()
        + deltaStatus.values().stream().mapToLong(DeltaCacheStatusInfo::numWatches).sum() > 0;
  }

  /**
   * Returns delta status info for group identifier and creates new one if it doesn't exist.
   *
   * @param group        group identifier.
   * @param resourceType resource type.
   */
  public DeltaCacheStatusInfo getOrAddDeltaStatusInfo(T group, Resources.ResourceType resourceType) {
    return deltaStatuses.computeIfAbsent(group, g -> new ConcurrentHashMap<>())
        .computeIfAbsent(resourceType, s -> new DeltaCacheStatusInfo<>(group));
  }

  /**
   * Returns status info for group identifier and creates new one if it doesn't exist.
   *
   * @param group        group identifier.
   * @param resourceType resource type.
   */
  public CacheStatusInfo getOrAddStatusInfo(T group, Resources.ResourceType resourceType) {
    return statuses.computeIfAbsent(group, g -> new ConcurrentHashMap<>())
        .computeIfAbsent(resourceType, s -> new CacheStatusInfo<>(group));
  }
}