Skip to content

Commit 1daf19d

Browse files
Merge pull request eBay#4 from VasilievAleksey/add_get_list_of_sheets
Add method to retrieve a list of sheets
2 parents bc22127 + 9568112 commit 1daf19d

File tree

1 file changed

+23
-0
lines changed

1 file changed

+23
-0
lines changed

src/main/java/com/ebay/xcelite/Xcelite.java

Lines changed: 23 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -24,6 +24,8 @@
2424
import org.apache.poi.xssf.usermodel.XSSFWorkbook;
2525

2626
import java.io.*;
27+
import java.util.ArrayList;
28+
import java.util.List;
2729

2830
/**
2931
* Class description...
@@ -98,6 +100,27 @@ public XceliteSheet getSheet(String sheetName) {
98100
return new XceliteSheetImpl(sheet);
99101
}
100102

103+
/**
104+
* Gets all sheets.
105+
*
106+
* @return the list of sheets (a list of {@link XceliteSheet} objects.)
107+
*/
108+
public List<XceliteSheet> getSheets() {
109+
final int numberOfSheets = workbook.getNumberOfSheets();
110+
111+
if (numberOfSheets == 0) {
112+
throw new XceliteException("Could not find any sheet");
113+
}
114+
115+
List<XceliteSheet> xceliteSheets = new ArrayList<>();
116+
117+
for (int i = 0; i < numberOfSheets; i++) {
118+
xceliteSheets.add(new XceliteSheetImpl(workbook.getSheetAt(i)));
119+
}
120+
121+
return xceliteSheets;
122+
}
123+
101124
/**
102125
* Saves data to the input file.
103126
*/

0 commit comments

Comments
 (0)