Spring Batch：无写入项时跳过邮件步骤的实现咨询

嘿，这个需求我之前做Spring Batch时也碰到过，其实要把上一步的写入计数传给决策器，核心是利用Spring Batch提供的StepExecution对象，它里面就存着我们需要的写入统计数据。我给你一步步拆解实现方式：

1. 自定义决策器，读取上一步的写入计数 #

我们需要实现JobExecutionDecider接口，在决策逻辑里拿到目标步骤的StepExecution，从中提取writeCount（成功写入的项数）。

public class WriteCountDecider implements JobExecutionDecider {

    private String targetStepName; // 要检查的前置步骤名称

    // 通过构造器注入目标步骤名称，也可以用setter方法
    public WriteCountDecider(String targetStepName) {
        this.targetStepName = targetStepName;
    }

    @Override
    public FlowExecutionStatus decide(JobExecution jobExecution, StepExecution currentStepExecution) {
        // 从JobExecution中找到目标步骤的StepExecution
        StepExecution targetStepExec = jobExecution.getStepExecutions().stream()
                .filter(stepExec -> stepExec.getStepName().equals(targetStepName))
                .findFirst()
                .orElseThrow(() -> new IllegalArgumentException("找不到步骤: " + targetStepName));

        int writeCount = targetStepExec.getWriteCount();
        // 根据写入计数返回不同状态，供流程跳转判断
        if (writeCount == 0) {
            return new FlowExecutionStatus("NO_WRITES");
        } else {
            return new FlowExecutionStatus("COMPLETED");
        }
    }
}

2. 配置Job流程，关联决策器与步骤 #

接下来在你的Batch配置里，注册决策器Bean，然后在流程中插入决策节点，根据决策结果跳转。以XML配置为例：

<!-- 注册步骤1后的决策器，指定要检查的步骤是step1 -->
<bean id="afterStep1Decider" class="com.yourpackage.WriteCountDecider">
    <constructor-arg value="step1"/> <!-- 替换成你实际的步骤1名称 -->
</bean>

<!-- 注册步骤3后的决策器，指定要检查的步骤是step3 -->
<bean id="afterStep3Decider" class="com.yourpackage.WriteCountDecider">
    <constructor-arg value="step3"/> <!-- 替换成你实际的步骤3名称 -->
</bean>

<!-- 作业主流程配置 -->
<batch:job id="yourBatchJob">
    <!-- 步骤1：读取、处理、写入File1 -->
    <batch:step id="step1">
        <!-- 这里放你的ItemReader、ItemProcessor、ItemWriter配置 -->
        <batch:next on="COMPLETED" to="afterStep1Decision"/>
    </batch:step>

    <!-- 步骤1后的决策节点 -->
    <batch:decision id="afterStep1Decision" decider="afterStep1Decider">
        <batch:next on="NO_WRITES" to="step3"/> <!-- 无写入项，跳过邮件直接到步骤3 -->
        <batch:next on="COMPLETED" to="sendFile1Email"/> <!-- 有写入项，执行邮件步骤 -->
    </batch:decision>

    <!-- 发送File1邮件的步骤 -->
    <batch:step id="sendFile1Email">
        <!-- 这里放你的邮件发送逻辑（比如用Tasklet实现） -->
        <batch:next on="COMPLETED" to="step3"/>
    </batch:step>

    <!-- 步骤3：读取、处理、写入File2 -->
    <batch:step id="step3">
        <!-- 这里放你的ItemReader、ItemProcessor、ItemWriter配置 -->
        <batch:next on="COMPLETED" to="afterStep3Decision"/>
    </batch:step>

    <!-- 步骤3后的决策节点 -->
    <batch:decision id="afterStep3Decision" decider="afterStep3Decider">
        <batch:next on="NO_WRITES" to="jobEnd"/> <!-- 无写入项，跳过邮件直接结束作业 -->
        <batch:next on="COMPLETED" to="sendFile2Email"/> <!-- 有写入项，执行邮件步骤 -->
    </batch:decision>

    <!-- 发送File2邮件的步骤 -->
    <batch:step id="sendFile2Email">
        <!-- 这里放你的邮件发送逻辑 -->
        <batch:next on="COMPLETED" to="jobEnd"/>
    </batch:step>

    <batch:end id="jobEnd"/>
</batch:job>

3. 关键注意点 #

• 确保决策器中指定的targetStepName和配置里的步骤ID完全一致，否则会找不到对应的StepExecution
• getWriteCount()统计的是成功写入的项数，如果步骤中有跳过的项，会自动排除，正好符合我们“只有真正写入内容才发邮件”的需求
• 如果用Java Config代替XML配置，逻辑完全一致，只是用JobBuilderFactory和StepBuilderFactory构建流程时，通过decider()方法插入决策节点即可

内容的提问来源于stack exchange，提问作者Manvi