spring 批处理：查询 tasklet 中存储过程结果的有效方法

Question

我的目标是在 spring 批处理 tasklet 中使用自定义行映射器查询存储过程（sql server）的结果。

我正在使用 Spring Boot（版本 2.2.2.RELEASE）和 Spring Batch（4.2.1.RELEASE）。

例子：

存储过程：

CREATE PROCEDURE storedProcName @numbers VARCHAR(max), @day DATE
AS
SET NOCOUNT ON;

SELECT something, something2, something3
FROM sometable
WHERE ids in (select value from string_split(@numbers,','))
AND day = @day

自定义行映射器：

public class CustomRowMapper implements RowMapper<CustomObject> {
    private static final String SOMETHING = "something";
    private static final String SOMETHING2 = "something2";
    private static final String SOMETHING3 = "something3";

    @Override
    public CustomObject mapRow(ResultSet resultSet, int i) throws SQLException {
        CustomObject customObject = new CustomObject();
        customObject.setSomething(resultSet.getString(SOMETHING));
        customObject.setSomething2(resultSet.getString(SOMETHING2));
        customObject.setSomething3(resultSet.getInt(SOMETHING3));

        return customObject;
    }
}

执行存储过程并查询结果：

SimpleJdbcCall jdbcCall = new SimpleJdbcCall(jdbcTemplate)
        .withProcedureName("storedProcName")
        .returningResultSet("test", new CustomRowMapper());
Map<String, Object> out = jdbcCall.execute(parameterSource);
List<CustomObject> customObjects = (List<CustomObject>) out.get("test");

它工作正常，但必须有更有效的方法来做到这一点？ 但我没有发现任何有用的东西。

Answer 1

必须有更有效的方法来做到这一点

Spring Batch 提供了StoredProcedureItemReader ，它可以调用给定的存储过程并迭代其结果。 它还允许您使用自定义映射器作为您定义的映射器。

根据您的代码片段，您调用该过程并获得一个List<CustomObject> ，您需要在某个时候对其进行迭代。 所以我认为带有StoredProcedureItemReader<CustomObject>面向块的 tasklet 是一个不错的选择：

@Bean
public StoredProcedureItemReader<CustomObject> itemReader() {
    return new StoredProcedureItemReaderBuilder<CustomObject>()
            .procedureName("storedProcName")
            .rowMapper(new CustomRowMapper())
            // set other properties
            .build();
}

编辑：展示如何返回同步阅读器

@Bean
@StepScope
public SynchronizedItemStreamReader<CustomObject> itemReader() {
    StoredProcedureItemReader<CustomObject> reader = new StoredProcedureItemReaderBuilder<>()
            .procedureName("storedProcName")
            .rowMapper(new CustomRowMapper())
            // set other properties
            .build();

    SynchronizedItemStreamReader<CustomObject> synchronizedReader = new SynchronizedItemStreamReader<>();
    synchronizedReader.setDelegate(reader);
    return synchronizedReader;
}