EXCEL 导出

package com.pigmis.common.utils;

import com.google.common.base.Strings;
import com.pigmis.utils.ReflexUtil;
import lombok.Data;
import org.apache.poi.hssf.usermodel.*;
import org.apache.poi.hssf.util.HSSFColor;
import org.apache.poi.ss.util.CellRangeAddress;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.OutputStream;
import java.math.BigDecimal;
import java.text.ParseException;
import java.text.SimpleDateFormat;
import java.util.Date;
import java.util.List;
import java.util.Locale;

@Data
public class ExcelExportUtil<T> {
    //表头
    private String title;
    //各个列的表头
    private List<String> heardList;
    //各个列的元素key值
    private List<String> heardKey;
    //各个列的对其方式
    private List<String> align;
    //需要填充的数据信息
    private List<T> data;
    //字体大小
    private int fontSize = 14;
    //行高
    private int rowHeight = 30;
    //列宽
    private int columWidth = 200;
    //工作表
    private String sheetName = "sheet1";

    /**
     * 开始导出数据信息
     *
     * @return
     */
    public String exportExport(HttpServletRequest request, HttpServletResponse response) throws IOException {
        //检查参数配置信息
        checkConfig();
        //创建工作簿
        HSSFWorkbook wb = new HSSFWorkbook();
        //创建工作表
        HSSFSheet wbSheet = wb.createSheet(this.sheetName);
        //设置默认行宽
        wbSheet.setDefaultColumnWidth(20);

        // 标题样式（加粗，垂直居中）
        HSSFCellStyle cellStyle = wb.createCellStyle();
        cellStyle.setAlignment(HSSFCellStyle.ALIGN_CENTER);//水平居中
        HSSFFont fontStyle = wb.createFont();
        fontStyle.setBoldweight(HSSFFont.BOLDWEIGHT_BOLD);
        fontStyle.setFontHeightInPoints((short)16);  //设置标题字体大小
        cellStyle.setFont(fontStyle);

        //在第0行创建rows  (表标题)
        HSSFRow title = wbSheet.createRow((int) 0);
        title.setHeightInPoints(30);//行高
        HSSFCell cellValue = title.createCell(0);
        cellValue.setCellValue(this.title);
        cellValue.setCellStyle(cellStyle);
        wbSheet.addMergedRegion(new CellRangeAddress(0,0,0,(this.heardList.size()-1)));
        //设置表头样式，表头居中
        HSSFCellStyle style = wb.createCellStyle();
        //设置单元格样式
        style.setAlignment(HSSFCellStyle.ALIGN_CENTER);
        style.setVerticalAlignment(HSSFCellStyle.VERTICAL_CENTER);
        //设置字体
        HSSFFont font = wb.createFont();
        font.setFontHeightInPoints((short) this.fontSize);
        style.setFont(font);
        //在第1行创建rows
        HSSFRow row = wbSheet.createRow((int) 1);
        //设置列头元素
        HSSFCell cellHead = null;
        for (int i = 0; i < heardList.size(); i++) {
            cellHead = row.createCell(i);
            cellHead.setCellValue(heardList.get(i));
            cellHead.setCellStyle(style);
        }

        //设置每格数据的样式 （字体红色）
        HSSFCellStyle cellParamStyle = wb.createCellStyle();
        HSSFFont ParamFontStyle = wb.createFont();
        cellParamStyle.setAlignment(HSSFCellStyle.ALIGN_CENTER);
        cellParamStyle.setVerticalAlignment(HSSFCellStyle.VERTICAL_CENTER);
        ParamFontStyle.setColor(HSSFColor.DARK_RED.index);   //设置字体颜色 (红色)
        ParamFontStyle.setFontHeightInPoints((short) this.fontSize);
        cellParamStyle.setFont(ParamFontStyle);
        //设置每格数据的样式2（字体蓝色）
        HSSFCellStyle cellParamStyle2 = wb.createCellStyle();
        cellParamStyle2.setAlignment(HSSFCellStyle.ALIGN_CENTER);
        cellParamStyle2.setVerticalAlignment(HSSFCellStyle.VERTICAL_CENTER);
        HSSFFont ParamFontStyle2 = wb.createFont();
        ParamFontStyle2.setColor(HSSFColor.BLUE.index);   //设置字体颜色 (蓝色)
        ParamFontStyle2.setFontHeightInPoints((short) this.fontSize);
        cellParamStyle2.setFont(ParamFontStyle2);
        //开始写入实体数据信息
        int a = 2;
        for (int i = 0; i < data.size(); i++) {
            HSSFRow roww = wbSheet.createRow((int) a);
            Object map = data.get(i);
            HSSFCell cell = null;
            for (int j = 0; j < heardKey.size(); j++) {
                cell = roww.createCell(j);
                //设置左右对齐
//                if (align.get(j).equals("left")){
//                    style.setAlignment(HSSFCellStyle.ALIGN_RIGHT);
//                }else if(align.get(j).equals("right")){
//                    style.setAlignment(HSSFCellStyle.ALIGN_LEFT);
//                }
                cell.setCellStyle(style);
                Object valueObject = ReflexUtil.getFieldValueByName(heardKey.get(j),map);
                String value = null;
                if (valueObject == null) {
                    valueObject = "";
                }
                if (valueObject instanceof String) {
                    //取出的数据是字符串直接赋值
                    value = (String) valueObject;
                } else if (valueObject instanceof Integer) {
                    //取出的数据是Integer
                    value = String.valueOf(((Integer) (valueObject)).floatValue());
                } else if (valueObject instanceof BigDecimal) {
                    //取出的数据是BigDecimal
                    value = String.valueOf(((BigDecimal) (valueObject)).floatValue());
                }else if (valueObject instanceof Date){
                    Date date = parse(valueObject.toString(), "EEE MMM dd HH:mm:ss zzz yyyy", Locale.US);
                    String time = format(date, "yyyy-MM-dd", Locale.CHINA);
                    value = time;
                }
                else {
                    value = valueObject.toString();
                }

                //设置单个单元格的字体颜色
//                if(heardKey[j].equals("ddNum") || heardKey[j].equals("sjNum")){
//                    if((Long)map.get("ddNum")!=null){
//                        if((Long)map.get("sjNum")==null){
//                            cell.setCellStyle(cellParamStyle);
//                        } else if((Long) map.get("ddNum") != (Long) map.get("sjNum")){
//                            if ((Long) map.get("ddNum") > (Long) map.get("sjNum")) {
//                                cell.setCellStyle(cellParamStyle);
//                            }
//                            if ((Long) map.get("ddNum") < (Long) map.get("sjNum")) {
//                                cell.setCellStyle(cellParamStyle2);
//                            }
//                        }else {
//                            cell.setCellStyle(style);
//                        }
//                    }
//                }
                cell.setCellValue(Strings.isNullOrEmpty(value) ? "" : value);
            }
            a++;
        }
        SimpleDateFormat df = new SimpleDateFormat("yyyyMMddHHmmss");
        String excelName = df.format(new Date());
        File file = new File(request.getSession().getServletContext().getRealPath("/") + "/FileUpload/download/"+excelName+".xls");
        String filePath ="/FileUpload/download/"+excelName+".xls";
        if(!file.exists()){
            //目标文件不存在，创建目标文件
            file.getParentFile().mkdirs();
            file.createNewFile();
        }
        FileOutputStream outputStream = new FileOutputStream(file);
        wb.write(outputStream);
        outputStream.close();
        return filePath;
//        //导出数据
//        try {
//            //设置Http响应头告诉浏览器下载这个附件
//            System.out.println(System.currentTimeMillis()+"fuck.xml");
//            response.setHeader("Content-Disposition", "attachment;Filename=" + System.currentTimeMillis() + ".xls");
//            response.setContentType("application/octet-stream");
           response.setContentType("application/vnd.ms-excel");
//            OutputStream outputStream = response.getOutputStream();
//            System.out.println(outputStream);
//            wb.write(outputStream);
//            outputStream.close();
//            return wb.getBytes();
//        } catch (Exception ex) {
//            ex.printStackTrace();
//            throw new IOException("导出Excel出现严重异常，异常信息：" + ex.getMessage());
//        }
    }

    /**
     * 检查数据配置问题
     *
     * @throws IOException 抛出数据异常类
     */
    protected void checkConfig() throws IOException {
        if (heardKey == null || heardList.size() == 0) {
            throw new IOException("列名数组不能为空或者为NULL");
        }

        if (fontSize < 0 || rowHeight < 0 || columWidth < 0) {
            throw new IOException("字体、宽度或者高度不能为负值");
        }

        if (Strings.isNullOrEmpty(sheetName)) {
            throw new IOException("工作表表名不能为NULL");
        }
    }

    public static void main(String[] args) {
        String str = "Mon Dec 31 00:00:00 CST 2012";
        Date date = parse(str, "EEE MMM dd HH:mm:ss zzz yyyy", Locale.US);
        System.out.printf("%tF %<tT%n", date);
    }

    public static Date parse(String str, String pattern, Locale locale) {
        if(str == null || pattern == null) {
            return null;
        }
        try {
            return new SimpleDateFormat(pattern, locale).parse(str);
        } catch (ParseException e) {
            e.printStackTrace();
        }
        return null;
    }

    public static String format(Date date, String pattern, Locale locale) {
        if(date == null || pattern == null) {
            return null;
        }
        return new SimpleDateFormat(pattern, locale).format(date);
    }
}