From e8f93fd437f68f1d2ab34fae6b1aaf406a4e1f3a Mon Sep 17 00:00:00 2001
From: haerong22 <haerong22@naver.com>
Date: Thu, 23 Jun 2022 02:55:17 +0900
Subject: [PATCH] #11 spring batch: Custom ExitStatus

---
 .../flow/CustomExitStatusConfiguration.java   | 74 +++++++++++++++++++
 .../basic/flow/PassCheckingListener.java      | 24 ++++++
 .../basic/flow/TransitionConfiguration.java   |  2 +-
 3 files changed, 99 insertions(+), 1 deletion(-)
 create mode 100644 spring-batch/basic/src/main/java/io/springbatch/basic/flow/CustomExitStatusConfiguration.java
 create mode 100644 spring-batch/basic/src/main/java/io/springbatch/basic/flow/PassCheckingListener.java

diff --git a/spring-batch/basic/src/main/java/io/springbatch/basic/flow/CustomExitStatusConfiguration.java b/spring-batch/basic/src/main/java/io/springbatch/basic/flow/CustomExitStatusConfiguration.java
new file mode 100644
index 00000000..9bd7ccd3
--- /dev/null
+++ b/spring-batch/basic/src/main/java/io/springbatch/basic/flow/CustomExitStatusConfiguration.java
@@ -0,0 +1,74 @@
+package io.springbatch.basic.flow;
+
+import lombok.RequiredArgsConstructor;
+import org.springframework.batch.core.ExitStatus;
+import org.springframework.batch.core.Job;
+import org.springframework.batch.core.Step;
+import org.springframework.batch.core.configuration.annotation.JobBuilderFactory;
+import org.springframework.batch.core.configuration.annotation.StepBuilderFactory;
+import org.springframework.batch.core.launch.support.RunIdIncrementer;
+import org.springframework.batch.repeat.RepeatStatus;
+import org.springframework.context.annotation.Bean;
+import org.springframework.context.annotation.Configuration;
+
+@Configuration
+@RequiredArgsConstructor
+public class CustomExitStatusConfiguration {
+
+    private final JobBuilderFactory jobBuilderFactory;
+    private final StepBuilderFactory stepBuilderFactory;
+
+    /*
+        Custom ExitStatus
+
+        - ExitStatus 에 존재하지 않는 exitCode 를 새롭게 정의해서 설정
+        - StepExecutionListener 의 afterStep() 메소드에서 Custom exitCode 생성 후 새로운 ExitStatus 반환
+        - Step 실행 후 완료 시점에서 현재 exitCode 를 사용자 정의 exitCode 로 수정할 수 있음
+     */
+
+    @Bean
+    public Job batchJob() {
+        return jobBuilderFactory.get("batchJob")
+                .incrementer(new RunIdIncrementer())
+                .start(step1())
+                    .on("PASS") // custom exitCode
+                    .to(step2())
+                .from(step1())
+                    .on("*")
+                    .to(step3())
+                .end()
+                .build();
+    }
+
+    @Bean
+    public Step step1() {
+        return stepBuilderFactory.get("step1")
+                .tasklet((contribution, chunkContext) -> {
+                    System.out.println("step1 was executed");
+                    return RepeatStatus.FINISHED;
+                })
+                .listener(new PassCheckingListener())
+                .build();
+    }
+
+    @Bean
+    public Step step2() {
+        return stepBuilderFactory.get("step2")
+                .tasklet((contribution, chunkContext) -> {
+                    System.out.println("step2 was executed");
+                    return RepeatStatus.FINISHED;
+                })
+                .build();
+    }
+
+    @Bean
+    public Step step3() {
+        return stepBuilderFactory.get("step3")
+                .tasklet((contribution, chunkContext) -> {
+                    System.out.println("step3 was executed");
+                    return RepeatStatus.FINISHED;
+                })
+                .build();
+    }
+
+}
diff --git a/spring-batch/basic/src/main/java/io/springbatch/basic/flow/PassCheckingListener.java b/spring-batch/basic/src/main/java/io/springbatch/basic/flow/PassCheckingListener.java
new file mode 100644
index 00000000..f3dc478a
--- /dev/null
+++ b/spring-batch/basic/src/main/java/io/springbatch/basic/flow/PassCheckingListener.java
@@ -0,0 +1,24 @@
+package io.springbatch.basic.flow;
+
+import org.springframework.batch.core.ExitStatus;
+import org.springframework.batch.core.StepExecution;
+import org.springframework.batch.core.StepExecutionListener;
+
+public class PassCheckingListener implements StepExecutionListener {
+    @Override
+    public void beforeStep(StepExecution stepExecution) {
+
+    }
+
+    @Override
+    public ExitStatus afterStep(StepExecution stepExecution) {
+
+        String exitCode = stepExecution.getExitStatus().getExitCode();
+
+        if (!exitCode.equals(ExitStatus.FAILED.getExitCode())) {
+            return new ExitStatus("PASS");
+        }
+
+        return null;
+    }
+}
diff --git a/spring-batch/basic/src/main/java/io/springbatch/basic/flow/TransitionConfiguration.java b/spring-batch/basic/src/main/java/io/springbatch/basic/flow/TransitionConfiguration.java
index 8faca6a7..7dc30a25 100644
--- a/spring-batch/basic/src/main/java/io/springbatch/basic/flow/TransitionConfiguration.java
+++ b/spring-batch/basic/src/main/java/io/springbatch/basic/flow/TransitionConfiguration.java
@@ -11,7 +11,7 @@ import org.springframework.batch.repeat.RepeatStatus;
 import org.springframework.context.annotation.Bean;
 import org.springframework.context.annotation.Configuration;
 
-@Configuration
+//@Configuration
 @RequiredArgsConstructor
 public class TransitionConfiguration {