maven pom.xml下载poi包

org.apache.poi

poi

3.17

org.apache.poi

poi-ooxml

3.17

org.apache.poi

poi-ooxml-schemas

3.17

首先是实体类，也就是我们的数据集合

import java.util.List;

import com.baomidou.mybatisplus.annotation.TableField;

import com.baomidou.mybatisplus.annotation.TableId;

import com.baomidou.mybatisplus.annotation.TableName;

import lombok.Data;

import lombok.experimental.Accessors;

@Data

@Accessors(chain = true)

@TableName("test")

public class TestDomain {

private String phone;

private String parentPhone;

@TableId

private String id;

private String parentId;

@TableField(exist = false)

private List children;

/**列数*/

@TableField(exist = false)

private Integer colNumber;

}

再写个接口接收前端的导出请求，这里是使用的springboot 框架

@GetMapping("/test")

public void test(HttpServletResponse response, HttpServletRequest request) throws IOException {

// 查询数据库

List testDomains = testMapper.selectList(null);

// 封装成树形结构

TestTreeHandler testTreeHandler = new TestTreeHandler(testDomains);

List treeList = testTreeHandler.buildTree();

//创建poi导出数据对象

SXSSFWorkbook sxssfWorkbook = new SXSSFWorkbook();

//创建sheet页

SXSSFSheet sheet = sxssfWorkbook.createSheet("sheet页名字");

//创建表头

SXSSFRow headRow = sheet.createRow(0);

//设置表头信息

headRow.createCell(0).setCellValue("父号码");

headRow.createCell(1).setCellValue("子号码");

// 遍历上面数据库查到的数据

for (TestDomain testDomain : treeList) {

//填充数据

SXSSFRow dataRow = sheet.createRow(sheet.getLastRowNum() + 1);

//序号

//看你实体类在进行填充

dataRow.createCell(0).setCellValue(testDomain.getParentPhone());

dataRow.createCell(1).setCellValue(testDomain.getPhone());

testDomain.setColNumber(1);

if(!CollectionUtils.isEmpty(testDomain.getChildren())) {

child(testDomain.getChildren(), sheet,testDomain,dataRow);

}

}

// 下载导出

String filename = "导出excel表格名字";

// 设置头信息

response.setCharacterEncoding("UTF-8");

response.setContentType("application/vnd.ms-excel");

//一定要设置成xlsx格式

response.setHeader("Content-Disposition", "attachment;filename=" + URLEncoder.encode(filename + ".xlsx", "UTF-8"));

//创建一个输出流

ServletOutputStream outputStream = response.getOutputStream();

//写入数据

sxssfWorkbook.write(outputStream);

// 关闭

outputStream.close();

sxssfWorkbook.close();

return null;

}

递归子节点，往文件中逐级写入数据

/**

*

* @Title: child

* @Description: 填充子节点数据

* @param childList 子节点数据

* @param sheet 表格对象

* @param parentDomain 父节点数据

* @param parentRow 父节点行对象

* void

* @throws

* @Author XX

*/

public static void child(List childList,SXSSFSheet sheet,TestDomain parentDomain,SXSSFRow parentRow) {

for (TestDomain testDomain : childList) {

// 获取sheet的最后一行

SXSSFRow childRow = sheet.createRow(sheet.getLastRowNum()+1);

// 合并单元格

sheet.addMergedRegion(new CellRangeAddress(childRow.getRowNum(), childRow.getRowNum(), 0, parentDomain.getColNumber()));

// 设置子节点表头 parentDomain.getColNumber()+1 和 parentDomain.getColNumber()+2 表示根据父节点最大列数往后挪移1-2列，如果是10个字段就往后挪移1-10列

parentRow.createCell(parentDomain.getColNumber()+1).setCellValue("父号码");

parentRow.createCell(parentDomain.getColNumber()+2).setCellValue("子号码");

// 填充数据

childRow.createCell(parentDomain.getColNumber()+1).setCellValue(testDomain.getParentPhone());

childRow.createCell(parentDomain.getColNumber()+2).setCellValue(testDomain.getPhone());

testDomain.setColNumber(parentDomain.getColNumber()+2);

if(!CollectionUtils.isEmpty(testDomain.getChildren())) {

child(testDomain.getChildren(), sheet,testDomain,childRow);

}

}

}

将LIST装换成树形结构的处理类

import java.util.ArrayList;

import java.util.List;

import com.dadi01.scrm.service.mot.provider.domain.TestDomain;

public class TestTreeHandler {

/**

* 递归封装树形菜单

*/

List nodes = new ArrayList();

public TestTreeHandler(List nodes) {

super();

this.nodes = nodes;

}

/**

*

* @Title: buildTree

* @Description: 构建树形结构

* @return

* List

* @throws

* @Author XX

*/

public List buildTree() {

List TestDomains = new ArrayList();

List rootNodes = getRootNodes();

for (TestDomain rootNode : rootNodes) {

buildChildNodes(rootNode);

TestDomains.add(rootNode);

}

return TestDomains;

}

/**

*

* @Title: buildChildNodes

* @Description: 递归子节点 找出父节点下所有的层级

* @param node 上级节点

* void

* @throws

* @Author XX

*/

public void buildChildNodes(TestDomain node) {

List children = getChildNodes(node);

if (!children.isEmpty()) {

for (TestDomain child : children) {

buildChildNodes(child);

}

node.setChildren(children);

}

}

/**

*

* @Title: getChildNodes

* @Description: 获取父节点下所有的子节点

* @param pnode 父节点

* @return

* List

* @throws

* @Author XX

*/

public List getChildNodes(TestDomain pnode) {

List childNodes = new ArrayList();

for (TestDomain n : nodes) {

if (pnode.getId().equals(n.getParentId())) {

childNodes.add(n);

}

}

return childNodes;

}

/**

*

* @Title: rootNode

* @Description: 判断是否为根节点

* @param node 当前节点

* @return

* boolean

* @throws

* @Author XX

*/

public boolean rootNode(TestDomain node) {

return "0".equals(node.getParentId());

}

/**

*

* @Title: getRootNodes

* @Description: 获取集合中所有的根节点

* @return

* List

* @throws

* @Author XX

*/

public List getRootNodes() {

List rootNodes = new ArrayList();

for (TestDomain n : nodes) {

if (rootNode(n)) {

rootNodes.add(n);

}

}

return rootNodes;

}

}

浏览器敲地址访问就行了

导出的文件内容 0为根节点

本文地址：https://blog.csdn.net/yangzongzhuan/article/details/110875951

如您对本文有疑问或者有任何想说的，请点击进行留言回复，万千网友为您解惑！