POI封装之四

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));
	}

}