From 195d136b0a56d2fe13c1d1bbd0a3c1f55162957a Mon Sep 17 00:00:00 2001
From: Tobias Soloschenko <tsoloschenko@apache.org>
Date: Mon, 17 Jul 2023 08:58:55 +0200
Subject: [PATCH] fix: use ExitCodeGenerator

---
 .../support/UnexpectedTaskExecutionException.java            | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/spring-cloud-dataflow-composed-task-runner/src/main/java/org/springframework/cloud/dataflow/composedtaskrunner/support/UnexpectedTaskExecutionException.java b/spring-cloud-dataflow-composed-task-runner/src/main/java/org/springframework/cloud/dataflow/composedtaskrunner/support/UnexpectedTaskExecutionException.java
index 7e1deb939f..73a21e9ee2 100644
--- a/spring-cloud-dataflow-composed-task-runner/src/main/java/org/springframework/cloud/dataflow/composedtaskrunner/support/UnexpectedTaskExecutionException.java
+++ b/spring-cloud-dataflow-composed-task-runner/src/main/java/org/springframework/cloud/dataflow/composedtaskrunner/support/UnexpectedTaskExecutionException.java
@@ -21,6 +21,7 @@
 import java.util.List;
 
 import org.springframework.batch.core.UnexpectedJobExecutionException;
+import org.springframework.boot.ExitCodeGenerator;
 import org.springframework.cloud.task.repository.TaskExecution;
 
 /**
@@ -29,7 +30,7 @@
  *
  * @author Tobias Soloschenko
  */
-public class UnexpectedTaskExecutionException extends UnexpectedJobExecutionException {
+public class UnexpectedTaskExecutionException extends UnexpectedJobExecutionException implements ExitCodeGenerator {
 
 	private static final long serialVersionUID = 3494615323781500165L;
 
@@ -153,7 +154,7 @@ public long getExecutionId() {
 		return this.executionId;
 	}
 
-	public Integer getExitCode() {
+	public int getExitCode() {
 		return this.exitCode;
 	}