提示:文章写完后,目录可以自动生成,如何生成可参考右边的帮助文档
前言
最近有个需求需要导出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 {
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进行导入导出数据