Add sheetNames to WorkbookReader

crealytics · Jan 27, 2020 · 21b440f · 21b440f
1 parent 0bef78f
commit 21b440f
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/src/main/scala/com/crealytics/spark/excel/WorkbookReader.scala b/src/main/scala/com/crealytics/spark/excel/WorkbookReader.scala
@@ -16,6 +16,13 @@ trait WorkbookReader {
     workbook.close()
     res
   }
+  def sheetNames: Seq[String] = {
+    withWorkbook(workbook =>
+        for (sheetIx <- (0 until workbook.getNumberOfSheets())) yield {
+          workbook.getSheetAt(sheetIx).getSheetName()
+        }
+      )
+  }
 }
 
 object WorkbookReader {