poi实现对树形结构数据导出excel并合并表格
poi实现对树形结构数据导出excel并合并表格
- 1.主要逻辑！
- 2.完整代码
最近好像得罪了poi，遇到的都是导出word、Excel、pdf的问题。下面我记录一下poi对树形结构的处理，前面先梳理整体思路，文章末尾会贴上完整代码。
首先我们看一下模板和效果图吧：
怎么样，效果还行吧，有没有达到你心目中的标准？如果你也遇到了这样的问题，那么请静下心来花费几分钟时间，我会详细的梳理出自己的思路，希望能够帮助到你。
1.主要逻辑！
1.首先，我们来看一下树形结构的数据、正如效果图一样，我这里处理的是一个三层结构。
---- 祖先节点(包含祖先节点id，祖先节点name，children，total(当前对象的第三级子节点的数量))
－－－父亲节点（包含祖先节点id，祖先节点name，本身的id，本身的name，children）
－－－－子节点（包含上面两个祖先的数据和自己本身的数据，无children）
数据是提前处理好了的，在代码里的体现就是一个集合List ，StatisticsByPartVo是一个java bean对象，是个树形结构。
2.因为涉及到表格的合并，我再提一下关键的代码
sheet0.addMergedRegion(new CellRangeAddress(Parma1，Parma2，Parma3，Parma4));
四个参数分别表示
Parma1：要合并的开始行
Parma2：要合并的结束行
Parma3：要合并的开始列
Parma4：要合并的结束列
举个例子，因为坐标都是从(0,0)开始的，我要合并三行四列，参数就是(2,2,3,3)
现在开始整理一下整体思路：循环整个list集合，用HSSFRow 来创建行，用HSSFCell来创建一行的一个元素，在循环的过程中，将需要合并的数据的坐标，单独用一个List
怕一段文字太多影响你们阅读，我上面说的是总体思路，下面我再简单说一下针对我这个demo和数据的逻辑。
1.循环整个list，得到当前对象StatisticsByPartVo，取到StatisticsByPartVo的total，根据total的值和当前rowNum的值，来为第一级添加坐标，我代码里添加了两次，因为序号也是跟第一级一起合并的。
2.循环当前对象StatisticsByPartVo的children，也是一个集合，得到当前对象secondChild，在循环secondChild的children，分别创建行列，在secondChild的children循环完毕后，为第二级添加坐标，代码的231行，其中注意rowNum和curNum等计数器的变化和位置，如果这个有错也会导致表格格式不对。
3.循环完之后，循环计步集合，合并表格，整体完毕
for(Mapmap:stepList){
sheet0.addMergedRegion(new CellRangeAddress(map.get("startIndexRow"), map.get("endIndexRow"), map.get("startIndexCol"), map.get("endIndexCol")));
}
以上就是我的整体思路，如果你也被这个问题困扰，不如跟着我的思路走一下，整体代码和逻辑都不复杂，如果我的文章能够帮助你解决掉问题，别吝啬你的小指头给作者点个赞哦。
下面贴上完整逻辑代码，关键地方已经打上注释，欢迎下方留言，有不对的地方欢迎指出。转载请附上原文链接。
2.完整代码
package cdcb.govpublic.base.entity;
import cdcb.util.PropertiesUtils;
import org.apache.poi.hssf.usermodel.*;
import org.apache.poi.ss.usermodel.Font;
import org.apache.poi.ss.usermodel.Workbook;
import org.apache.poi.ss.util.CellRangeAddress;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.File;
import java.io.FileInputStream;
import java.io.OutputStream;
import java.net.URLEncoder;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
/**
* @author wq
* @date 2020/4/2.
*/
public class StatisticsByPartView {
public StatisticsByPartView() {
}
public static void buildExcelDocument(Map mode, HttpServletRequest request, HttpServletResponse response,String year) throws Exception {
String path = PropertiesUtils.getValueByKey("excelPath", request.getServletContext().getRealPath("/"));
String excelModelPath = path + "static/template/listByPart.xls";
File excel = new File(excelModelPath);
FileInputStream is = new FileInputStream(excel);
Workbook workbook = new HSSFWorkbook(is);
is.close();
String excelName = year+"年政府网各栏目数据报送情况汇总表";
//这个方法就是主要创建逻辑方法
setWorkBookValue(workbook, mode,excelName);
String userAgent = request.getHeader("User-Agent");
userAgent = userAgent == null ? "" : userAgent.toLowerCase();
if (!userAgent.contains("msie") && !userAgent.contains("trident")) {
excelName = new String(excelName.getBytes(), "iso-8859-1");
} else {
excelName = URLEncoder.encode(excelName, "UTF-8");
}
response.setContentType("application/octet-stream");
response.setHeader("Content-disposition", "attachment;filename=\"" + excelName + ".xls\"");
OutputStream ouputStream = response.getOutputStream();
workbook.write(ouputStream);
ouputStream.flush();
ouputStream.close();
}
private static void setWorkBookValue(Workbook wbs, Map model,String excelName) {
List list = (List)model.get("list");
HSSFCellStyle style = (HSSFCellStyle)wbs.createCellStyle();
style.setVerticalAlignment((short)1);
style.setAlignment((short)2);
style.setWrapText(false);
style.setBorderBottom((short)1);
style.setBorderRight((short)1);
style.setBorderTop((short)1);
style.setBorderLeft((short)1);
//这里设置style2的目的是给下面的循环做判断，如果数据为零则加粗标红
HSSFCellStyle style2 = (HSSFCellStyle) wbs.createCellStyle();
style2.setVerticalAlignment((short) 1);
style2.setAlignment((short) 2);
style2.setWrapText(false);
style2.setBorderBottom((short) 1);
style2.setBorderRight((short) 1);
style2.setBorderTop((short) 1);
style2.setBorderLeft((short) 1);
Font ztFont = wbs.createFont();
ztFont.setColor(Font.COLOR_RED);
ztFont.setBoldweight(HSSFFont.BOLDWEIGHT_BOLD);
style2.setFont(ztFont);
//创建工作簿
HSSFSheet sheet0 = (HSSFSheet)wbs.getSheetAt(0);
//记录步数的集合
List>stepList = new ArrayList<>();
//下面几行是为了替换模板第一行标题的文字
HSSFRow headRow = sheet0.getRow(0);
HSSFCell headCell = headRow.getCell(0);
headCell.setCellValue(excelName);
sheet0.createFreezePane(0,3,0,3);
//因为模板的起始行是第三行，所以这里定义为3
int rowNum = 3;
for(int i = 0; i < list.size(); ++i) {
Mapmap2 = new HashMap<>();
Mapmap3 = new HashMap<>();
StatisticsByPartVo vo = list.get(i);
//为当前对象的第一级和序号列添加步数
if(vo.getTotal()>1){
map2.put("startIndexRow",rowNum);
map2.put("endIndexRow",vo.getTotal()+rowNum-1);
map2.put("startIndexCol",0);
map2.put("endIndexCol",0);
map3.put("startIndexRow",rowNum);
map3.put("endIndexRow",vo.getTotal()+rowNum-1);
map3.put("startIndexCol",1);
map3.put("endIndexCol",1);
stepList.add(map2);
stepList.add(map3);
}
//循环第二级
int curNum = rowNum;
for(int k=0;kmap4 = new HashMap<>();
//创建第二级和第三级
for (int j = 0; j < secondChild.getChildren().size(); j++) {
StatisticsByPartVo third = secondChild.getChildren().get(j);
HSSFRow rowi = sheet0.createRow(curNum +j);
HSSFCell col0 = rowi.createCell(0);
col0.setCellStyle(style);
HSSFCell col1 = rowi.createCell(1);
col1.setCellStyle(style);
//创建第一级的数据，下面判断原因：因为要合并，合并内容的数据只需要创建一次就好，其他的为空
if(j==0){
col1.setCellValue(vo.getFirstName());
col0.setCellValue((double)(i + 1));
}else{
col1.setCellValue("");
col0.setCellValue("");
}
//创建第二级的数据
HSSFCell col2 = rowi.createCell(2);
col2.setCellStyle(style);
if(j == 0){
col2.setCellValue(third.getSecondName());
}else{
col2.setCellValue("");
}
HSSFCell col3 = rowi.createCell(3);
col3.setCellValue(third.getThirdName());
col3.setCellStyle(style);
//下面都是创建第三级的数据了
HSSFCell col4 = rowi.createCell(4);
col4.setCellValue(third.getMonth1());
if(third.getMonth1() == 0){
col4.setCellStyle(style2);
}else{
col4.setCellStyle(style);
}
HSSFCell col5 = rowi.createCell(5);
col5.setCellValue(third.getMonth2());
if(third.getMonth2() == 0){
col5.setCellStyle(style2);
}else{
col5.setCellStyle(style);
}
HSSFCell col6 = rowi.createCell(6);
col6.setCellValue(third.getMonth3());
if(third.getMonth3() == 0){
col6.setCellStyle(style2);
}else{
col6.setCellStyle(style);
}
HSSFCell col7 = rowi.createCell(7);
col7.setCellValue(third.getMonth4());
if(third.getMonth4() == 0){
col7.setCellStyle(style2);
}else{
col7.setCellStyle(style);
}
HSSFCell col8 = rowi.createCell(8);
col8.setCellValue(third.getMonth5());
if(third.getMonth5() == 0){
col8.setCellStyle(style2);
}else{
col8.setCellStyle(style);
}
HSSFCell col9 = rowi.createCell(9);
col9.setCellValue(third.getMonth6());
if(third.getMonth6() == 0){
col9.setCellStyle(style2);
}else{
col9.setCellStyle(style);
}
HSSFCell col10 = rowi.createCell(10);
col10.setCellValue(third.getMonth7());
if(third.getMonth7() == 0){
col10.setCellStyle(style2);
}else{
col10.setCellStyle(style);
}
HSSFCell col11 = rowi.createCell(11);
col11.setCellValue(third.getMonth8());
if(third.getMonth8() == 0){
col11.setCellStyle(style2);
}else{
col11.setCellStyle(style);
}
HSSFCell col12 = rowi.createCell(12);
col12.setCellValue(third.getMonth9());
if(third.getMonth9() == 0){
col12.setCellStyle(style2);
}else{
col12.setCellStyle(style);
}
HSSFCell col13 = rowi.createCell(13);
col13.setCellValue(third.getMonth10());
if(third.getMonth10() == 0){
col13.setCellStyle(style2);
}else{
col13.setCellStyle(style);
}
HSSFCell col14 = rowi.createCell(14);
col14.setCellValue(third.getMonth11());
if(third.getMonth11() == 0){
col14.setCellStyle(style2);
}else{
col14.setCellStyle(style);
}
HSSFCell col15 = rowi.createCell(15);
col15.setCellValue(third.getMonth12());
if(third.getMonth12() == 0){
col15.setCellStyle(style2);
}else{
col15.setCellStyle(style);
}
HSSFCell col16 = rowi.createCell(16);
col16.setCellValue(third.getTotal());
if(third.getTotal() == 0){
col16.setCellStyle(style2);
}else{
col16.setCellStyle(style);
}
}
//记录第二级合并的步数
if(secondChild.getChildren().size()>1){
map4.put("startIndexRow",curNum);
map4.put("endIndexRow",secondChild.getChildren().size()+curNum-1);
map4.put("startIndexCol",2);
map4.put("endIndexCol",2);
stepList.add(map4);
}
curNum+=secondChild.getChildren().size();
}
rowNum += vo.getTotal();
}
//循环合并表格
for(Mapmap:stepList){
sheet0.addMergedRegion(new CellRangeAddress(map.get("startIndexRow"), map.get("endIndexRow"), map.get("startIndexCol"), map.get("endIndexCol")));
}
}
}