apache-poi-xssf 查询列表导出excel封装工具类

package yycg.util;

import java.io.FileOutputStream;
import java.lang.reflect.Method;
import java.util.ArrayList;
import java.util.List;

import org.apache.poi.ss.usermodel.Cell;
import org.apache.poi.ss.usermodel.Row;
import org.apache.poi.ss.usermodel.Sheet;
import org.apache.poi.xssf.streaming.SXSSFSheet;
import org.apache.poi.xssf.streaming.SXSSFWorkbook;
/**
 * 
***********************************版权声明******************************
*版权所有： 登风
*项目名称： XX系统
***********************************创建记录******************************
*创建者 ：登风       
*创建日期：2019年8月29日 下午 8：58:46
*主要功能： 网页列表导出excel工具类
*
*修改者 :登风
*修改日期： 2019年8月29日  下午8:58:46
************************************备注********************************
*
 */
public class ExcelExportSXXSSF {

    // 定义工作表
    private SXSSFWorkbook wb;

    /**
     * 定义工作表中的sheet
     */
    private Sheet sh;


    /**
     * 定义保存在内存中的数量,-1表示手动控制
     */
    private int flushRows;
    /** 导出文件行数 */
    private int rownum;
    /** 导出文件列数 */
    private int colnum;

    /** 导出文件的存放路径 */
    private String filePath;
    /** 下载导出文件的路径 */
    private String fileWebPath;
    /**文件名称前缀*/
    private String filePrefix;
    /**导出文件全路径*/
    private String fileAllPath;
    /** 导出文件列标题 */
    private List<String> fieldNames;
    /**导出文件每列代码，用于反射获取对象属性值*/
    private List<String> fieldCodes;

    private ExcelExportSXXSSF() {

    }

    /**
     * 开始导出方法
     * 
     * @param filePath
     *            导出文件存放物理路径
     * @param fileWebPath
     *            导出文件web下载路径
     * @param filePrefix
     *            导出文件名的前缀          
     * @param flushRows
     *            存放在内存的数据量
     * @param fieldNames
     *            导出文件列标题
     * @param fieldCodes
     *               导出数据对象的字段名称     
     * @param flushRows
     *             写磁盘控制参数
     * @return
     */
    public static ExcelExportSXXSSF start(String filePath, String fileWebPath,String filePrefix,
            List<String> fieldNames,List<String> fieldCodes, int flushRows) throws Exception {
        ExcelExportSXXSSF excelExportSXXSSF = new ExcelExportSXXSSF();
        excelExportSXXSSF.setFilePath(filePath);
        excelExportSXXSSF.setFileWebPath(fileWebPath);
        excelExportSXXSSF.setFilePrefix(filePrefix);
        excelExportSXXSSF.setFieldNames(fieldNames);
        excelExportSXXSSF.setFieldCodes(fieldCodes);
        excelExportSXXSSF.setWb(new SXSSFWorkbook(flushRows));//创建workbook
        excelExportSXXSSF.setSh(excelExportSXXSSF.getWb().createSheet());//创建sheet
        excelExportSXXSSF.writeTitles();
        return excelExportSXXSSF;
    }

    /**
     * 设置导入文件的标题
     * 开始生成导出excel的标题
     * @throws Exception
     */
    private void writeTitles() throws Exception {
        rownum = 0;//第0行
        colnum = fieldNames.size();//根据列标题得出列数
        Row row = sh.createRow(rownum);
        for (int cellnum = 0; cellnum < colnum; cellnum++) {
            Cell cell = row.createCell(cellnum);
            cell.setCellValue(fieldNames.get(cellnum));
        }
    }

    /**
     * 向导出文件写数据
     * 
     * @param datalist
     *            存放Object对象，仅支持单个自定义对象，不支持对象中嵌套自定义对象
     * @return
     */
    public void writeDatasByObject(List datalist) throws Exception {

        for (int j = 0; j < datalist.size(); j++) {
            rownum = rownum + 1;
            Row row = sh.createRow(rownum);
            for (int cellnum = 0; cellnum < fieldCodes.size(); cellnum++) {
                Object owner = datalist.get(j);
                Object value = invokeMethod(owner, fieldCodes.get(cellnum),
                        new Object[] {});
                Cell cell = row.createCell(cellnum);
                cell.setCellValue(value!=null?value.toString():"");
            }

        }

    }
    /**
     * 向导出文件写数据
     * 
     * @param datalist
     *            存放字符串数组
     * @return
     */
    public void writeDatasByString(List<String> datalist) throws Exception {
            rownum = rownum + 1;
            Row row = sh.createRow(rownum);
            int datalist_size = datalist.size();
            for (int cellnum = 0; cellnum < colnum; cellnum++) {
                Cell cell = row.createCell(cellnum);
                if(datalist_size>cellnum){
                    cell.setCellValue(datalist.get(cellnum));
                }else{
                    cell.setCellValue("");
                }
                
            }
    }

    /**
     * 手动刷新方法,如果flushRows为-1则需要使用此方法手动刷新内存
     * 
     * @param flushRows
     * @throws Exception
     */
    public void flush(int flushNum) throws Exception {
        ((SXSSFSheet) sh).flushRows(flushNum);
    }

    /**
     * 导出文件
     * 
     * @throws Exception
     */
    public String exportFile() throws Exception {
        String filename = filePrefix+"_"+MyUtil.getCurrentTimeStr() + ".xlsx";
        FileOutputStream out = new FileOutputStream(filePath + filename);
        wb.write(out);
        out.flush();
        out.close();
        setFileAllPath(fileWebPath + filename);
        return fileWebPath + filename;
    }

    /**
     * 反射方法，通过get方法获取对象属性
     * 
     * @param owner
     * @param fieldname
     * @param args
     * @return
     * @throws Exception
     */
    private Object invokeMethod(Object owner, String fieldname, Object[] args)
            throws Exception {

        String methodName = "get" + fieldname.substring(0, 1).toUpperCase()
                + fieldname.substring(1);
        Class ownerClass = owner.getClass();

        Class[] argsClass = new Class[args.length];

        for (int i = 0, j = args.length; i < j; i++) {
            argsClass[i] = args[i].getClass();
        }

        Method method = ownerClass.getMethod(methodName, argsClass);
        return method.invoke(owner, args);
    }

    public SXSSFWorkbook getWb() {
        return wb;
    }

    public void setWb(SXSSFWorkbook wb) {
        this.wb = wb;
    }

    public Sheet getSh() {
        return sh;
    }

    public void setSh(Sheet sh) {
        this.sh = sh;
    }


    public int getFlushRows() {
        return flushRows;
    }

    public void setFlushRows(int flushRows) {
        this.flushRows = flushRows;
    }

    public String getFilePath() {
        return filePath;
    }

    public void setFilePath(String filePath) {
        this.filePath = filePath;
    }

    public String getFileWebPath() {
        return fileWebPath;
    }

    public void setFileWebPath(String fileWebPath) {
        this.fileWebPath = fileWebPath;
    }

    public List<String> getFieldNames() {
        return fieldNames;
    }

    public void setFieldNames(List<String> fieldNames) {
        this.fieldNames = fieldNames;
    }

    public List<String> getFieldCodes() {
        return fieldCodes;
    }

    public void setFieldCodes(List<String> fieldCodes) {
        this.fieldCodes = fieldCodes;
    }

    public int getRownum() {
        return rownum;
    }

    public String getFilePrefix() {
        return filePrefix;
    }

    public void setFilePrefix(String filePrefix) {
        this.filePrefix = filePrefix;
    }

    public int getColnum() {
        return colnum;
    }

    public String getFileAllPath() {
        return fileAllPath;
    }

    public void setFileAllPath(String fileAllPath) {
        this.fileAllPath = fileAllPath;
    }
    
    //demo
    public static void main(String[] args) throws Exception {
        /**            导出文件存放物理路径
         * @param fileWebPath
         *            导出文件web下载路径
         * @param filePrefix
         *            导出文件名的前缀          
         * @param flushRows
         *            存放在内存的数据量
         * @param fieldNames
         *            导出文件列标题
         * @param fieldCodes
         *               导出数据对象的字段名称     
         * @param flushRows*/
        //导出文件存放的路径，并且是虚拟目录指向的路径
        String filePath = "I:\JavaItem\第14项目：药品集中采购系统视频教程\自己笔记\poiDemo\";
        //导出文件的前缀
        String filePrefix="ypxx";
        //-1表示关闭自动刷新，手动控制写磁盘的时机，其它数据表示多少数据在内存保存，超过的则写入磁盘
        int flushRows=100;
        
        //指导导出数据的title
        List<String> fieldNames=new ArrayList<String>();
        fieldNames.add("流水号");
        fieldNames.add("通用名");
        fieldNames.add("价格");
        
        //告诉导出类数据list中对象的属性，让ExcelExportSXXSSF通过反射获取对象的值
        List<String> fieldCodes=new ArrayList<String>();
        fieldCodes.add("bm");//药品流水号
        fieldCodes.add("mc");//通用名
        fieldCodes.add("price");//价格
        
        
        
        //注意：fieldCodes和fieldNames个数必须相同且属性和title顺序一一对应，这样title和内容才一一对应
        
        
        //开始导出，执行一些workbook及sheet等对象的初始创建
        ExcelExportSXXSSF excelExportSXXSSF = ExcelExportSXXSSF.start(filePath, "/upload/", filePrefix, fieldNames, fieldCodes, flushRows);
        
        //准备导出的数据，将数据存入list，且list中对象的字段名称必须是刚才传入ExcelExportSXXSSF的名称
        List<Ypxx> list = new ArrayList<Ypxx>();
        
        Ypxx ypxx1 = new Ypxx("001", "青霉素", 5);
        Ypxx ypxx2 = new Ypxx("002", "感冒胶囊", 2.5f);
        list.add(ypxx1);
        list.add(ypxx2);
        
        //执行导出
        excelExportSXXSSF.writeDatasByObject(list);
        //输出文件，返回下载文件的http地址
        String webpath = excelExportSXXSSF.exportFile();
        
        System.out.println(webpath);
        
        
    }

}