• Home
• io.github.creekmoon
• excel-utils-boot-starter
• 1.2.10
• source code
• SheetWriter.java

×

Project download

Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $

You can buy this project and download/modify it how often you want.

cn.creekmoon.excelUtils.core.SheetWriter Maven / Gradle / Ivy

package cn.creekmoon.excelUtils.core;

import cn.creekmoon.excelUtils.core.SheetWriterContext.Title;
import cn.hutool.core.collection.ListUtil;
import cn.hutool.core.util.StrUtil;
import cn.hutool.poi.excel.BigExcelWriter;
import cn.hutool.poi.excel.style.StyleUtil;
import lombok.AllArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.apache.poi.ss.usermodel.CellStyle;
import org.apache.poi.xssf.usermodel.XSSFCellStyle;

import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.util.*;
import java.util.function.Consumer;
import java.util.function.Function;
import java.util.function.Predicate;
import java.util.stream.Collectors;

@Slf4j
public class SheetWriter {

    public SheetWriterContext sheetWriterContext;

    protected ExcelExport parent;


    /**
     * 添加标题
     */
    public SheetWriter addTitle(String titleName, Function valueFunction) {
        sheetWriterContext.titles.add(Title.of(titleName, valueFunction));
        return this;
    }

    /**
     * 获取当前的表头数量
     */
    public int countTitles() {
        return sheetWriterContext.titles.size();
    }

    /**
     * 为当前列设置一个样式
     *
     * @param condition        样式触发的条件
     * @param styleInitializer 样式初始化器
     * @return
     */
    public SheetWriter setDataStyle(Predicate condition, Consumer styleInitializer) {
        return setDataStyle(sheetWriterContext.titles.size() - 1, condition, styleInitializer);
    }

    /**
     * 添加excel样式映射,在写入的时候会读取这个映射
     *
     * @param colIndex         对应的列号
     * @param condition        样式的触发条件
     * @param styleInitializer 样式的初始化内容
     * @return
     */
    public SheetWriter setDataStyle(int colIndex, Predicate condition, Consumer styleInitializer) {
        /*初始化样式*/
//        CellStyle newCellStyle = parent.getBigExcelWriter().createCellStyle();
        XSSFCellStyle newCellStyle = (XSSFCellStyle) StyleUtil.createDefaultCellStyle(parent.getBigExcelWriter().getWorkbook());
        styleInitializer.accept(newCellStyle);
        ConditionStyle conditionStyle = new ConditionStyle(condition, newCellStyle);

        /*保存映射结果*/
        if (!sheetWriterContext.colIndex2Styles.containsKey(colIndex)) {
            sheetWriterContext.colIndex2Styles.put(colIndex, new ArrayList<>());
        }
        sheetWriterContext.colIndex2Styles.get(colIndex).add(conditionStyle);
        return this;
    }


    /**
     * 为当前列的数据设置一个样式
     *
     * @param styleInitializer 样式初始化器
     * @return
     */
    public SheetWriter setDataStyle(Consumer styleInitializer) {
        return this.setDataStyle(x -> true, styleInitializer);
    }

    /**
     * 写入对象
     *
     * @param data
     * @return
     */
    public SheetWriter write(List data) {
        return write(data, ExcelExport.WriteStrategy.CONTINUE_ON_ERROR);
    }


    /**
     * 以map形式写入
     *
     * @param data key=标题 value=行内容
     * @return
     */
    protected SheetWriter writeByMap(List> data) {
        preWritingContextAdjust();
        parent.getBigExcelWriter().write(data);
        postWritingContextAdjust();
        return this;
    }

    /**
     * 以对象形式写入
     *
     * @param vos           数据集
     * @param writeStrategy 写入策略
     * @return
     */
    private SheetWriter write(List vos, ExcelExport.WriteStrategy writeStrategy) {
        preWritingContextAdjust();

        this.initTitles();
        List> rows =
                vos.stream()
                        .map(
                                vo -> {
                                    List