#BAEL-8695: add main source

Author: hmdrzsharifi

spring-batch-2/src/main/java/com/baeldung/restartjob/BatchConfig.java

@@ -0,0 +1,72 @@
+package com.baeldung.restartjob;
+
+import org.springframework.batch.core.Job;
+import org.springframework.batch.core.Step;
+import org.springframework.batch.core.configuration.annotation.StepScope;
+import org.springframework.batch.core.job.builder.JobBuilder;
+import org.springframework.batch.core.repository.JobRepository;
+import org.springframework.batch.core.step.builder.StepBuilder;
+import org.springframework.batch.item.ItemProcessor;
+import org.springframework.batch.item.ItemWriter;
+import org.springframework.batch.item.file.FlatFileItemReader;
+import org.springframework.batch.item.file.builder.FlatFileItemReaderBuilder;
+import org.springframework.batch.item.file.mapping.PassThroughLineMapper;
+import org.springframework.context.annotation.Bean;
+import org.springframework.context.annotation.Configuration;
+import org.springframework.core.io.ClassPathResource;
+import org.springframework.transaction.PlatformTransactionManager;
+
+@Configuration
+public class BatchConfig {
+
+    @Bean
+    public Job simpleJob(JobRepository jobRepository, PlatformTransactionManager transactionManager) {
+        return new JobBuilder("simpleJob", jobRepository)
+          .start(step1(jobRepository, transactionManager))
+          .build();
+    }
+
+    @Bean
+    public Step step1(JobRepository jobRepository, PlatformTransactionManager transactionManager) {
+        return new StepBuilder("step1", jobRepository)
+          .<String, String>chunk(2, transactionManager)
+          .reader(flatFileItemReader())
+          .processor(itemProcessor())
+          .writer(itemWriter())
+          .build();
+    }
+
+    @Bean
+    @StepScope
+    public FlatFileItemReader<String> flatFileItemReader() {
+        return new FlatFileItemReaderBuilder<String>()
+          .name("itemReader")
+          .resource(new ClassPathResource("data.csv"))
+          .lineMapper(new PassThroughLineMapper())
+          .saveState(true)
+          .build();
+    }
+
+    @Bean
+    public ItemProcessor<String, String> itemProcessor() {
+        return item -> {
+          System.out.println("Processing: " + item);
+
+          if (item.equals("Item3")) {
+              throw new RuntimeException("Simulated failure on Item3");
+          }
+
+          return "PROCESSED " + item;
+        };
+    }
+
+    @Bean
+    public ItemWriter<String> itemWriter() {
+        return items -> {
+          System.out.println("Writing items:");
+          for (String item : items) {
+              System.out.println("- " + item);
+          }
+        };
+    }
+}

spring-batch-2/src/main/java/com/baeldung/restartjob/RestartJobBatchApp.java

@@ -0,0 +1,58 @@
+package com.baeldung.restartjob;
+
+import java.util.List;
+
+import org.springframework.batch.core.BatchStatus;
+import org.springframework.batch.core.Job;
+import org.springframework.batch.core.JobExecution;
+import org.springframework.batch.core.JobInstance;
+import org.springframework.batch.core.JobParameters;
+import org.springframework.batch.core.JobParametersBuilder;
+import org.springframework.batch.core.explore.JobExplorer;
+import org.springframework.batch.core.launch.JobLauncher;
+import org.springframework.batch.core.launch.JobOperator;
+import org.springframework.boot.CommandLineRunner;
+import org.springframework.boot.SpringApplication;
+import org.springframework.boot.autoconfigure.SpringBootApplication;
+import org.springframework.context.annotation.Bean;
+
+@SpringBootApplication
+public class RestartJobBatchApp {
+
+    public static void main(String[] args) {
+        SpringApplication app = new SpringApplication(RestartJobBatchApp.class);
+        app.setAdditionalProfiles("restart");
+        app.run(args);
+    }
+
+    @Bean
+    CommandLineRunner run(JobLauncher jobLauncher, Job job, JobExplorer jobExplorer, JobOperator jobOperator) {
+        return args -> {
+          JobParameters jobParameters = new JobParametersBuilder()
+              .addString("jobId", "test-job-" + System.currentTimeMillis())
+              .toJobParameters();
+
+          List<JobInstance> instances = jobExplorer.getJobInstances("simpleJob", 0, 1);
+          if (!instances.isEmpty()) {
+              JobInstance lastInstance = instances.get(0);
+              List<JobExecution> executions = jobExplorer.getJobExecutions(lastInstance);
+              if (!executions.isEmpty()) {
+                  JobExecution lastExecution = executions.get(0);
+                  if (lastExecution.getStatus() == BatchStatus.FAILED) {
+                      System.out.println("Restarting failed job execution with ID: " + lastExecution.getId());
+
+                      final Long restartId = jobOperator.restart(lastExecution.getId());
+                      final JobExecution restartedExecution = jobExplorer.getJobExecution(restartId);
+
+                      System.out.println("Restarted job status: " + restartedExecution.getStatus());
+                      return;
+                  }
+              }
+          }
+
+          System.out.println("Starting new job execution...");
+          JobExecution jobExecution = jobLauncher.run(job, jobParameters);
+          System.out.println("Job started with status: " + jobExecution.getStatus());
+      };
+    }
+}