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<Map<String,String>>stepList 保存起来,Map里面的数据就是坐标的四个Parma,循环完数据之后,再单独循环坐标集合,统一调用上面的一行代码搞定合并,(记得合并是最后来做的事情)。
怕一段文字太多影响你们阅读,我上面说的是总体思路,下面我再简单说一下针对我这个demo和数据的逻辑。
1.循环整个list,得到当前对象StatisticsByPartVo,取到StatisticsByPartVo的total,根据total的值和当前rowNum的值,来为第一级添加坐标,我代码里添加了两次,因为序号也是跟第一级一起合并的。
2.循环当前对象StatisticsByPartVo的children,也是一个集合,得到当前对象secondChild,在循环secondChild的children,分别创建行列,在secondChild的children循环完毕后,为第二级添加坐标,代码的231行,其中注意rowNum和curNum等计数器的变化和位置,如果这个有错也会导致表格格式不对。
3.循环完之后,循环计步集合,合并表格,整体完毕
for(Map<String,Integer>map: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<String, Object> 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<String, Object> model,String excelName) {List<StatisticsByPartVo> 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<Map<String,Integer>>stepList = new ArrayList<>();//下面几行是为了替换模板第一行标题的文字HSSFRow headRow = sheet0.getRow(0);HSSFCell headCell = headRow.getCell(0);headCell.setCellValue(excelName);sheet0.createFreezePane(0,3,0,3);//因为模板的起始行是第三行,所以这里定义为3int rowNum = 3;for(int i = 0; i < list.size(); ++i) {Map<String,Integer>map2 = new HashMap<>();Map<String,Integer>map3 = 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;k<vo.getChildren().size();k++){StatisticsByPartVo secondChild = vo.getChildren().get(k);Map<String,Integer>map4 = 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(Map<String,Integer>map:stepList){sheet0.addMergedRegion(new CellRangeAddress(map.get("startIndexRow"), map.get("endIndexRow"), map.get("startIndexCol"), map.get("endIndexCol")));}}
}