com.hubspot.singularity.data.history.DeployHistoryHelper Maven / Gradle / Ivy

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

import com.google.common.collect.Lists;
import com.google.inject.Inject;
import com.google.inject.Singleton;
import com.hubspot.singularity.SingularityDeploy;
import com.hubspot.singularity.SingularityDeployHistory;
import com.hubspot.singularity.SingularityDeployKey;
import com.hubspot.singularity.config.SingularityConfiguration;
import com.hubspot.singularity.data.DeployManager;
import java.util.Collections;
import java.util.List;
import java.util.Optional;

@Singleton
public class DeployHistoryHelper
  extends BlendedHistoryHelper {
  private final DeployManager deployManager;
  private final HistoryManager historyManager;

  @Inject
  public DeployHistoryHelper(
    DeployManager deployManager,
    HistoryManager historyManager,
    SingularityConfiguration configuration
  ) {
    super(configuration.getDatabaseConfiguration().isPresent());
    this.deployManager = deployManager;
    this.historyManager = historyManager;
  }

  @Override
  protected List getFromZk(String requestId) {
    final List deployKeys = deployManager.getDeployIdsFor(
      requestId
    );
    final List histories = Lists.newArrayListWithCapacity(
      deployKeys.size()
    );

    for (SingularityDeployKey key : deployKeys) {
      Optional deployHistory = deployManager.getDeployHistory(
        key.getRequestId(),
        key.getDeployId(),
        false
      );
      if (deployHistory.isPresent()) {
        histories.add(deployHistory.get());
      }
    }

    Collections.sort(histories);

    return histories;
  }

  @Override
  protected List getFromHistory(
    String requestId,
    int historyStart,
    int numFromHistory
  ) {
    return historyManager.getDeployHistoryForRequest(
      requestId,
      historyStart,
      numFromHistory
    );
  }

  public boolean isDeployIdAvailable(String requestId, String deployId) {
    Optional deploy = deployManager.getDeploy(requestId, deployId);

    if (deploy.isPresent()) {
      return false;
    }

    return !historyManager.getDeployHistory(requestId, deployId).isPresent();
  }

  @Override
  protected Optional getTotalCount(String requestId, boolean canSkipZk) {
    final int numFromZk;
    if (sqlEnabled && canSkipZk) {
      numFromZk = 0;
    } else {
      numFromZk = deployManager.getDeployIdsFor(requestId).size();
    }
    final int numFromHistory = historyManager.getDeployHistoryForRequestCount(requestId);

    return Optional.of(numFromZk + numFromHistory);
  }
}