Excel导入工具类

最新推荐文章于 2024-05-15 16:59:24 发布

m0_49341808

最新推荐文章于 2024-05-15 16:59:24 发布

阅读量52

点赞数

文章标签： excel java

本文链接：https://blog.csdn.net/m0_49341808/article/details/133814696

版权

package model;

import lombok.AllArgsConstructor;
import lombok.Data;
import lombok.NoArgsConstructor;

import java.io.Serializable;
import java.util.List;

@Data
@AllArgsConstructor
@NoArgsConstructor
public class ExcelMC implements Serializable {
    private static final long serialVersionUID = 1L;

    private String sheetName;
    private List<String> ltTitle;
    private List<List<String>> ltBody;
}

package utils.excel;

import model.ExcelMC;
import org.apache.poi.xssf.usermodel.XSSFRow;
import org.apache.poi.xssf.usermodel.XSSFSheet;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;
import org.springframework.web.context.request.RequestContextHolder;
import org.springframework.web.context.request.ServletRequestAttributes;
import sun.misc.BASE64Encoder;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.net.URLEncoder;
import java.util.List;

public class QQExcelTool {
    public static void ExcelSheet(String fileName,List<ExcelMC> ltData) {
        try {
            XSSFWorkbook workbook = new XSSFWorkbook();
            for (ExcelMC mSheet : ltData) {
                XSSFSheet sheet1 = workbook.createSheet(mSheet.getSheetName());
                XSSFRow row = sheet1.createRow(0);  //创建row信息
                //创建单元格头
                for (int i = 0; i < mSheet.getLtTitle().size(); i++) {
                    row.createCell(i).setCellValue(mSheet.getLtTitle().get(i));
                }
                //创建单元格头标
                List<List<String>> ltBody = mSheet.getLtBody();
                for (int i = 0; i < ltBody.size(); i++) {
                    row = sheet1.createRow(i + 1);
                    int k = 0;
                    for (String s : ltBody.get(i)) {
                        row.createCell(k).setCellValue(s);
                        k++;
                    }
                }
            }
            ServletRequestAttributes requestModel = (ServletRequestAttributes) RequestContextHolder.currentRequestAttributes();
            HttpServletRequest request = requestModel.getRequest();
            HttpServletResponse response = requestModel.getResponse();
            //设置文件输出头
            assert response != null;
            response.setHeader("Content-Disposition", "attachment;filename=" + encodeDownloadFilename(fileName, request.getHeader("user-agent")));
            response.setContentType("application/vnd.openxmlformats-officedocument.spreadsheetml.sheet");
            response.setCharacterEncoding("utf-8");
            //下载输出流
            workbook.write(response.getOutputStream());
        }
        catch (Exception ignored){
            ignored.printStackTrace();
        }
    }
    public static String encodeDownloadFilename(String filename, String agent) throws IOException {
        if (agent.contains("Firefox")) { // 火狐浏览器
            filename = "=?UTF-8?B?" + new BASE64Encoder().encode(filename.getBytes("utf-8")) + "?=";
            filename = filename.replaceAll("\r\n", "");
        } else { // IE及其他浏览器
            filename = URLEncoder.encode(filename, "utf-8");
            filename = filename.replace("+", " ");
        }
        return filename;
    }
}