From 8939611e36a779e07893c50a255b4bb9a1643f23 Mon Sep 17 00:00:00 2001 From: kimjihun Date: Fri, 2 Aug 2024 20:18:13 +0900 Subject: [PATCH] =?UTF-8?q?*=20=EB=A6=AC=ED=8C=A9=ED=84=B0=EB=A7=81?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../java/com/example/springbatch/batch/ExcelRowReader.java | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/src/main/java/com/example/springbatch/batch/ExcelRowReader.java b/src/main/java/com/example/springbatch/batch/ExcelRowReader.java index e94e26c..d7b817d 100644 --- a/src/main/java/com/example/springbatch/batch/ExcelRowReader.java +++ b/src/main/java/com/example/springbatch/batch/ExcelRowReader.java @@ -28,7 +28,8 @@ public class ExcelRowReader implements ItemReader { Workbook workbook = WorkbookFactory.create(fileInputStream); Sheet sheet = workbook.getSheetAt(0); this.rowCursor = sheet.iterator(); - // Skip header row if necessary + + if (rowCursor.hasNext()) { rowCursor.next(); } @@ -36,10 +37,11 @@ public class ExcelRowReader implements ItemReader { @Override public Row read() { + if (rowCursor != null && rowCursor.hasNext()) { return rowCursor.next(); } else { - return null; // No more rows to read + return null; } } }