com.hubspot.singularity.config.SMTPConfiguration Maven / Gradle / Ivy

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

import com.fasterxml.jackson.annotation.JsonProperty;
import com.google.common.collect.ImmutableList;
import com.google.common.collect.ImmutableMap;
import com.google.common.collect.Maps;
import com.hubspot.singularity.SingularityEmailDestination;
import com.hubspot.singularity.SingularityEmailType;
import java.util.Arrays;
import java.util.Collections;
import java.util.List;
import java.util.Map;
import java.util.Optional;
import java.util.TimeZone;
import java.util.concurrent.TimeUnit;
import javax.validation.constraints.NotNull;

public class SMTPConfiguration {
  @JsonProperty
  private String username;

  @JsonProperty
  private String password;

  @JsonProperty
  private int taskLogLength = 512;

  @NotNull
  @JsonProperty
  private String host = "localhost";

  @JsonProperty
  private Integer port = 25; // SMTP

  @JsonProperty
  @NotNull
  private String from = "[email protected]";

  @JsonProperty
  private String mailerDatePattern = "MMM dd h:mm:ss a zzz";

  @JsonProperty
  private int mailMaxThreads = 3;

  @NotNull
  @JsonProperty
  private boolean ssl = false;

  @NotNull
  @JsonProperty
  private boolean startTLS = false;

  @NotNull
  @JsonProperty
  private List admins = Collections.emptyList();

  @JsonProperty
  private int rateLimitAfterNotifications = 5;

  @JsonProperty
  private long rateLimitPeriodMillis = TimeUnit.MINUTES.toMillis(10);

  @JsonProperty
  private long rateLimitCooldownMillis = TimeUnit.HOURS.toMillis(1);

  // Files to tail when sending a task email.
  @JsonProperty
  private List taskEmailTailFiles = Arrays.asList("stdout", "stderr");

  @JsonProperty
  private Optional taskLogErrorRegex = Optional.empty();

  @JsonProperty
  private Optional taskLogErrorRegexCaseSensitive = Optional.empty();

  @NotNull
  @JsonProperty
  private Long maxTaskLogSearchOffset = 100000L;

  @JsonProperty
  private TimeZone mailerTimeZone = TimeZone.getTimeZone("UTC");

  @JsonProperty
  private Optional subjectPrefix = Optional.empty();

  @JsonProperty
  private Optional uiBaseUrl = Optional.empty();

  @JsonProperty("emails")
  private Map> emailConfiguration = Maps.newHashMap(
    ImmutableMap
      .>builder()
      .put(
        SingularityEmailType.REQUEST_IN_COOLDOWN,
        ImmutableList.of(
          SingularityEmailDestination.ADMINS,
          SingularityEmailDestination.OWNERS
        )
      )
      .put(
        SingularityEmailType.REPLACEMENT_TASKS_FAILING,
        ImmutableList.of(SingularityEmailDestination.OWNERS)
      )
      .put(
        SingularityEmailType.SINGULARITY_ABORTING,
        ImmutableList.of(SingularityEmailDestination.ADMINS)
      )
      .put(
        SingularityEmailType.TASK_FAILED,
        ImmutableList.of(
          SingularityEmailDestination.ADMINS,
          SingularityEmailDestination.OWNERS,
          SingularityEmailDestination.ACTION_TAKER
        )
      )
      .put(
        SingularityEmailType.TASK_LOST,
        ImmutableList.of(
          SingularityEmailDestination.ADMINS,
          SingularityEmailDestination.ACTION_TAKER
        )
      )
      .put(
        SingularityEmailType.TASK_FINISHED_LONG_RUNNING,
        ImmutableList.of(
          SingularityEmailDestination.OWNERS,
          SingularityEmailDestination.ADMINS
        )
      )
      .put(
        SingularityEmailType.TASK_FINISHED_ON_DEMAND,
        ImmutableList.of(
          SingularityEmailDestination.OWNERS,
          SingularityEmailDestination.ACTION_TAKER
        )
      )
      .put(
        SingularityEmailType.TASK_SCHEDULED_OVERDUE_TO_FINISH,
        ImmutableList.of(
          SingularityEmailDestination.OWNERS,
          SingularityEmailDestination.ADMINS
        )
      )
      .put(
        SingularityEmailType.TASK_KILLED_UNHEALTHY,
        ImmutableList.of(
          SingularityEmailDestination.OWNERS,
          SingularityEmailDestination.ADMINS
        )
      )
      .put(
        SingularityEmailType.REQUEST_PAUSED,
        ImmutableList.of(
          SingularityEmailDestination.OWNERS,
          SingularityEmailDestination.ADMINS
        )
      )
      .put(
        SingularityEmailType.REQUEST_REMOVED,
        ImmutableList.of(
          SingularityEmailDestination.OWNERS,
          SingularityEmailDestination.ADMINS
        )
      )
      .put(
        SingularityEmailType.REQUEST_UNPAUSED,
        ImmutableList.of(
          SingularityEmailDestination.OWNERS,
          SingularityEmailDestination.ADMINS
        )
      )
      .put(
        SingularityEmailType.REQUEST_SCALED,
        ImmutableList.of(SingularityEmailDestination.OWNERS)
      )
      .put(
        SingularityEmailType.DISASTER_DETECTED,
        ImmutableList.of(SingularityEmailDestination.ADMINS)
      )
      .build()
  );

  public Map> getEmailConfiguration() {
    return emailConfiguration;
  }

  public void setEmailConfiguration(
    Map> emailConfiguration
  ) {
    this.emailConfiguration.putAll(emailConfiguration);
  }

  public int getTaskLogLength() {
    return taskLogLength;
  }

  public void setTaskLogLength(int length) {
    taskLogLength = length;
  }

  public Optional getUsername() {
    return Optional.ofNullable(username);
  }

  public void setUsername(String username) {
    this.username = username;
  }

  public int getMailMaxThreads() {
    return mailMaxThreads;
  }

  public void setMailMaxThreads(int mailMaxThreads) {
    this.mailMaxThreads = mailMaxThreads;
  }

  public Optional getPassword() {
    return Optional.ofNullable(password);
  }

  public void setPassword(String password) {
    this.password = password;
  }

  public String getHost() {
    return host;
  }

  public void setHost(String host) {
    this.host = host;
  }

  public int getPort() {
    return port;
  }

  public void setPort(int port) {
    this.port = port;
  }

  public String getFrom() {
    return from;
  }

  public void setFrom(String from) {
    this.from = from;
  }

  public boolean isSsl() {
    return ssl;
  }

  public void setSsl(boolean ssl) {
    this.ssl = ssl;
  }

  public boolean isStartTLS() {
    return startTLS;
  }

  public void setStartTLS(boolean startTLS) {
    this.startTLS = startTLS;
  }

  public List getAdmins() {
    return admins;
  }

  public void setAdmins(List admins) {
    this.admins = admins;
  }

  public int getRateLimitAfterNotifications() {
    return rateLimitAfterNotifications;
  }

  public void setRateLimitAfterNotifications(int rateLimitAfterNotifications) {
    this.rateLimitAfterNotifications = rateLimitAfterNotifications;
  }

  public long getRateLimitPeriodMillis() {
    return rateLimitPeriodMillis;
  }

  public void setRateLimitPeriodMillis(long rateLimitPeriodMillis) {
    this.rateLimitPeriodMillis = rateLimitPeriodMillis;
  }

  public long getRateLimitCooldownMillis() {
    return rateLimitCooldownMillis;
  }

  public void setRateLimitCooldownMillis(long rateLimitCooldownMillis) {
    this.rateLimitCooldownMillis = rateLimitCooldownMillis;
  }

  public List getTaskEmailTailFiles() {
    return taskEmailTailFiles;
  }

  public void setTaskEmailTailFiles(List taskEmailTailFiles) {
    this.taskEmailTailFiles = taskEmailTailFiles;
  }

  public String getMailerDatePattern() {
    return mailerDatePattern;
  }

  public void setMailerDatePattern(String mailerDatePattern) {
    this.mailerDatePattern = mailerDatePattern;
  }

  public TimeZone getMailerTimeZone() {
    return mailerTimeZone;
  }

  public void setMailerTimeZone(TimeZone mailerTimeZone) {
    this.mailerTimeZone = mailerTimeZone;
  }

  public Optional getTaskLogErrorRegex() {
    return taskLogErrorRegex;
  }

  public void setTaskLogErrorRegex(Optional taskLogErrorRegex) {
    this.taskLogErrorRegex = taskLogErrorRegex;
  }

  public Optional getTaskLogErrorRegexCaseSensitive() {
    return taskLogErrorRegexCaseSensitive;
  }

  public void setTaskLogErrorRegexCaseSensitive(
    Optional taskLogErrorRegexCaseSensitive
  ) {
    this.taskLogErrorRegexCaseSensitive = taskLogErrorRegexCaseSensitive;
  }

  public Long getMaxTaskLogSearchOffset() {
    return maxTaskLogSearchOffset;
  }

  public void setMaxTaskLogSearchOffset(Long maxTaskLogSearchOffset) {
    this.maxTaskLogSearchOffset = maxTaskLogSearchOffset;
  }

  public Optional getSubjectPrefix() {
    return subjectPrefix;
  }

  public void setSubjectPrefix(Optional subjectPrefix) {
    this.subjectPrefix = subjectPrefix;
  }

  public Optional getUiBaseUrl() {
    return uiBaseUrl;
  }

  public void setUiBaseUrl(Optional uiBaseUrl) {
    this.uiBaseUrl = uiBaseUrl;
  }
}