PoiSingleExcelView.java 1.55 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 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.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;
        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();
    }
}