com.breuninger.boot.jobs.repository.inmem.InMemJobRepository Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of spring-boot-starter-breuninger-jobs Show documentation
spring-boot-starter-breuninger-jobs
There is a newer version: 2.0.5.RELEASE
Show newest version
package com.breuninger.boot.jobs.repository.inmem;

import static java.util.Collections.emptyList;
import static java.util.Comparator.comparing;
import static java.util.Comparator.reverseOrder;
import static java.util.Objects.nonNull;
import static java.util.Optional.ofNullable;
import static java.util.stream.Collectors.toList;

import java.time.OffsetDateTime;
import java.util.Comparator;
import java.util.HashSet;
import java.util.List;
import java.util.Optional;
import java.util.Set;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.ConcurrentMap;

import com.breuninger.boot.jobs.domain.JobInfo;
import com.breuninger.boot.jobs.domain.JobInfo.JobStatus;
import com.breuninger.boot.jobs.domain.JobMessage;
import com.breuninger.boot.jobs.repository.JobRepository;

public class InMemJobRepository implements JobRepository {

  private static final Comparator STARTED_TIME_DESC_COMPARATOR = comparing(JobInfo::getStarted, reverseOrder());

  private final ConcurrentMap jobs = new ConcurrentHashMap<>();

  @Override
  public Optional findOne(final String uri) {
    return ofNullable(jobs.get(uri));
  }

  @Override
  public List findLatest(final int maxCount) {
    return jobs.values()
      .stream()
      .sorted(STARTED_TIME_DESC_COMPARATOR)
      .limit(maxCount)
      .collect(toList());
  }

  @Override
  public List findLatestJobsDistinct() {
    final Set typeSet = new HashSet<>();

    return jobs.values()
      .stream()
      .sorted(STARTED_TIME_DESC_COMPARATOR)
      .filter(j -> nonNull(j.getJobType()))
      .filter(j -> typeSet.add(j.getJobType()))
      .collect(toList());
  }

  @Override
  public List findLatestBy(final String type, final int maxCount) {
    return jobs.values()
      .stream()
      .sorted(STARTED_TIME_DESC_COMPARATOR)
      .filter(jobInfo -> jobInfo.getJobType().equalsIgnoreCase(type))
      .limit(maxCount)
      .collect(toList());
  }

  @Override
  public List findRunningWithoutUpdateSince(final OffsetDateTime timeOffset) {
    return jobs.values()
      .stream()
      .filter(jobInfo -> !jobInfo.isStopped() && jobInfo.getLastUpdated().isBefore(timeOffset))
      .collect(toList());
  }

  @Override
  public List findAll() {
    return jobs.values().stream().sorted(STARTED_TIME_DESC_COMPARATOR).collect(toList());
  }

  @Override
  public List findAllJobInfoWithoutMessages() {
    return jobs.values()
      .stream()
      .sorted(STARTED_TIME_DESC_COMPARATOR)
      .map(job -> job.copy().setMessages(emptyList()).build())
      .collect(toList());
  }

  @Override
  public List findByType(final String jobType) {
    return jobs.values().stream().filter(jobInfo -> jobInfo.getJobType().equals(jobType)).collect(toList());
  }

  @Override
  public JobInfo createOrUpdate(final JobInfo job) {
    jobs.put(job.getJobId(), job);
    return job;
  }

  @Override
  public void removeIfStopped(final String id) {
    final var jobInfo = jobs.get(id);
    if (jobInfo != null && jobInfo.isStopped()) {
      jobs.remove(id);
    }
  }

  @Override
  public JobStatus findStatus(final String jobId) {
    return jobs.get(jobId).getStatus();
  }

  @Override
  public void appendMessage(final String jobId, final JobMessage jobMessage) {
    final var jobInfo = jobs.get(jobId);
    jobs.replace(jobId, jobInfo.copy().setLastUpdated(jobMessage.getTimestamp()).addMessage(jobMessage).build());
  }

  @Override
  public void setJobStatus(final String jobId, final JobStatus jobStatus) {
    final var jobInfo = jobs.get(jobId);
    jobs.replace(jobId, jobInfo.copy().setStatus(jobStatus).build());
  }

  @Override
  public void setLastUpdate(final String jobId, final OffsetDateTime lastUpdate) {
    final var jobInfo = jobs.get(jobId);
    jobs.replace(jobId, jobInfo.copy().setLastUpdated(lastUpdate).build());
  }

  @Override
  public long size() {
    return jobs.size();
  }

  @Override
  public void deleteAll() {
    jobs.clear();
  }
}