org.sfm.poi.impl.PoiEnumGetter Maven / Gradle / Ivy

Go to download
package org.sfm.poi.impl;


import org.apache.poi.ss.usermodel.Cell;
import org.apache.poi.ss.usermodel.Row;
import org.sfm.reflect.EnumHelper;
import org.sfm.reflect.Getter;

public class PoiEnumGetter> implements Getter {

    private final int index;
    private final Class enumClass;
    private final E[] values;
    public PoiEnumGetter(int index, Class enumClass) {
        this.index = index;
        this.enumClass = enumClass;
        this.values = EnumHelper.getValues(enumClass);
    }

    @Override
    public E get(Row target) throws Exception {
        final Cell cell = target.getCell(index);
        if (cell != null) {
            switch (cell.getCellType()) {
                case Cell.CELL_TYPE_BLANK : return null;
                case Cell.CELL_TYPE_STRING : return Enum.valueOf(enumClass, cell.getStringCellValue());
                case Cell.CELL_TYPE_NUMERIC : return values[(int)cell.getNumericCellValue()];
                default:
                    throw new UnsupportedOperationException("Cannot convert cell to enum");
            }
        } else {
            return null;
        }
    }

}