java实现Excel表格导出

Excel表格导出工具类

import org.apache.poi.hssf.usermodel.*;
import org.apache.poi.ss.usermodel.Workbook;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;
import org.springframework.http.HttpStatus;
import org.springframework.http.ResponseEntity;
import org.springframework.web.multipart.MultipartFile;

import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.io.InputStream;
import java.util.List;

public class ExcelUtil {

    public static Workbook getWorkbook(MultipartFile file) throws IOException {
        String fileName = file.getOriginalFilename();
        if (!fileName.matches("^.+\\.(?i)(xls)$") && !fileName.matches("^.+\\.(?i)(xlsx)$")) {
            return null;
        }
        boolean isExcel2003 = true;
        if (fileName.matches("^.+\\.(?i)(xlsx)$")) {
            isExcel2003 = false;
        }
        InputStream input = file.getInputStream();
        Workbook work = null;
        if (isExcel2003) {
            work = new HSSFWorkbook(input);
        } else {
            work = new XSSFWorkbook(input);
        }
        return work;
        /**
         *
         * Sheet sheet = work.getSheetAt(0);//得到sheet
         * int colLength = row.getLastCellNum();//得到总列数
         * for(int i=2;i<=sheet.getLastRowNum();i++){//遍历行
         *    Cell cell = row.getCell(j);//当前单元格对象,根据行和列确定单元格
         *    String value = (cell == null) == true ? null : cell.getStringCellValue().trim();//得到字符串
         *    Date data = cell.getDateCellValue();//得到时间
         * }
         */
    }
    /**
     * Excel表格导出
     * @param response HttpServletResponse对象
     * @param excelData Excel表格的数据,封装为List>
     * @param sheetName sheet的名字
     * @param fileName 导出Excel的文件名
     * @param columnWidth Excel表格的宽度,建议为15
     * @throws IOException 抛IO异常
     */
    public static void exportExcel(HttpServletResponse response,
                                   List<List<String>> excelData,
                                   String sheetName,
                                   String fileName,
                                   int columnWidth) throws IOException {

        //声明一个工作簿
        HSSFWorkbook workbook = new HSSFWorkbook();

        //生成一个表格,设置表格名称
        HSSFSheet sheet = workbook.createSheet(sheetName);

        //设置表格列宽度
        sheet.setDefaultColumnWidth(columnWidth);

        //写入List>中的数据
        int rowIndex = 0;
        for(List<String> data : excelData){
            //创建一个row行,然后自增1
            HSSFRow row = sheet.createRow(rowIndex++);

            //遍历添加本行数据
            for (int i = 0; i < data.size(); i++) {
                //创建一个单元格
                HSSFCell cell = row.createCell(i);

                //创建一个内容对象
                HSSFRichTextString text = new HSSFRichTextString(data.get(i));

                //将内容对象的文字内容写入到单元格中
                cell.setCellValue(text);
            }
        }

        //准备将Excel的输出流通过response输出到页面下载
        //八进制输出流
        response.setContentType("application/octet-stream");

        //设置导出Excel的名称
        response.setHeader("Content-disposition", "attachment;filename=" + fileName);

        try {
            //刷新缓冲
            response.flushBuffer();

            //workbook将Excel写入到response的输出流中,供页面下载该Excel文件
            workbook.write(response.getOutputStream());

        } catch (IOException e) {
            e.printStackTrace();
        }


//        //关闭workbook
//        workbook.close();
    }
}

控制层
PageMerchantDTO 封装了请求参数,分页和一些其他的参数,去掉也可以,直接全查


    @ApiOperation("导出")
    @GetMapping("/auth/export")
    public void export(PageMerchantDTO merchantDTO, HttpServletResponse response) throws IOException {
   		 //返回结果集
        List<List<String>> result = merchantService.export(merchantDTO);
        //获取时间的工具自己生产就行了
        String date = DateUtil.getCurrentDate(DateUtil.DATE_FORMAT_YYYY_MM_DD);
        String sheetName = "XXX列表";
        String fileName = "XXXX列表"+date+".xls";

        ExcelUtil.exportExcel(response, result, sheetName, fileName, 15);
    }

接口

    List<List<String>> export(PageMerchantDTO merchantDTO);

service层

    @Override
    public List<List<String>> export(PageMerchantDTO merchantDTO) {

        List<List<String>> strList=new ArrayList<>();
        //头部也是一个list 
        List<String> head = new ArrayList<>();
        head.add("名称");
        head.add("项目");
        head.add("联系人");
        head.add("入驻时间");
        head.add("到期时间");
        head.add("状态");
        //放到第一行显示,下面的数据以此类推
        strList.add(head);
        //这个方法就是从数据库查询出来的一个分页list 我这是调用了一个分页的数据,根据项目本身的业务逻辑写就可以了,不用管我这里的方法
        PageResult page = findPage(merchantDTO);
        if(page.getContent().size()!=0){
        //强转了对象,可以忽略
            List<Merchant> contentList = (List<Merchant>) page.getContent();
            for (int i = 0; i <contentList.size() ; i++) {
            //给表格进行赋值
                List<String> data = new ArrayList<>();
                data.add(contentList.get(i).getXXX());
                data.add(contentList.get(i).getXXX());
                data.add(contentList.get(i).getXXX());
                data.add(contentList.get(i).getXXX());
                data.add(contentList.get(i).getXXX());
                data.add(contentList.get(i).getXXX());
                strList.add(data);
            }

        }
        return strList;
    }

你可能感兴趣的:(java实现Excel表格导出)