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

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

import com.codahale.metrics.MetricRegistry;
import com.google.common.base.Charsets;
import com.google.common.io.BaseEncoding;
import com.google.inject.Inject;
import com.hubspot.singularity.SingularityUserSettings;
import com.hubspot.singularity.config.SingularityConfiguration;
import com.hubspot.singularity.data.transcoders.Transcoder;
import java.util.Optional;
import java.util.Set;
import org.apache.curator.framework.CuratorFramework;
import org.apache.curator.utils.ZKPaths;

public class UserManager extends CuratorManager {
  private final Transcoder settingsTranscoder;
  private final SingularityValidator validator;

  private static final String USER_ROOT = "/users";
  private static final String SETTINGS_ROOT = USER_ROOT + "/settings";

  @Inject
  public UserManager(
    CuratorFramework curator,
    SingularityConfiguration configuration,
    MetricRegistry metricRegistry,
    Transcoder settingsTranscoder,
    SingularityValidator validator
  ) {
    super(curator, configuration, metricRegistry);
    this.settingsTranscoder = settingsTranscoder;
    this.validator = validator;
  }

  private String encodeUserId(String userId) {
    validator.checkUserId(userId);
    return BaseEncoding.base64Url().encode(userId.getBytes(Charsets.UTF_8));
  }

  private String getUserSettingsPath(String userId) {
    return ZKPaths.makePath(SETTINGS_ROOT, encodeUserId(userId));
  }

  public void updateUserSettings(String userId, SingularityUserSettings userSettings) {
    save(getUserSettingsPath(userId), userSettings, settingsTranscoder);
  }

  public void addStarredRequestIds(String userId, Set starredRequestIds) {
    final String path = getUserSettingsPath(userId);
    final Optional settings = getData(path, settingsTranscoder);
    if (!settings.isPresent()) {
      validator.checkStarredRequests(starredRequestIds);
      save(path, new SingularityUserSettings(starredRequestIds), settingsTranscoder);
      return;
    }
    settings.get().addStarredRequestIds(starredRequestIds);
    validator.checkStarredRequests(settings.get().getStarredRequestIds());
    save(path, settings.get(), settingsTranscoder);
  }

  public void deleteStarredRequestIds(String userId, Set starredRequestIds) {
    final String path = getUserSettingsPath(userId);
    final Optional settings = getData(path, settingsTranscoder);
    if (!settings.isPresent()) {
      return;
    }
    save(
      path,
      settings.get().deleteStarredRequestIds(starredRequestIds),
      settingsTranscoder
    );
  }

  public Optional getUserSettings(String userId) {
    return getData(getUserSettingsPath(userId), settingsTranscoder);
  }

  public void deleteUserSettings(String userId) {
    delete(getUserSettingsPath(userId));
  }
}