From 3f0857b4149c510833fdf591b4e4c84ee20a2564 Mon Sep 17 00:00:00 2001
From: Stefano Cordio <stefano.cordio@gmail.com>
Date: Sat, 10 May 2025 00:59:50 +0200
Subject: [PATCH] Remove unnecessary reflection in
 `RemoteChunkHandlerFactoryBean`

Signed-off-by: Stefano Cordio <stefano.cordio@gmail.com>
---
 .../chunk/RemoteChunkHandlerFactoryBean.java        | 13 ++-----------
 1 file changed, 2 insertions(+), 11 deletions(-)

diff --git a/spring-batch-integration/src/main/java/org/springframework/batch/integration/chunk/RemoteChunkHandlerFactoryBean.java b/spring-batch-integration/src/main/java/org/springframework/batch/integration/chunk/RemoteChunkHandlerFactoryBean.java
index 9def1ae7c..26ffef4ff 100644
--- a/spring-batch-integration/src/main/java/org/springframework/batch/integration/chunk/RemoteChunkHandlerFactoryBean.java
+++ b/spring-batch-integration/src/main/java/org/springframework/batch/integration/chunk/RemoteChunkHandlerFactoryBean.java
@@ -122,12 +122,12 @@ public class RemoteChunkHandlerFactoryBean<T> implements FactoryBean<ChunkHandle
 			stepContributionSource = (StepContributionSource) chunkWriter;
 		}
 
-		Assert.state(step instanceof TaskletStep, "Step [" + step.getName() + "] must be a TaskletStep");
+		Assert.state(step != null, "A TaskletStep must be provided");
 		if (logger.isDebugEnabled()) {
 			logger.debug("Converting TaskletStep with name=" + step.getName());
 		}
 
-		Tasklet tasklet = getTasklet(step);
+		Tasklet tasklet = step.getTasklet();
 		Assert.state(tasklet instanceof ChunkOrientedTasklet<?>,
 				"Tasklet must be ChunkOrientedTasklet in step=" + step.getName());
 
@@ -227,15 +227,6 @@ public class RemoteChunkHandlerFactoryBean<T> implements FactoryBean<ChunkHandle
 		return (ChunkProcessor<T>) getField(tasklet, "chunkProcessor");
 	}
 
-	/**
-	 * Pull a Tasklet out of a step.
-	 * @param step a TaskletStep
-	 * @return the Tasklet
-	 */
-	private Tasklet getTasklet(TaskletStep step) {
-		return (Tasklet) getField(step, "tasklet");
-	}
-
 	private static Object getField(Object target, String name) {
 		Assert.notNull(target, "Target object must not be null");
 		Field field = ReflectionUtils.findField(target.getClass(), name);