github/spring_batch_5
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

* 리팩터링

Browse Source
This commit is contained in:
kimjihun
2024-08-02 20:18:13 +09:00
parent b597977108
commit 8939611e36
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/main/java/com/example/springbatch/batch/ExcelRowReader.java
View File

@@ -28,7 +28,8 @@ public class ExcelRowReader implements ItemReader<Row> {
Workbook workbook = WorkbookFactory.create(fileInputStream); Workbook workbook = WorkbookFactory.create(fileInputStream);
Sheet sheet = workbook.getSheetAt(0); Sheet sheet = workbook.getSheetAt(0);
this.rowCursor = sheet.iterator(); this.rowCursor = sheet.iterator();
// Skip header row if necessary
if (rowCursor.hasNext()) { if (rowCursor.hasNext()) {
rowCursor.next(); rowCursor.next();
} }
@@ -36,10 +37,11 @@ public class ExcelRowReader implements ItemReader<Row> {
@Override @Override
public Row read() { public Row read() {
if (rowCursor != null && rowCursor.hasNext()) { if (rowCursor != null && rowCursor.hasNext()) {
return rowCursor.next(); return rowCursor.next();
} else { } else {
return null; // No more rows to read return null;
} }
} }
} }