com.transferwise.tasks.dao.ITaskDao Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of tw-tasks-core Show documentation
Transferwise Tasks Executor - Fire and forget until Alert.
The newest version!
package com.transferwise.tasks.dao;

import com.transferwise.tasks.ITasksService.AddTaskRequest.CompressionRequest;
import com.transferwise.tasks.domain.BaseTask;
import com.transferwise.tasks.domain.IBaseTask;
import com.transferwise.tasks.domain.Task;
import com.transferwise.tasks.domain.TaskContext;
import com.transferwise.tasks.domain.TaskStatus;
import com.transferwise.tasks.domain.TaskVersionId;
import edu.umd.cs.findbugs.annotations.SuppressFBWarnings;
import java.time.Duration;
import java.time.Instant;
import java.time.ZonedDateTime;
import java.util.List;
import java.util.Map;
import java.util.UUID;
import lombok.Data;
import lombok.experimental.Accessors;
import org.apache.commons.lang3.tuple.Pair;

public interface ITaskDao {

  @Data
  @Accessors(chain = true)
  class StuckTask implements IBaseTask {

    private TaskVersionId versionId;
    private int priority;
    private String type;
    private String status;

    @Override
    public UUID getId() {
      return versionId.getId();
    }

    @Override
    public long getVersion() {
      return versionId.getVersion();
    }
  }

  @Data
  @Accessors(chain = true)
  class InsertTaskRequest {

    private String type;
    private String subType;
    @SuppressFBWarnings("EI_EXPOSE_REP")
    private byte[] data;
    private UUID taskId;
    private String key;
    private ZonedDateTime runAfterTime;
    private TaskStatus status;
    private ZonedDateTime maxStuckTime;
    private Integer priority;
    private CompressionRequest compression;
    private TaskContext taskContext;
  }

  @Data
  @Accessors(chain = true)
  class InsertTaskResponse {

    private UUID taskId;
    private boolean inserted;
  }

  @Data
  @Accessors(chain = true)
  class GetStuckTasksResponse {

    private List stuckTasks;
    private boolean hasMore;
  }

  @Data
  @Accessors(chain = true)
  class DeleteFinishedOldTasksResult {

    private int foundTasksCount;
    private int deletedTasksCount;
    private int deletedUniqueKeysCount;
    private int deletedTaskDatasCount;
    private UUID firstDeletedTaskId;
    private ZonedDateTime firstDeletedTaskNextEventTime;
    private ZonedDateTime deletedBeforeTime;
  }

  ZonedDateTime getEarliestTaskNextEventTime(TaskStatus status);

  List prepareStuckOnProcessingTasksForResuming(String clientId, ZonedDateTime maxStuckTime);

  GetStuckTasksResponse getStuckTasks(int batchSize, TaskStatus status);

  InsertTaskResponse insertTask(InsertTaskRequest request);

  int getTasksCountInStatus(int maxCount, TaskStatus status);

  Map getErronousTasksCountByType(int maxCount);

  int getStuckTasksCount(ZonedDateTime age, int maxCount);

  Map, Integer> getStuckTasksCountByStatusAndType(ZonedDateTime age, int maxCount);

   T getTask(UUID taskId, Class clazz);

  DeleteFinishedOldTasksResult deleteOldTasks(TaskStatus taskStatus, Duration age, int batchSize);

  @SuppressWarnings("UnusedReturnValue")
  boolean deleteTask(UUID taskId, long version);

  boolean clearPayloadAndMarkDone(UUID taskId, long version);

  boolean setToBeRetried(UUID id, ZonedDateTime retryTime, long version, boolean resetTriesCount);

  Task grabForProcessing(BaseTask task, String nodeId, Instant maxProcessingEndTime);

  boolean setStatus(UUID taskId, TaskStatus status, long version);

  boolean setNextEventTime(UUID taskId, ZonedDateTime nextEventTime, long version, String state);

  boolean markAsSubmitted(UUID taskId, long version, ZonedDateTime maxStuckTime);

  Long getTaskVersion(UUID id);

  long getApproximateTasksCount();

  long getApproximateUniqueKeysCount();

  long getApproximateTaskDatasCount();
}