Skip to content

Commit 658bc8f

Browse files
committed
style: Added reader() method
1 parent 13d3760 commit 658bc8f

File tree

1 file changed

+6
-4
lines changed

1 file changed

+6
-4
lines changed

plugins/riot/src/main/java/com/redis/riot/DatabaseImport.java

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33
import java.util.Map;
44

55
import org.springframework.batch.core.Job;
6-
import org.springframework.batch.item.database.builder.JdbcCursorItemReaderBuilder;
6+
import org.springframework.batch.item.database.JdbcCursorItemReader;
77

88
import picocli.CommandLine.ArgGroup;
99
import picocli.CommandLine.Command;
@@ -23,10 +23,12 @@ public class DatabaseImport extends AbstractRedisImportCommand {
2323

2424
@Override
2525
protected Job job() {
26+
return job(step(reader()));
27+
}
28+
29+
protected JdbcCursorItemReader<Map<String, Object>> reader() {
2630
log.info("Creating JDBC reader with sql=\"{}\" {} {}", sql, dataSourceArgs, readerArgs);
27-
JdbcCursorItemReaderBuilder<Map<String, Object>> reader = JdbcCursorItemReaderFactory.create(sql,
28-
dataSourceArgs, readerArgs);
29-
return job(step(reader.build()));
31+
return JdbcCursorItemReaderFactory.create(sql, dataSourceArgs, readerArgs).build();
3032
}
3133

3234
public String getSql() {

0 commit comments

Comments
 (0)