【无标题】easyExcel+Apache Poi 的实现模板导出并加密

最新推荐文章于 2024-02-23 11:29:47 发布

double_h_t

最新推荐文章于 2024-02-23 11:29:47 发布

阅读量339

点赞数 1

文章标签： apache

本文链接：https://blog.csdn.net/double_h_t/article/details/132609959

版权

import org.apache.poi.poifs.crypt.EncryptionInfo;
import org.apache.poi.poifs.crypt.EncryptionMode;
import org.apache.poi.poifs.crypt.Encryptor;
import org.apache.poi.poifs.filesystem.POIFSFileSystem;
import org.apache.poi.ss.usermodel.Workbook;
import org.apache.poi.ss.usermodel.WorkbookFactory;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.*;
import java.util.List;
import java.util.Map;

public class ExcelUtil {

    public static void exportExcel(HttpServletRequest request, HttpServletResponse response,
                                   String name, List<?> list, Map<?, Object> map, String teName) throws Exception {

        // 使用EasyExcel将数据写入到一个临时文件中
        File tempFile = File.createTempFile("temp", ".xlsx");
        try (OutputStream tempOut = new FileOutputStream(tempFile)) {
            InputStream templateFileName = ExcelUtil.class.getClassLoader().getResourceAsStream(teName);
            ExcelWriter excelWriter = EasyExcel.write(tempOut).withTemplate(templateFileName).build();
            WriteSheet sheet = EasyExcel.writerSheet().build();
            FillConfig fillConfig = FillConfig.builder().forceNewRow(Boolean.TRUE).build();
            if (list != null && list.size() != 0)
                excelWriter.fill(list, fillConfig, sheet);
            if (map != null)
                excelWriter.fill(map, sheet);
            excelWriter.finish();
        }

        // 使用Apache POI读取临时文件并加密数据
        Workbook workbook = WorkbookFactory.create(tempFile);
        POIFSFileSystem fs = new POIFSFileSystem();
        EncryptionInfo info = new EncryptionInfo(EncryptionMode.agile);
        Encryptor enc = info.getEncryptor();
        enc.confirmPassword("123456"); // 设置密码

        try (OutputStream os = enc.getDataStream(fs)) {
            workbook.write(os);
        }

        // 设置响应头，以便下载
        String fileName = name + System.currentTimeMillis() + ".xlsx";
        DownLoadHelper.setDownloadHeader(response, request, fileName);

        // 将加密后的数据写入到客户端的输出流
        try (OutputStream out = response.getOutputStream()) {
            fs.writeFilesystem(out);
        }

        // 删除临时文件
        tempFile.delete();
    }
}







import com.google.common.net.HttpHeaders;


import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import java.io.UnsupportedEncodingException;

import java.net.URLEncoder;


public class DownLoadHelper {
    /**
     * 文件下载设置
     *
     * @param request
     * @param fileName
     * @return
     * @throws UnsupportedEncodingException
     */
    public static void setDownloadHeader(HttpServletResponse response, HttpServletRequest request, String fileName) throws UnsupportedEncodingException {
        response.setCharacterEncoding("utf-8");
        // 出现下载的对话框
        response.setContentType("application/vnd.openxmlformats-officedocument.spreadsheetml.sheet");
        //中文文件名乱码 重新编码文件名
        String userAgent = request.getHeader("USER-AGENT");
        if (userAgent.contains("MSIE") || userAgent.contains("MSEdge")) {
            fileName = URLEncoder.encode(fileName, "UTF-8");
            fileName = fileName.replace("+", " ");
        } else if (userAgent.contains("Firefox")) {
            // 火狐浏览器
            fileName = new String(fileName.getBytes(), "ISO8859-1");
        } else if (userAgent.contains("Chrome")) {
            // google浏览器
            fileName = URLEncoder.encode(fileName, "utf-8");
        } else {
            // 其它浏览器
            fileName = URLEncoder.encode(fileName, "utf-8");
        }
        response.setHeader(HttpHeaders.CONTENT_DISPOSITION, "attachment;filename=" + fileName);
    }


}