spring batch：在tasklet中查询存储过程结果的有效方式

Question

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

我正在使用 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

there has to be a more efficient way to do this

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();
}

编辑： 显示如何 return 同步 reader

@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;
}

spring batch：在tasklet中查询存储过程结果的有效方式

spring batch: efficient way to query results of a stored procedure within a tasklet

sql-server

spring

stored-procedures

spring-batch

spring-jdbc