使用POI导出Excel

r囧r小猫 2024-03-23 16:30 31阅读 0赞

> 提示：文章写完后，目录可以自动生成，如何生成可参考右边的帮助文档

#### 文章目录 ####

*  前言
 *  一、导入依赖
 *  二、使用步骤
 *   *  1.编写工具类
     *  2.传入数据
 *  总结

--------------------

## 前言 ##

最近有个需求需要导出Excel，要求使用POI，这里记录一下

--------------------

## 一、导入依赖 ##

首先在项目中导入POI需要的依赖

<dependency>
        <groupId>org.apache.poi</groupId>
        <artifactId>poi</artifactId>
        <version>4.1.2</version>
    </dependency>
    <dependency>
        <groupId>org.apache.poi</groupId>
        <artifactId>poi-ooxml</artifactId>
        <version>4.1.2</version>
    </dependency>

## 二、使用步骤 ##

### 1.编写工具类 ###

因为导出Excel在项目中的多处地方都有使用，所以仅仅在一个地方去编写显然不符合代码复用的要求，私以为抽出一个工具类是最佳方案，工具类如下：

import org.apache.poi.ss.usermodel.Cell;
    import org.apache.poi.ss.usermodel.CellStyle;
    import org.apache.poi.ss.usermodel.Row;
    import org.apache.poi.xssf.usermodel.XSSFSheet;
    import org.apache.poi.xssf.usermodel.XSSFWorkbook;
    import org.springframework.stereotype.Component;
    
    import java.io.File;
    import java.io.FileNotFoundException;
    import java.io.FileOutputStream;
    import java.io.OutputStream;
    import java.util.List;
    
    
    @Component
    public class ExcelUtil {
        
    
        /**
         * 导出表格
         *
         * @param data
         * @param filePath
         */
        public void createExcel(List<List<String>> data, String filePath) {
        
            XSSFWorkbook workbook = new XSSFWorkbook();
            XSSFSheet sheet = workbook.createSheet();
            for (int i = 0; i < data.size(); i++) {
        
                List<String> rowDate = data.get(i);
                Row row = sheet.createRow(i);
                for (int j = 0; j < rowDate.size(); j++) {
        
                    Cell cell = row.createCell(j);
                    cell.setCellValue(rowDate.get(j));
                }
            }
            File fileExcel = new File(filePath);
            fileExcel.deleteOnExit();
            File folder = new File(filePath.substring(0, filePath.indexOf(File.separator)));
            if (!folder.exists()) {
        
                folder.mkdirs();
            }
            OutputStream os = null;
            try {
        
                os = new FileOutputStream(filePath);
                workbook.write(os);
            } catch (Exception e) {
        
                e.printStackTrace();
            }
        }
    }

工具类中编写了一个导出Excel的方法，核心思想是：将需要导出的数据传入，data是一个双层List集合，里边的每一个子集合都是需要导出的每一行的数据，第一行是表头，将集合中的每一行拿出创建excel的行，再输出到工作簿中。而filePath就是我们需要输出Excel的路径，其实也可以将方法改造成流的形式，让用户自己决定去保存到某个位置。但是，，，，产品的需求是生成的excel要上传到OSS云存储，然后前端会拿到一个地址，然后再执行下载，咱也不知道，就按照需求来吧。

### 2.传入数据 ###

这里需要我们自己在业务层去将需要导出的数据进行整合。具体看下边代码

//根据查询条件查询场景记录
      List<SceneRecord> sceneRecordList = baseMapper.selectList(queryWrapper);
       List<List<String>> data = new ArrayList<>();
       List<String> headerData = new ArrayList<>(sceneRecordList.size());
       headerData.add("时间");
       headerData.add("会员ID");
       headerData.add("场景名称");
       headerData.add("减排量变化");
       headerData.add("附带信息");
       data.add(headerData);
       SimpleDateFormat simpleDateFormat = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
       //查询场景名称
       List<SceneRule> sceneRuleList = sceneRuleMapper.selectList(new QueryWrapper<>());
       HashMap<String, String> map = new HashMap<>(sceneRuleList.size());
       //遍历场景规则
       for (SceneRule rule : sceneRuleList) {
        
           map.put(rule.getId(), rule.getName());
       }
       for (SceneRecord sceneRecord : sceneRecordList) {
        
           List<String> bodyData = new ArrayList<>(sceneRecordList.size());
           bodyData.add(simpleDateFormat.format(sceneRecord.getCreateTime()));
           bodyData.add(sceneRecord.getMemberId());
           //将对应的场景名称添加
           bodyData.add(map.get(sceneRecord.getSceneRuleId()));
           bodyData.add(String.valueOf(sceneRecord.getEmission()));
           bodyData.add(sceneRecord.getAdditional());
           data.add(bodyData);
       }
       SimpleDateFormat simpleDateFormat1 = new SimpleDateFormat("yyyyMMddHHmmss");
       Date date = new Date();
       String filename = "场景减排量记录-" + simpleDateFormat1.format(date) + date.getTime() + ".xlsx";
       excelUtil.createExcel(data, BasicFilePath.PATH + File.separator + filename);
       String path = "";
       try {
        
           File file = new File(BasicFilePath.PATH + File.separator + filename);
           path = ossUtil.upload(filename, new FileInputStream(file));
           file.delete();
       } catch (FileNotFoundException e) {
        
           e.printStackTrace();
       }
       return ossUtil.getUrl(path);

其中部分业务代码，没有贴上去，应该可以看出个大概，主要操作就是先将header表头数据存入，然后将需要导出的数据一行行的添加进去，场景名称和场景规则这些是前端需要的数据处理，可以不用看，以上就是这次需求的解决办法

--------------------

## 总结 ##

另外一种EasyExcel之前已经记录过详情看这一篇博客  
[使用EasyExcel进行导入导出数据][EasyExcel]

[EasyExcel]: http://t.csdn.cn/eGEn6