hsim.checkpoint.util.excel.PoiWorkSheet Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of check-point Show documentation
Http request auto validation library for spring framework
There is a newer version: 0.1.2
package hsim.checkpoint.util.excel;

import hsim.checkpoint.util.ValidationObjUtil;
import lombok.Getter;
import org.apache.poi.hssf.usermodel.HSSFSheet;
import org.apache.poi.ss.usermodel.Cell;
import org.apache.poi.ss.usermodel.CellType;
import org.apache.poi.ss.usermodel.Row;

import java.util.ArrayList;
import java.util.Date;
import java.util.List;

/**
 * The type Poi work sheet.
 */
@Getter
public class PoiWorkSheet {

    private static final double DEFAULT_WIDTH = 1;

    private HSSFSheet sheet;
    private PoiCellStyle style;

    /**
     * Instantiates a new Poi work sheet.
     *
     * @param workBook the work book
     */
    public PoiWorkSheet(PoiWorkBook workBook) {
        this.init(workBook, null);
    }

    /**
     * Instantiates a new Poi work sheet.
     *
     * @param workBook  the work book
     * @param sheetName the sheet name
     */
    public PoiWorkSheet(PoiWorkBook workBook, String sheetName) {
        this.init(workBook, sheetName);
    }

    /**
     * Next row row.
     *
     * @param cnt the cnt
     * @return the row
     */
    public Row nextRow(int cnt) {
        Row lastrow = null;
        for (int i = 0; i < cnt; i++) {
            lastrow = this.nextRow();
        }
        return lastrow;
    }

    /**
     * Next row row.
     *
     * @return the row
     */
    public Row nextRow() {
        return this.sheet.createRow(this.sheet.getLastRowNum() + 1);
    }

    /**
     * Gets last row.
     *
     * @return the last row
     */
    public Row getLastRow() {
        return this.sheet.getRow(this.sheet.getLastRowNum());
    }

    private String checkSheetName(String sheetName){
        return sheetName.replaceAll("\\*", "");
    }

    private void init(PoiWorkBook workBook, String sheetName) {

        this.style = new PoiCellStyle(workBook);

        if (sheetName != null) {
            this.sheet = workBook.getWorkBook().createSheet(this.checkSheetName(sheetName));
        } else {
            this.sheet = workBook.getWorkBook().createSheet();
        }

        this.sheet.createRow(0);

    }

    private int getCellCnt() {
        int cellCnt = this.getLastRow().getLastCellNum();
        return cellCnt < 0 ? 0 : cellCnt;
    }

    /**
     * Create title cells list.
     *
     * @param strs the strs
     * @return the list
     */
    public List createTitleCells(String... strs) {
        List cells = new ArrayList<>();

        for (String s : strs) {
            Cell cell = this.createTitleCell(s, DEFAULT_WIDTH);
            cells.add(cell);
        }
        return cells;
    }

    /**
     * Create title cells.
     *
     * @param width the width
     * @param strs  the strs
     */
    public void createTitleCells(double width, String... strs) {
        for (String s : strs) {
            this.createTitleCell(s, width);
        }
    }

    /**
     * Create title cell cell.
     *
     * @param str   the str
     * @param width the width
     * @return the cell
     */
    public Cell createTitleCell(String str, double width) {

        int cellCnt = this.getCellCnt();

        Cell cell = this.getLastRow().createCell(cellCnt);
        cell.setCellValue(str);
        cell.setCellType(CellType.STRING);
        cell.setCellStyle(this.style.getStringCs());

        sheet.setColumnWidth(cellCnt, (int) (sheet.getColumnWidth(cellCnt) * width));

        return cell;
    }


    /**
     * Create value cells.
     *
     * @param values the values
     */
    public void createValueCells(Object... values) {
        for (Object value : values) {
            if (value == null) {
                this.createCell("");
                continue;
            }

            if (value instanceof String) {
                this.createCell((String) value);
            } else if (value instanceof Date) {
                this.createCell((Date) value);
            } else if (ValidationObjUtil.isIntType(value.getClass())) {
                long longValue = Long.valueOf(value + "");
                this.createCell(longValue);
            } else if (ValidationObjUtil.isDoubleType(value.getClass())) {
                double doubleValue = Double.valueOf(value + "");
                this.createCell(doubleValue);
            } else {
                this.createCell(value);
            }
        }
    }

    private Cell getNextCell(CellType cellType) {
        int cellCnt = this.getCellCnt();
        Cell cell = this.getLastRow().createCell(cellCnt);
        cell.setCellType(cellType);
        cell.setCellStyle(this.style.getNumberCs());
        return cell;
    }

    /**
     * Create cell cell.
     *
     * @param value the value
     * @return the cell
     */
    public Cell createCell(double value) {

        Cell cell = this.getNextCell(CellType.NUMERIC);
        cell.setCellValue(value);
        cell.setCellStyle(this.style.getNumberCs());
        return cell;
    }

    /**
     * Create cell cell.
     *
     * @param value the value
     * @return the cell
     */
    public Cell createCell(Long value) {
        Cell cell = this.getNextCell(CellType.NUMERIC);
        cell.setCellValue(value);
        cell.setCellStyle(this.style.getNumberCs());
        return cell;
    }

    /**
     * Create cell cell.
     *
     * @param value the value
     * @return the cell
     */
    public Cell createCell(long value) {
        Cell cell = this.getNextCell(CellType.NUMERIC);
        cell.setCellValue(value);
        cell.setCellStyle(this.style.getNumberCs());
        return cell;
    }


    /**
     * Create cell cell.
     *
     * @param obj the obj
     * @return the cell
     */
    public Cell createCell(Object obj) {

        Cell cell = this.getNextCell(CellType.STRING);
        cell.setCellValue(String.valueOf(obj));
        cell.setCellStyle(this.style.getStringCs());
        return cell;
    }

    /**
     * Create cell cell.
     *
     * @param str the str
     * @return the cell
     */
    public Cell createCell(String str) {

        Cell cell = this.getNextCell(CellType.STRING);
        cell.setCellValue(str);
        cell.setCellStyle(this.style.getStringCs());
        return cell;
    }

    /**
     * Create cell cell.
     *
     * @param date the date
     * @return the cell
     */
    public Cell createCell(Date date) {

        Cell cell = this.getNextCell(CellType.STRING);
        if (date != null) {
            cell.setCellValue(date);
        }
        cell.setCellStyle(this.style.getDateCs());
        return cell;
    }


}