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

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

import com.google.inject.Inject;
import com.google.inject.Singleton;
import com.hubspot.singularity.ExtendedTaskState;
import com.hubspot.singularity.OrderDirection;
import com.hubspot.singularity.SingularityDeployKey;
import com.hubspot.singularity.SingularityTaskId;
import com.hubspot.singularity.SingularityTaskIdHistory;
import com.hubspot.singularity.config.SingularityConfiguration;
import com.hubspot.singularity.data.TaskManager;
import java.util.List;
import java.util.Optional;

@Singleton
public class DeployTaskHistoryHelper
  extends BlendedHistoryHelper {
  private final TaskManager taskManager;
  private final HistoryManager historyManager;

  @Inject
  public DeployTaskHistoryHelper(
    TaskManager taskManager,
    HistoryManager historyManager,
    SingularityConfiguration configuration
  ) {
    super(configuration.getDatabaseConfiguration().isPresent());
    this.taskManager = taskManager;
    this.historyManager = historyManager;
  }

  @Override
  protected List getFromZk(
    final SingularityDeployKey deployKey
  ) {
    List deployTaskIds = taskManager.getInactiveTaskIdsForDeploy(
      deployKey.getRequestId(),
      deployKey.getDeployId()
    );
    return getTaskHistoriesFor(taskManager, deployTaskIds);
  }

  @Override
  protected List getFromHistory(
    final SingularityDeployKey deployKey,
    int historyStart,
    int numFromHistory
  ) {
    return historyManager.getTaskIdHistory(
      Optional.of(deployKey.getRequestId()),
      Optional.of(deployKey.getDeployId()),
      Optional.empty(),
      Optional.empty(),
      Optional.empty(),
      Optional.empty(),
      Optional.empty(),
      Optional.empty(),
      Optional.empty(),
      Optional.empty(),
      Optional.of(historyStart),
      numFromHistory
    );
  }

  @Override
  protected Optional getTotalCount(
    SingularityDeployKey deployKey,
    boolean canSkipZk
  ) {
    final int numFromZk;
    if (sqlEnabled && canSkipZk) {
      numFromZk = 0;
    } else {
      numFromZk =
        taskManager
          .getInactiveTaskIdsForDeploy(deployKey.getRequestId(), deployKey.getDeployId())
          .size();
    }
    final int numFromHistory = historyManager.getTaskIdHistoryCount(
      Optional.of(deployKey.getRequestId()),
      Optional.of(deployKey.getDeployId()),
      Optional.empty(),
      Optional.empty(),
      Optional.empty(),
      Optional.empty(),
      Optional.empty(),
      Optional.empty(),
      Optional.empty()
    );
    return Optional.of(numFromZk + numFromHistory);
  }
}