extienda MongoItemReader y proporcione su propia implementación para el método doPageRead(). De esta manera tendrá soporte completo de paginación y esta lectura de documentos será parte de un paso.
public class CustomMongoItemReader<T, O> extends MongoItemReader<T> {
private MongoTemplate template;
private Class<? extends T> inputType;
private Class<O> outputType
private MatchOperation match;
private ProjectionOperation projection;
private String collection;
@Override
protected Iterator<T> doPageRead() {
Pageable page = PageRequest.of(page, pageSize) //page and page size are coming from the class that MongoItemReader extends
Aggregation agg = newAggregation(match, projection, skip(page.getPageNumber() * page.getPageSize()), limit(page.getPageSize()));
return (Iterator<T>) template.aggregate(agg, collection, outputType).iterator();
}
}
Y otros getter y setters y otros métodos. Solo eche un vistazo al código fuente de MongoItemReader aquí También eliminé el soporte de Query. Puede tener eso también en el mismo método, simplemente copie y péguelo desde MongoItemReader. Lo mismo con Ordenar.
Y en la clase donde tienes un lector, harías algo como:
public MongoItemReader<T> reader() {
CustomMongoItemReader reader = new CustomMongoItemReader();
reader.setTemplate(mongoTemplate);
reader.setName("abc");
reader.setTargetType(input.class);
reader.setOutputType(output.class);
reader.setCollection(myMongoCollection);
reader.setMatch(Aggregation.match(new Criteria()....)));
reader.setProjection(Aggregation.project("..","..");
return reader;
}