封装poi导出篇

其他 2018-11-25 16:54:25 阅读次数: 0

前言

先写的导入，以为导出会很简单，没想到导出的东西也不少，基于常用的几种样式和校验写了一个简单的导出，包括时间，数字，文字长度，下拉框，提示框校验，基础样式包括字体，字体颜色，背景颜色等功能，可以使用rgb，基本上满足了我的需求。

基础校验类

package cc.vvxtoys.vhelper.excelhelper.exporter;

import java.text.SimpleDateFormat;

import org.apache.poi.hssf.usermodel.DVConstraint;
import org.apache.poi.hssf.usermodel.HSSFDataValidation;
import org.apache.poi.hssf.usermodel.HSSFSheet;
import org.apache.poi.ss.usermodel.DataValidation;
import org.apache.poi.ss.usermodel.DataValidationConstraint;
import org.apache.poi.ss.usermodel.DataValidationConstraint.ValidationType;
import org.apache.poi.ss.usermodel.DateUtil;
import org.apache.poi.ss.usermodel.Sheet;
import org.apache.poi.ss.util.CellRangeAddressList;
import org.apache.poi.xssf.usermodel.XSSFDataValidation;
import org.apache.poi.xssf.usermodel.XSSFDataValidationConstraint;
import org.apache.poi.xssf.usermodel.XSSFDataValidationHelper;
import org.apache.poi.xssf.usermodel.XSSFSheet;

/**
 * @author vvxtoys
 * @date   2018年11月8日下午5:30:02
 * @Description: TODO
 * @version V1.0
 */
public class BaseValidation implements Validation{
    
    private int version;  //版本
    private Sheet sheet;
    private ValidationConfig config;
    
    
    public BaseValidation(ValidationConfig config){
        this.config = config;
    }
    @Override
    public DataValidation createPrompt(){
        if(version == 0){
            return xlsPrompt();
        }else{
            return xlsxPrompt();
        }
    }
    private HSSFDataValidation xlsPrompt(){
        CellRangeAddressList addressList = new CellRangeAddressList(config.getStartRow(), config.getEndRow(),
                                                                    config.getStartCol(), config.getEndCol());
        DVConstraint dv = DVConstraint.createCustomFormulaConstraint("BB1");
        HSSFDataValidation dataValidation = new HSSFDataValidation(addressList, dv);
        dataValidation.setSuppressDropDownArrow(false);
        dataValidation.createPromptBox("输入提示", config.getPromptText());
        dataValidation.setShowPromptBox(true);
        return dataValidation;
    }
    private XSSFDataValidation xlsxPrompt(){    
        XSSFDataValidationHelper help = new XSSFDataValidationHelper((XSSFSheet)sheet);
        CellRangeAddressList addressList = new CellRangeAddressList(config.getStartRow(), config.getEndRow(),
                config.getStartCol(), config.getEndCol());
        XSSFDataValidationConstraint dv = (XSSFDataValidationConstraint)help.createCustomConstraint("BB1");
        XSSFDataValidation validation = (XSSFDataValidation)help.createValidation(dv, addressList);
        validation.setSuppressDropDownArrow(false);
        validation.createPromptBox("输入提示",config.getPromptText());
        validation.setShowPromptBox(true);
        return validation;
    }
    @Override
    public DataValidation createDropDownBox(){
        if(version == 0){
            return xlsDropDownBox();
        }else{
            return xlsxDropDownBox();
        }
    }
    
    private HSSFDataValidation xlsDropDownBox(){
        CellRangeAddressList regions = new CellRangeAddressList(config.getStartRow(), config.getEndRow(),
                config.getStartCol(), config.getEndCol()); 
        DVConstraint constraint = DVConstraint.createExplicitListConstraint(config.getTextlist());      
        HSSFDataValidation datavalidation = new HSSFDataValidation(regions,constraint);
        datavalidation.setSuppressDropDownArrow(false);
        datavalidation.createPromptBox("输入提示", config.getPromptText());
        datavalidation.setShowPromptBox(true);
        return datavalidation;
    }
    private XSSFDataValidation xlsxDropDownBox(){
        XSSFDataValidationHelper help = new XSSFDataValidationHelper((XSSFSheet)sheet);
        XSSFDataValidationConstraint dvConstraint= (XSSFDataValidationConstraint) help.createExplicitListConstraint(config.getTextlist());
        CellRangeAddressList addressList = new CellRangeAddressList(config.getStartRow(), config.getEndRow(),
                config.getStartCol(), config.getEndCol());
        XSSFDataValidation validation = (XSSFDataValidation) help.createValidation(
                dvConstraint, addressList);
        validation.setSuppressDropDownArrow(true);
        validation.createPromptBox("输入提示", config.getPromptText());
        validation.setShowPromptBox(true);
        return validation;
    }
    @Override
    public DataValidation createValidateTime(){
        if(version == 0){
            return xlsValidateTime();
        }else{
            return xlsxValidateTime();
        }
    }
    
    private HSSFDataValidation xlsValidateTime(){
        CellRangeAddressList addressList = new CellRangeAddressList(config.getStartRow(), config.getEndRow(),
                config.getStartCol(), config.getEndCol());
        DVConstraint dvConstraint = DVConstraint.createDateConstraint(DVConstraint.OperatorType.BETWEEN, config.getStartDate(),
                config.getEndDate(), config.getPattern());
        HSSFDataValidation dataValidation = new HSSFDataValidation(addressList, dvConstraint);
        dataValidation.setSuppressDropDownArrow(false);
        dataValidation.createPromptBox("输入提示",config.getPromptText());
        dataValidation.createErrorBox("日期格式错误提示", config.getErrorText());
        dataValidation.setShowPromptBox(true);
        return dataValidation;
    }
    private XSSFDataValidation xlsxValidateTime(){
        XSSFDataValidation validation = null;
        try{
            SimpleDateFormat sdf = new SimpleDateFormat(config.getPattern());
            XSSFDataValidationHelper help = new XSSFDataValidationHelper((XSSFSheet)sheet);
            CellRangeAddressList addressList = new CellRangeAddressList(config.getStartRow(), config.getEndRow(),
                    config.getStartCol(), config.getEndCol());
            XSSFDataValidationConstraint dvConstraint=(XSSFDataValidationConstraint)
                    help.createDateConstraint(DataValidationConstraint.OperatorType.BETWEEN, 
                            ""+DateUtil.getExcelDate(sdf.parse(config.getStartDate())), 
                            ""+DateUtil.getExcelDate(sdf.parse(config.getEndDate())), "");

            validation = (XSSFDataValidation) help.createValidation(
                    dvConstraint, addressList);

            validation.createPromptBox("输入提示",config.getPromptText());
            validation.createErrorBox("日期格式错误提示",config.getErrorText());
            validation.setShowPromptBox(true);
            validation.setShowErrorBox(true);
            validation.setErrorStyle(0);
            sheet.addValidationData(validation);
        }catch(Exception e){
            e.printStackTrace();
        }
        return validation;
    }
    @Override
    public DataValidation createTextLength(){
        if(version == 0){
            return xlsTextLength();
        }else{
            return xlsxTextLength();
        }
    }
    private HSSFDataValidation xlsTextLength(){
        DVConstraint constraint = DVConstraint.createNumericConstraint(  
                DVConstraint.ValidationType.TEXT_LENGTH,  
                DVConstraint.OperatorType.BETWEEN, String.valueOf(config.getMinLen()), String.valueOf(config.getMaxLen()));  
        CellRangeAddressList addressList = new CellRangeAddressList(config.getStartRow(), config.getEndRow(),
                config.getStartCol(), config.getEndCol());
        HSSFDataValidation dataValidation = new HSSFDataValidation(addressList, constraint);
        dataValidation.setSuppressDropDownArrow(false);
        dataValidation.createPromptBox("输入提示", config.getPromptText());
        // 设置输入错误提示信息
        dataValidation.createErrorBox("格式错误",config.getErrorText());
        dataValidation.setShowPromptBox(true);
        return dataValidation;
    }
    private XSSFDataValidation xlsxTextLength(){
        XSSFDataValidationHelper help = new XSSFDataValidationHelper((XSSFSheet)sheet);
        CellRangeAddressList addressList = new CellRangeAddressList(config.getStartRow(), config.getEndRow(),
                config.getStartCol(), config.getEndCol());
        XSSFDataValidationConstraint dvConstraint=(XSSFDataValidationConstraint)
        help.createNumericConstraint(ValidationType.TEXT_LENGTH, DataValidationConstraint.OperatorType.BETWEEN, String.valueOf(config.getMinLen()), String.valueOf(config.getMaxLen()));
        XSSFDataValidation validation = (XSSFDataValidation) help.createValidation(
                dvConstraint, addressList);
        validation.createPromptBox("输入提示", config.getPromptText());
        validation.createErrorBox("格式错误提示", config.getErrorText());
        validation.setShowPromptBox(true);
        validation.setShowErrorBox(true);
        return validation;
    }
    @Override
    public DataValidation createValidateNumber(){
        if(version == 0){
            return xlsValidateNumber();
        }else{
            return xlsxValidateNumber();
        }
    }
    
    private HSSFDataValidation xlsValidateNumber(){
        DVConstraint constraint = DVConstraint.createNumericConstraint(  
                DVConstraint.ValidationType.DECIMAL,  
                DVConstraint.OperatorType.BETWEEN, String.valueOf(config.getMin()), String.valueOf(config.getMax()));  
        CellRangeAddressList addressList = new CellRangeAddressList(config.getStartRow(), config.getEndRow(),
                config.getStartCol(), config.getEndCol());
        HSSFDataValidation dataValidation = new HSSFDataValidation(addressList, constraint);
        dataValidation.createPromptBox("输入提示", config.getPromptText());
        dataValidation.createErrorBox("格式错误", config.getErrorText());
        dataValidation.setShowPromptBox(true);
        dataValidation.setShowErrorBox(true);
        return dataValidation;
    }   
    private XSSFDataValidation xlsxValidateNumber(){
        XSSFDataValidationHelper help = new XSSFDataValidationHelper((XSSFSheet)sheet);
        CellRangeAddressList addressList = new CellRangeAddressList(config.getStartRow(), config.getEndRow(),
                config.getStartCol(), config.getEndCol());
        XSSFDataValidationConstraint dvConstraint=(XSSFDataValidationConstraint)
        help.createNumericConstraint(ValidationType.DECIMAL, DataValidationConstraint.OperatorType.BETWEEN, String.valueOf(config.getMin()), String.valueOf(config.getMax()));
        XSSFDataValidation validation = (XSSFDataValidation) help.createValidation(
                dvConstraint, addressList);
        validation.createPromptBox("输入提示", config.getPromptText());
        validation.createErrorBox("格式错误", config.getErrorText());
        validation.setShowPromptBox(true);
        validation.setShowErrorBox(true);
        return validation;
    }
    
    public void generatorValidation(){
        if(version == 0 && config.getEndRow() == 0){
            config.setEndRow(65535);
        }else if(version == 1 && config.getEndRow() == 0){
            config.setEndRow(1048575);
        }
        if(config.getCustomize()){
            addValidation(config.getDataValidation());
        }else{
            switch (config.getEv()) {
            case PROMPT:
                addValidation(createPrompt());
                break;
            case DROPDOWN:
                addValidation(createDropDownBox());
                break;
            case DATE:
                addValidation(createValidateTime());
                break;
            case LENGTH:
                addValidation(createTextLength());
                break;
            case NUMBER:
                addValidation(createValidateNumber());
                break;
            default:
                break;
            }
        }
    }
    public void setSheet(Sheet sheet){
        if(sheet instanceof HSSFSheet){
            this.version = 0;
            this.sheet = (HSSFSheet)sheet;
        }else if(sheet instanceof XSSFSheet){
            this.version = 1;
            this.sheet = (XSSFSheet)sheet;
        }
    }
    
    public void addValidation(DataValidation dataValidation){
        sheet.addValidationData(dataValidation);
    }
}

基础样式类

package cc.vvxtoys.vhelper.excelhelper.exporter;

import org.apache.poi.hssf.usermodel.HSSFCellStyle;
import org.apache.poi.hssf.usermodel.HSSFDataFormat;
import org.apache.poi.hssf.usermodel.HSSFFont;
import org.apache.poi.hssf.usermodel.HSSFPalette;
import org.apache.poi.hssf.usermodel.HSSFWorkbook;
import org.apache.poi.hssf.util.HSSFColor.HSSFColorPredefined;
import org.apache.poi.ss.usermodel.BorderStyle;
import org.apache.poi.ss.usermodel.CellStyle;
import org.apache.poi.ss.usermodel.FillPatternType;
import org.apache.poi.ss.usermodel.HorizontalAlignment;
import org.apache.poi.ss.usermodel.Workbook;
import org.apache.poi.xssf.usermodel.DefaultIndexedColorMap;
import org.apache.poi.xssf.usermodel.XSSFCellStyle;
import org.apache.poi.xssf.usermodel.XSSFColor;
import org.apache.poi.xssf.usermodel.XSSFDataFormat;
import org.apache.poi.xssf.usermodel.XSSFFont;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;

public class BaseStyle {
    
    private int version;
    private Workbook workbook;
    private StyleConfig sconfig;
    
    
    public BaseStyle(Workbook workbook,StyleConfig config) {
        if(workbook instanceof HSSFWorkbook){
            this.version = 0;
            this.workbook = workbook;
            this.sconfig = config;
        }else if(workbook instanceof XSSFWorkbook){
            version = 1;
            this.workbook = workbook;
            this.sconfig = config;
        }
    }
    
    /**
     * @author vvxtoys
     * @date   2018年11月13日下午3:38:58
     * @Description: 调用优先级 rgb index
     * @return
     */
    public CellStyle generatorStyle(){
        if(sconfig.isOuter()){
            return sconfig.getCellStyle();
        }
        if(version == 0){
            HSSFCellStyle style = ((HSSFWorkbook)workbook).createCellStyle();
            Object bgColor = sconfig.getBackground();
            if(bgColor instanceof byte[]){
                byte[] bs = (byte[])bgColor;
                style.setFillForegroundColor(HSSFColorPredefined.WHITE.getIndex());
                HSSFPalette palette = ((HSSFWorkbook) workbook).getCustomPalette();
                palette.setColorAtIndex(HSSFColorPredefined.WHITE.getIndex(), bs[0], bs[1], bs[2]);
            }else{
                short bgIndex = (short)bgColor;
                if(bgIndex == -1){
                    style.setFillForegroundColor(HSSFColorPredefined.GREY_40_PERCENT.getIndex());
                }else{
                    style.setFillForegroundColor(bgIndex);
                }
            }
            style.setFillPattern(FillPatternType.SOLID_FOREGROUND);
            style.setBorderBottom(BorderStyle.THIN);
            style.setBorderLeft(BorderStyle.THIN);
            style.setBorderRight(BorderStyle.THIN);
            style.setBorderTop(BorderStyle.THIN);
            style.setAlignment(HorizontalAlignment.CENTER);         
            HSSFDataFormat format = ((HSSFWorkbook)workbook).createDataFormat();
            style.setDataFormat(format.getFormat("@"));
            style.setWrapText(true);
            
            HSSFFont font = ((HSSFWorkbook)workbook).createFont();
            Object fColor = sconfig.getFontColor();
            if(fColor instanceof byte[]){
                byte[] frgb = (byte[])fColor;
                font.setColor(HSSFColorPredefined.BLACK.getIndex());
                style.setFillForegroundColor(HSSFColorPredefined.BLACK.getIndex());
                HSSFPalette palette1 = ((HSSFWorkbook) workbook).getCustomPalette();
                palette1.setColorAtIndex(HSSFColorPredefined.BLACK.getIndex(), frgb[0], frgb[1], frgb[2]);
            }else{
                short fIndex = (short)fColor;
                if(fIndex == -1){
                    font.setColor(HSSFColorPredefined.BLACK.getIndex());
                }else{
                    font.setColor(fIndex);
                }
            }
            font.setFontHeightInPoints(sconfig.getHeight());
            font.setFontName(sconfig.getFontName());
            font.setBold(sconfig.getBlob());
            
            style.setFont(font);
            
            return style;
        }else{
            XSSFCellStyle style = ((XSSFWorkbook)workbook).createCellStyle();
            Object bgColor = sconfig.getBackground();
            if(bgColor instanceof byte[]){
                byte[] rgb = (byte[])bgColor;
                XSSFColor color = new XSSFColor(rgb, new DefaultIndexedColorMap());
                style.setFillForegroundColor(color);
            }else{
                short bgIndex = (short)bgColor;
                if(bgIndex == -1){
                    style.setFillForegroundColor(HSSFColorPredefined.GREY_40_PERCENT.getIndex());
                }else{
                    style.setFillForegroundColor(bgIndex);
                }
            }
            
            style.setFillPattern(FillPatternType.SOLID_FOREGROUND);
            style.setBorderBottom(BorderStyle.THIN);
            style.setBorderLeft(BorderStyle.THIN);
            style.setBorderRight(BorderStyle.THIN);
            style.setBorderTop(BorderStyle.THIN);
            style.setAlignment(HorizontalAlignment.CENTER);
            style.setWrapText(true);
            XSSFDataFormat format = ((XSSFWorkbook)workbook).createDataFormat();
            style.setDataFormat(format.getFormat("@"));
            
            XSSFFont font = ((XSSFWorkbook)workbook).createFont();
            Object fColor = sconfig.getFontColor();
            if(fColor instanceof byte[]){
                byte[] frgb = (byte[])fColor;
                XSSFColor color1 = new XSSFColor(frgb,new DefaultIndexedColorMap());
                style.setFillForegroundColor(color1);
            }else{
                short fIndex = (short)fColor;
                if(fIndex == -1){
                    font.setColor(HSSFColorPredefined.BLACK.getIndex());
                }else{
                    font.setColor(fIndex);
                }
            }
            font.setFontHeightInPoints(sconfig.getHeight());
            font.setFontName(sconfig.getFontName());
            font.setBold(sconfig.getBlob());            
            style.setFont(font);
            
            return style;
        }
    }
    
    
    public CellStyle getDefaultStyle(){
        if(version == 0){
            HSSFCellStyle style = ((HSSFWorkbook)workbook).createCellStyle();
            style.setBorderBottom(BorderStyle.THIN);
            style.setBorderLeft(BorderStyle.THIN);
            style.setBorderRight(BorderStyle.THIN);
            style.setBorderTop(BorderStyle.THIN);
            style.setAlignment(HorizontalAlignment.CENTER);         
            HSSFDataFormat format = ((HSSFWorkbook)workbook).createDataFormat();
            style.setDataFormat(format.getFormat("@"));
            style.setWrapText(true);
            
            HSSFFont font = ((HSSFWorkbook)workbook).createFont();          
            font.setFontHeightInPoints(sconfig.getHeight());
            font.setFontName(sconfig.getFontName());
            font.setBold(sconfig.getBlob());
            
            style.setFont(font);
            
            return style;
        }else{
            XSSFCellStyle style = ((XSSFWorkbook)workbook).createCellStyle();
            style.setBorderBottom(BorderStyle.THIN);
            style.setBorderLeft(BorderStyle.THIN);
            style.setBorderRight(BorderStyle.THIN);
            style.setBorderTop(BorderStyle.THIN);
            style.setAlignment(HorizontalAlignment.CENTER);
            style.setWrapText(true);
            XSSFDataFormat format = ((XSSFWorkbook)workbook).createDataFormat();
            style.setDataFormat(format.getFormat("@"));
            
            XSSFFont font = ((XSSFWorkbook)workbook).createFont();          
            font.setFontHeightInPoints(sconfig.getHeight());
            font.setFontName(sconfig.getFontName());
            font.setBold(sconfig.getBlob());            
            style.setFont(font);
            
            return style;
        }
    }
}

结语

导出代码量比较大，贴出来两个比较核心的类，完整项目地址为：https://gitee.com/vvxtoys/vhelper

猜你喜欢

转载自www.cnblogs.com/vvxtoys/p/importhelper.html
今日推荐
Apache Doris 2.0.10 版本正式发布！
开源日报 | 大模型开战；大模型独角兽被曝卖身；周鸿祎建议谷歌开源所有产品；最大开源AI社区提供1000万美元共享GPU
开源日报 | Chrome内置Gemini的意义不在于Gemini；中国AI追随之路的五大误区；ECharts创始人“下海”养鱼；谷歌I/O开发者大会什么都有，只是没有惊喜
微软回应中国区AI团队“打包赴美”传闻
基于大语言模型的开源知识库问答系统 MaxKB GitHub Star 数量突破 5,000 个！
美国拟限制 AI 大模型出口中国和俄罗斯
苹果将与 OpenAI 达成协议，将 ChatGPT 应用于 iPhone
周排行
阿里云短信服务平台注册
Windows下的字符串处理(1)
sqoop: mysql导入数据到hdfs, hive, hbase
commons.lang中常用的工具类
离线安装PostgreSQL11.6
使用PyTorch简单实现卷积神经网络模型
一文彻底搞定谱聚类
一道面试题引发的血案
One Chat for Mac(聊天工具)
TCP/IP的底层队列是如何实现的？
每日归档
更多
2024-05-17(34)
2024-05-16(6)
2024-05-15(24)
2024-05-14(0)
2024-05-13(18)
2024-05-12(0)
2024-05-11(38)
2024-05-10(38)
2024-05-09(35)
2024-05-08(42)

代码天地 © 2018 codetd.com

境外服务器   最新电视剧2022