net.jodah.sarge.internal.SupervisionRegistry Maven / Gradle / Ivy

Go to download
package net.jodah.sarge.internal;

import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;

import net.jodah.sarge.Plan;
import net.jodah.sarge.Supervisor;

/**
 * Statically stores and retrieves SupervisionContext instances.
 * 
 * @author Jonathan Halterman
 */
public class SupervisionRegistry {
  private final Map supervisors = new ConcurrentHashMap();
  private final Map plans = new ConcurrentHashMap();

  /** Returns the Plan for the {@code supervised}. */
  public Plan planFor(Object supervised) {
    return plans.get(supervised);
  }

  public void supervise(Object supervisable, Plan strategy) {
    plans.put(supervisable, strategy);
  }

  /**
   * @throws IllegalArgumentException if {@code supervised} is already supervised
   */
  public void supervise(Object supervisable, Supervisor supervisor) {
    Supervisor existingSupervisor = supervisors.get(supervisable);
    if (existingSupervisor != null)
      throw new IllegalArgumentException(supervisable + " is already supervised");
    supervisors.put(supervisable, supervisor);
  }

  /** Returns the supervisor for the {@code supervised}. */
  public Supervisor supervisorFor(Object supervised) {
    return supervisors.get(supervised);
  }

  /**
   * @throws IllegalArgumentException if {@code supervised} is not supervised
   */
  public void unsupervise(Object supervised) {
    if (plans.remove(supervised) == null && supervisors.remove(supervised) == null)
      throw new IllegalArgumentException(supervised + " is not supervised");
  }
}