com.hubspot.singularity.data.NotificationsManager Maven / Gradle / Ivy

Go to download
package com.hubspot.singularity.data;

import java.util.List;
import java.util.concurrent.TimeUnit;

import org.apache.curator.framework.CuratorFramework;
import org.apache.curator.utils.ZKPaths;

import com.codahale.metrics.MetricRegistry;
import com.google.common.cache.CacheBuilder;
import com.google.common.cache.CacheLoader;
import com.google.common.cache.LoadingCache;
import com.google.inject.Inject;
import com.hubspot.singularity.config.SingularityConfiguration;

public class NotificationsManager extends CuratorManager {
  private static final String NOTIFICATIONS_ROOT = "/notifications";
  private static final String BLACKLIST_ROOT = NOTIFICATIONS_ROOT + "/blacklist";

  LoadingCache> cache;

  @Inject
  public NotificationsManager(CuratorFramework curator, SingularityConfiguration configuration, MetricRegistry metricRegistry) {
    super(curator, configuration, metricRegistry);

    cache = CacheBuilder.newBuilder()
        .expireAfterWrite(5, TimeUnit.MINUTES)
        .build(
            new CacheLoader>() {
              @Override public List load(String key) throws Exception {
                return getChildren(key);
              }
            }
        );
  }

  public void addToBlacklist(String email) {
    create(getEmailPath(email));
    cache.invalidate(BLACKLIST_ROOT);
  }

  public void removeFromBlacklist(String email) {
    delete(getEmailPath(email));
    cache.invalidate(BLACKLIST_ROOT);
  }

  public List getBlacklist() {
    return cache.getUnchecked(BLACKLIST_ROOT);
  }

  private String getEmailPath(String email) {
    return ZKPaths.makePath(BLACKLIST_ROOT, email);
  }
}