Implement unsupported operations in ResourcelessJobRepository

Author: fmbenhassine

spring-batch-core/src/main/java/org/springframework/batch/core/repository/support/ResourcelessJobRepository.java

@@ -27,6 +27,7 @@
 import org.springframework.batch.core.step.StepExecution;
 import org.springframework.batch.core.repository.JobRepository;
 import org.springframework.batch.support.transaction.ResourcelessTransactionManager;
+import org.springframework.lang.Nullable;
 
 /**
  * A {@link JobRepository} implementation that does not use or store batch meta-data. It
@@ -50,6 +51,12 @@ public class ResourcelessJobRepository implements JobRepository {
 
 	private JobExecution jobExecution;
 
+	/*
+	 * ===================================================================================
+	 * Job operations
+	 * ===================================================================================
+	 */
+
 	@Override
 	public List<String> getJobNames() {
 		if (this.jobInstance == null) {
@@ -58,8 +65,41 @@ public List<String> getJobNames() {
 		return Collections.singletonList(this.jobInstance.getJobName());
 	}
 
+	/*
+	 * ===================================================================================
+	 * Job instance operations
+	 * ===================================================================================
+	 */
+
+	@Override
+	public List<JobInstance> getJobInstances(String jobName, int start, int count) {
+		if (this.jobInstance == null) {
+			return Collections.emptyList();
+		}
+		return Collections.singletonList(this.jobInstance);
+	}
+
+	@Override
+	@Nullable
+	public JobInstance getJobInstance(@Nullable Long instanceId) {
+		return this.jobInstance;
+	}
+
+	@Override
+	@Nullable
+	public JobInstance getLastJobInstance(String jobName) {
+		return this.jobInstance;
+	}
+
+	@Override
+	@Nullable
+	public JobInstance getJobInstance(String jobName, JobParameters jobParameters) {
+		return this.jobInstance;
+	}
+
 	@SuppressWarnings("removal")
 	@Override
+	@Deprecated(since = "6.0", forRemoval = true)
 	public boolean isJobInstanceExists(String jobName, JobParameters jobParameters) {
 		return false;
 	}
@@ -75,50 +115,79 @@ public JobInstance createJobInstance(String jobName, JobParameters jobParameters
 		return this.jobInstance;
 	}
 
+	/*
+	 * ===================================================================================
+	 * Job execution operations
+	 * ===================================================================================
+	 */
+
 	@Override
-	public JobExecution createJobExecution(String jobName, JobParameters jobParameters) {
-		if (this.jobInstance == null) {
-			createJobInstance(jobName, jobParameters);
-		}
-		this.jobExecution = new JobExecution(this.jobInstance, 1L, jobParameters);
+	@Nullable
+	public JobExecution getJobExecution(Long executionId) {
 		return this.jobExecution;
 	}
 
 	@Override
-	public void update(JobExecution jobExecution) {
-		jobExecution.setLastUpdated(LocalDateTime.now());
-		this.jobExecution = jobExecution;
+	@Nullable
+	public JobExecution getLastJobExecution(String jobName, JobParameters jobParameters) {
+		return this.jobExecution;
 	}
 
 	@Override
-	public void add(StepExecution stepExecution) {
-		this.addAll(Collections.singletonList(stepExecution));
+	@Nullable
+	public JobExecution getLastJobExecution(JobInstance jobInstance) {
+		return this.jobExecution;
 	}
 
 	@Override
-	public void addAll(Collection<StepExecution> stepExecutions) {
-		this.jobExecution.addStepExecutions(new ArrayList<>(stepExecutions));
+	public List<JobExecution> getJobExecutions(JobInstance jobInstance) {
+		if (this.jobExecution == null) {
+			return Collections.emptyList();
+		}
+		return Collections.singletonList(this.jobExecution);
 	}
 
 	@Override
-	public void update(StepExecution stepExecution) {
-		stepExecution.setLastUpdated(LocalDateTime.now());
-		if (this.jobExecution.isStopping()) {
-			stepExecution.setTerminateOnly();
+	public JobExecution createJobExecution(String jobName, JobParameters jobParameters) {
+		if (this.jobInstance == null) {
+			createJobInstance(jobName, jobParameters);
 		}
+		this.jobExecution = new JobExecution(this.jobInstance, 1L, jobParameters);
+		return this.jobExecution;
 	}
 
 	@Override
-	public void updateExecutionContext(StepExecution stepExecution) {
-		stepExecution.setLastUpdated(LocalDateTime.now());
+	public void update(JobExecution jobExecution) {
+		jobExecution.setLastUpdated(LocalDateTime.now());
+		this.jobExecution = jobExecution;
 	}
 
 	@Override
 	public void updateExecutionContext(JobExecution jobExecution) {
 		jobExecution.setLastUpdated(LocalDateTime.now());
 	}
 
+	/*
+	 * ===================================================================================
+	 * Step execution operations
+	 * ===================================================================================
+	 */
+
 	@Override
+	@Nullable
+	public StepExecution getStepExecution(Long jobExecutionId, Long stepExecutionId) {
+		if (this.jobExecution == null || !this.jobExecution.getId().equals(jobExecutionId)) {
+			return null;
+		}
+		return this.jobExecution.getStepExecutions()
+			.stream()
+			.filter(stepExecution -> stepExecution.getId().equals(stepExecutionId))
+			.findFirst()
+			.orElse(null);
+	}
+
+	@Override
+	@Nullable
 	public StepExecution getLastStepExecution(JobInstance jobInstance, String stepName) {
 		return this.jobExecution.getStepExecutions()
 			.stream()
@@ -136,8 +205,26 @@ public long getStepExecutionCount(JobInstance jobInstance, String stepName) {
 	}
 
 	@Override
-	public JobExecution getLastJobExecution(String jobName, JobParameters jobParameters) {
-		return this.jobExecution;
+	public void add(StepExecution stepExecution) {
+		this.addAll(Collections.singletonList(stepExecution));
+	}
+
+	@Override
+	public void addAll(Collection<StepExecution> stepExecutions) {
+		this.jobExecution.addStepExecutions(new ArrayList<>(stepExecutions));
+	}
+
+	@Override
+	public void update(StepExecution stepExecution) {
+		stepExecution.setLastUpdated(LocalDateTime.now());
+		if (this.jobExecution.isStopping()) {
+			stepExecution.setTerminateOnly();
+		}
+	}
+
+	@Override
+	public void updateExecutionContext(StepExecution stepExecution) {
+		stepExecution.setLastUpdated(LocalDateTime.now());
 	}
 
 }