ExecutionContext 传递参数

关于 Spring Batch 的读数据（ ItemReader ）、处理数据（ ItemProcessor ）、写数据（ ItemWriter ）的配置流程，可以参考前面系列文章，本文不再详细描述。我们需要记住的是，当一个作业（ Job ）启动，Spring Batch 是通过作业名称（ Job name）及 作业参数（ JobParameters ）作为唯一标识来区分不同的作业。一个 Job 下可以有多个作业步（ Step ），每个 Step 中就是有具体的操作逻辑（读、处理、写）。在 Job 和 Step 下的各个操作步骤间，如何传递，，这里就需要理解 ExecutionContext 的概念。

ExecutionContext 概念

在 Job 的运行及 Step 的运行过程中，Spring Batch 提供 ExecutionContext 进行运行数据持久化，利用它，可以根据业务进行数据共享，如用来重启的静态数据与状态数据。如下图：

Execution Context 本质上来讲就是一个 Map<String,Object> ，它是Spring Batch 框架提供的持久化与控制的 key/value 对，可以让开发者在 Step 运行或Job 运行过程中保存需要进行持久化的状态，它可以。分为两类，一类是Job 运行的上下文（对应数据表：BATCH_JOB_EXECUTION_CONTEXT），另一类是Step Execution的上下文（对应数据表BATCH_STEP_EXECUTION_CONTEXT）。两类上下文关系：一个 Job 运行对应一个 Job Execution 的上下文（如上图中蓝色部分的 ExecutionContext ），每个 Step 运行对应一个 Step Execution 上下文（如上图中粉色部分的 ExecutionContext ）；同一个 Job 中的 Step Execution 共用 Job Execution 的上下文。也就是说，它们的作用范围有区别。因此，如果同一个 Job 的不同 Step 间需要共享数据时，可以通过 Job Execution 的上下文共享数据。根据 ExecutionContext 的共享数据特性，则可以实现在不同步骤间传递数据。

ExecutionContext 传递数据

一个 Job 启动后，会生成一个 JobExecution ，用于存放和记录 Job 运行的信息，同样，在 Step 启动后，也会有对应的 StepExecution 。如前面所说，在 JobExecution 和 StepExecution 中都会有一个 ExecutionContext ，用于存储上下文。因此，数据传递的思路就是确定数据使用范围，然后通过 ExecutionContext 传入数据，然后就可以在对应的范围内共享数据。如当前示例，需要 Job 范围内共享数据，在读组件（ ItemReader ）和写组件（ ItemWriter ）中传递读与写数据的数量（ size ），在 Job 结束时，输出读及写的数据量。实际上 Spring Batch 会自动计算读写数量，本示例仅为了显示数据共享功能。

那么，如何获取对应的 Execution ？，Spring Batch 提供了 JobExecutionListener 和 StepExecutionListener 监听器接口，通过实现监听器接口，分别可以在开启作业前（ beforeJob ）和 完成作业后（ afterJob ）afterJob ），开启作业步前（ beforeStep）及 完成作业步后（ afterStep ）获取对应的 Execution ，然后进行操作。

实现监听器接口

在自定义的 UserItemReader 和 UserItemWriter 中，实现 StepExecutionListener 接口，其中使用 StepExecution 作为成员，从 beforeStep 中获取。如下：

public class UserItemWriter implements ItemWriter<TargetUser>, StepExecutionListener {
    private StepExecution stepExecution;
    //...略
    @Override
    public void beforeStep(StepExecution stepExecution) {
        this.stepExecution = stepExecution;
    }
}

读组件（ UserItemReader ）也使用同样的方式。而在作业结束后，获取参数，则可以继承 JobExecutionListenerSupport ，实现自己感兴趣的方法，也从参数中获取 JobExecution，然后获取参数进行处理。

public class ParamJobEndListener extends JobExecutionListenerSupport {
    @Override
    public void afterJob(JobExecution jobExecution) {}
}

设置用于传递的数据

由于我们需要在 Job 范围内传递参数，获取到 StepExecution 后，可以获得相应的 JobExecution ，进而获取 Job 对应的 executionContext，这样，就可以在 Job 范围内共享参数数据了。如下是在读组件中进行配置

ExecutionContext executionContext = stepExecution.getJobExecution().getExecutionContext();
            executionContext.put(SyncConstants.PASS_PARAM_READ_NUM, items.size());

同样在写组件中，获取到 ExecutionContext 后，可以对参数进行处理。本示例中，是通过对 ItemReader 传递的处理数目参数进行累加处理，得到结果。

@Override
public void write(List<? extends TargetUser> items) {
    ExecutionContext executionContext = stepExecution.getJobExecution().getExecutionContext();
    Object currentWriteNum = executionContext.get(SyncConstants.PASS_PARAM_WRITE_NUM);
    if (Objects.nonNull(currentWriteNum)) {
        log.info("currentWriteNum:{}", currentWriteNum);
        executionContext.put(SyncConstants.PASS_PARAM_WRITE_NUM, items.size()+(Integer)currentWriteNum);
    } else {
        executionContext.put(SyncConstants.PASS_PARAM_WRITE_NUM, items.size());
    }

最后在作业结束后，在实现 JobExecutionListenerSupport 的接口中，afterJob 函数中，对参数进行输出。

public class ParamJobEndListener extends JobExecutionListenerSupport {
    @Override
    public void afterJob(JobExecution jobExecution) {
        ExecutionContext executionContext = jobExecution.getExecutionContext();
        Integer writeNum = (Integer)executionContext.get(SyncConstants.PASS_PARAM_WRITE_NUM);
        log.info(LogConstants.LOG_TAG + "writeNum:{}",writeNum);
    }
}

StepScope 动态绑定参数传递

StepScope及后期绑定

前面说到在 Job 及 Step 范围内，使用 ExecutionContext 进行数据共享，但，如果需要在 Job 启动前设置参数，并且每次启动输入的参数是动态变化的（比如增量同步时，日期是基于上一次同步的时间或者ID），也就是说，每次运行，需要根据参数新建一个操作步骤（如 ItemReader、ItemWriter等），我们知道，由于在 Spring IOC 中加载的Bean，默认都是单例模式的，因此，需要每次运行新建，运行完销毁，新建是在运行时进行的。这就需要用到StepScope 及后期绑定技术。

在之前的示例中，已出现过 StepScope，它的作用是提供了操作步骤的作用范围，某个 Spring Bean 使用注解StepScope，则表示此 Bean 在作业步（ Step ）开始的时候初始化，在 Step 结束的时候销毁，也就是说 Bean的作用范围是在 Step 这个生命周期中。而 Spring Batch 通过属性后期绑定技术，在运行期获取属性值，并使用 SPEL 的表达式进行属性绑定。而在 StepScope 中，Spring Batch 框架提供 JobParameters，JobExecutionContext，StepExecutionContext，当然也可以使用 Spring 容器中的 Bean ，如 JobExecution ，StepExecution。

作业参数传递及动态获取 StepExecution

一个 Job 是由 Job name 及 JobParameters 作为唯一标识的，也就是说只有 job name 和 JobParameters 不一致时，Spring Batch 才会启动一个新的 Job，一致的话就当作是同一个 Job ，若 此 Job 未执行过，则执行；若已执行过且是 FAILED 状态，则尝试重新运行此 Job ，若已执行过且是 COMPLETED 状态，则会报错。

本示例中，Job 启动时输入时间参数，在 ItemReader 中使用 StepScope 注解，然后把时间参数绑定到 ItemReader 中，同时绑定 StepExecution ，以便于在 ItemReader 对时间参数及 StepExecution 进行操作。

设置时间参数

在使用 JobLauncher 启动 Job 时，是需要输入 jobParameters 作为参数的。因此可以创建此对象，并设置参数。

JobParameters jobParameters = new JobParametersBuilder()
                .addLong("time",timMillis)
                .toJobParameters();

动态绑定参数

在配置 Step 时，需要创建ItemReader 的 Bean，为了使用动态参数，在 ItemReader 中设置 Map 存放参数，并设置 StepExecution 为成员，以便于后面使用 ExecutionContext。

public class UserItemReader implements ItemReader<User> {
    protected Map<String, Object> params;
    private StepExecution stepExecution;
    public void setStepExecution(StepExecution stepExecution) {
        this.stepExecution = stepExecution;
    }
}

使用 StepScope 进行配置：

@Bean
@StepScope
public ItemReader paramItemReader(@Value("#{stepExecution}") StepExecution stepExecution,
                                  @Value("#{jobParameters['time']}") Long timeParam) {
    UserItemReader userItemReader = new UserItemReader();
    //设置参数
    Map<String, Object> params = CollUtil.newHashMap();
    Date datetime = new Date(timeParam);
    params.put(SyncConstants.PASS_PARAM_DATETIME, datetime);
    userItemReader.setParams(params);
    userItemReader.setStepExecution(stepExecution);
    return userItemReader;
}

注意：此时 ItemReader 不可再使用实现 StepExecutionListener 的方式来对 stepExecution 赋值，由于 ItemReader 是动态绑定的，StepExecutionListener 将不再起作用，因此需要在后期绑定中来绑定 stepExecution Bean 的方式来赋值。

设置及传递参数

ItemReader 获取到 StepExecution 后即可获取 ExecutionContext，然后可以像前面说的使用 ExecutionContext 方式进行数据传递。如下：

ExecutionContext executionContext = stepExecution.getJobExecution().getExecutionContext();
//readNum参数
executionContext.put(SyncConstants.PASS_PARAM_READ_NUM, items.size());
//datetime参数
executionContext.put(SyncConstants.PASS_PARAM_DATETIME,params.get(SyncConstants.PASS_PARAM_DATETIME));

总结

在 Job 和 Step 不同的数据范围中，可使用 ExecutionContext 共享数据。本文以传递处理数量为例，使用 Mybatis-plus，基于 ExecutionContext ，结合 StepScope及后期绑定技术，实现在 Job 启动传入参数，然后在 ItemReader、ItemProcessor、ItemWriter 及 Job 完成后的数据共享及传递。如果你在使用 Spring Batch 过程中需要进行数据共享与传递，请试试这种方式吧。