com.github.bingoohuang.excel2beans.ExcelToBeans Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of excel2javabeans Show documentation

a little utility to convert excel rows to java beans

There is a newer version: 0.0.34

package com.github.bingoohuang.excel2beans;

import lombok.Getter;
import lombok.SneakyThrows;
import lombok.val;
import org.apache.commons.lang3.StringUtils;
import org.apache.poi.ss.usermodel.*;

import java.io.Closeable;
import java.io.IOException;
import java.io.InputStream;
import java.util.List;
import java.util.concurrent.atomic.AtomicInteger;

/**
 * Mapping excel rows to java beans.
 */
public class ExcelToBeans implements Closeable {
    private @Getter final Workbook workbook;
    private final boolean shouldBeClosedByMe;

    @SneakyThrows
    public ExcelToBeans(InputStream excelInputStream) {
        this.workbook = WorkbookFactory.create(excelInputStream);
        this.shouldBeClosedByMe = true;
    }

    public ExcelToBeans(Workbook workbook) {
        this.workbook = workbook;
        this.shouldBeClosedByMe = false;
    }

    @SuppressWarnings("unchecked")
    public  List convert(Class beanClass) {
        val converter = new ExcelSheetToBeans(workbook, beanClass);
        return converter.convert();
    }

    @SuppressWarnings("unchecked")
    public void writeError(Class beanClass, List rowRefs) {
        val sheet = PoiUtil.findSheet(workbook, beanClass);
        val sheetToBeans = new ExcelSheetToBeans(workbook, beanClass);
        int lastCellNum = getLastCellNum(sheet, sheetToBeans, rowRefs);
        if (lastCellNum <= 0) return;

        val redCellStyle = createRedCellStyle();

        rowRefs.stream().filter(x -> StringUtils.isNotEmpty(x.error())).forEach(x -> {
            val cell = sheet.getRow(x.getRowNum()).createCell(lastCellNum);
            cell.setCellStyle(redCellStyle);
            cell.setCellValue(x.error());
        });
    }

    public int getLastCellNum(Sheet sheet, ExcelSheetToBeans sheetToBeans,
                              List rowRefs) {
        if (sheetToBeans.isHasTitle()) {
            return sheet.getRow(sheetToBeans.findTitleRowNum()).getLastCellNum();
        }

        return rowRefs.isEmpty() ? 0 : sheet.getRow(rowRefs.get(0).getRowNum()).getLastCellNum();
    }

    private CellStyle createRedCellStyle() {
        val cellStyle = workbook.createCellStyle();
        val font = workbook.createFont();
        font.setColor(IndexedColors.RED.getIndex());
        cellStyle.setFont(font);

        return cellStyle;
    }

    public void removeOkRows(Class beanClass, List rowRefs) {
        val sheet = PoiUtil.findSheet(workbook, beanClass);
        val rowsRemoved = new AtomicInteger();

        rowRefs.stream().filter(x -> StringUtils.isEmpty(x.error()))
                .forEach(x -> PoiUtil.removeRow(sheet, x.getRowNum() - rowsRemoved.getAndIncrement()));
    }

    @Override public void close() throws IOException {
        if (shouldBeClosedByMe) workbook.close();
    }
}