POIWorkbook,应该通过工厂得到IWorkbook比较合理
public class POIWorkbook implements IWorkbook {
private Workbook wb;
@Override
public ISheet getSheetAt(int index) {
if (wb == null) {
throw new RuntimeException("call read first.");
}
POISheet sheet = new POISheet(wb.getSheetAt(index));
sheet.setWorkbook(this);
return sheet;
}
@Override
public void write(File file) throws IOException, Exception {
FileUtils.touch(file);
wb.write(new FileOutputStream(file));
}
@Override
public IWorkbook read(File file) throws IOException, Exception {
wb = WorkbookFactory.create(file);
return this;
}
@Override
public Iterator<ISheet> iterator() {
final Iterator<Sheet> i = wb.iterator();
return new Iterator<ISheet>() {
@Override
public boolean hasNext() {
return i.hasNext();
}
@Override
public ISheet next() {
return new POISheet(i.next());
}
@Override
public void remove() {
i.remove();
}
};
}
@Override
public Iterator<ISheet> iterator(final ISheetFilter filter) {
final Iterator<Sheet> i = wb.iterator();
return new Iterator<ISheet>() {
@Override
public boolean hasNext() {
return i.hasNext();
}
@Override
public ISheet next() {
ISheet sheet = null;
while (i.hasNext()) {
ISheet temp = new POISheet(i.next());
if (filter.filter(temp)) {
sheet = temp;
break;
}
}
return sheet;
}
@Override
public void remove() {
i.remove();
}
};
}
@Override
public ISheet createSheet(String name) {
POISheet sheet = new POISheet(wb.createSheet(name));
sheet.setWorkbook(this);
return sheet;
}
@Override
public ISheet cloneSheet(int index) {
POISheet sheet = new POISheet(wb.cloneSheet(index));
sheet.setWorkbook(this);
return sheet;
}
@Override
public ISheet cloneSheet(String name) {
return cloneSheet(wb.getSheetIndex(name));
}
@Override
public void removeSheet(String name) {
wb.removeSheetAt(wb.getSheetIndex(name));
}
}