PoiSingleExcelView.java 2.6 KB
package com.bckefu.excel.view;

import com.bckefu.excel.ExcelExportUtil;
import com.bckefu.excel.entity.ExportParams;
import com.bckefu.excel.entity.vo.ExcelConstants;
import com.bckefu.excel.export.base.ExcelExportServer;
import org.apache.poi.hssf.usermodel.HSSFWorkbook;
import org.apache.poi.ss.usermodel.Workbook;

import javax.servlet.ServletOutputStream;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.net.URLEncoder;
import java.util.Collection;
import java.util.List;
import java.util.Map;

/**
 * @author : caoliang
 * @date : 2017/11/6:下午2:01
 */
public class PoiSingleExcelView extends AbstractExcelView {

    @Override
    void renderMergedOutputModel(String fileName, Map<String, Object> model, HttpServletRequest request, HttpServletResponse response) throws Exception {
        Workbook workbook = null;
        if (model.containsKey(ExcelConstants.MAP_LIST)) {
            List<Map<String, Object>> list = (List<Map<String, Object>>) model
                    .get(ExcelConstants.MAP_LIST);
            if (list.size() == 0) {
                throw new RuntimeException("MAP_LIST IS NULL");
            }
            workbook = ExcelExportUtil.exportExcel(
                    (ExportParams) list.get(0).get(ExcelConstants.PARAMS), (Class<?>) list.get(0)
                            .get(ExcelConstants.CLASS),
                    (Collection<?>) list.get(0).get(ExcelConstants.DATA_LIST));
            for (int i = 1; i < list.size(); i++) {
                new ExcelExportServer().createSheet(workbook,
                        (ExportParams) list.get(i).get(ExcelConstants.PARAMS), (Class<?>) list
                                .get(i).get(ExcelConstants.CLASS),
                        (Collection<?>) list.get(i).get(ExcelConstants.DATA_LIST));
            }
        } else {
            workbook = ExcelExportUtil.exportExcel(
                    (ExportParams) model.get(ExcelConstants.PARAMS),
                    (Class<?>) model.get(ExcelConstants.CLASS),
                    (Collection<?>) model.get(ExcelConstants.DATA_LIST));
        }

        if (workbook instanceof HSSFWorkbook) {
            fileName += HSSF;
        } else {
            fileName += XSSF;
        }
        if (isIE(request)) {
            fileName = URLEncoder.encode(fileName, "UTF8");
        } else {
            fileName = new String(fileName.getBytes("UTF-8"), "ISO-8859-1");
        }
        response.setHeader("content-disposition", "attachment;filename=" + fileName);
        ServletOutputStream out = response.getOutputStream();
        workbook.write(out);
        out.flush();
    }
}