java poi拆分excel文件,每个文件保留标题行
java poi拆分excel文件,每个文件保留标题行
jiankang666
发布于 2022-05-12 13:00:28
发布于 2022-05-12 13:00:28
代码可运行
运行总次数：0
代码可运行
一、背景
在工作中，经常需要处理excel文件，将提供的excel文件导入数据库，有时候文件太大我们就需要做拆分了，下面实现将xlsx文件和xls文件按行数拆分，每个文件都保留标题行。这篇文件接着前面拆分csv大文件进行学习。
二、后端代码
1、我们需要引入poi的依赖
<dependency>
<groupId>org.apache.poi</groupId>
<artifactId>poi</artifactId>
<version>4.0.1</version>
</dependency>
<!-- 07版本以后的格式 -->
<dependency>
<groupId>org.apache.poi</groupId>
<artifactId>poi-ooxml</artifactId>
<version>4.0.1</version>
</dependency>
2、拆分xls类型的文件方法
（1）参数inputStream为要拆分的文件流
（2）参数fileName为拆分文件名
（3）参数splitSize为拆分后每个文件的最大行数
（4）思路：根据前端传的excel文件流转换为一个Workbook文件，创建Workbook文件目的就是能够更好的操作excel，然后获取总行数，根据传到splitSize计算出需要创建几个excel文件，然后创建需要的若干个空excel放入list集合中，遍历原始excel，遍历时有行，单元格的概念，这个poi都能够操作，原始文件第一行的所有单元格都遍历到每个拆分文件的第一行，其余行随机往拆分文件里面一行一行的写，写到最后，将拆分文件集合中的Workbook文件写入存放拆分文件的目录中，关流。
/**
* 拆分xls文件并返回文件夹路径
* @param inputStream
* @param fileName
* @param splitSize
* @return
*/
public static String getXlsZipPath(InputStream inputStream, String fileName, int splitSize) {
StopWatch stopWatch = new StopWatch();
stopWatch.start();
Workbook workBook = null;
try {
workBook = WorkbookFactory.create(inputStream);
} catch (Exception e) {
logger.error("Load excel file error!", e);
return null;
}
try {
Sheet sheet = workBook.getSheetAt(0);
Iterator<Row> rowIterator = sheet.rowIterator();
int totalRows = sheet.getPhysicalNumberOfRows();
int splitNumber = (int) ((totalRows % splitSize == 0) ? (totalRows / splitSize) : (totalRows / splitSize + 1));
logger.info("xls文件总行数： {}行 拆分文件个数：{}个", totalRows, splitNumber);
List<HSSFWorkbook> hssfWorkbooks = new ArrayList<>();
//创建存放拆分文件的目录，文件夹存在，可能里面有内容，删除所有内容
File dir = new File(defaultDir + fileName);
if (dir.exists()) {
delAllFile(dir.getAbsolutePath());
}
dir.mkdirs();
//创建的拆分文件写入流并放入集合中
for (int count = 0; count < splitNumber; count++) {
HSSFWorkbook hssfWorkbook = new HSSFWorkbook();
hssfWorkbook.createSheet().createRow(0);
hssfWorkbooks.add(hssfWorkbook);
}
int oldRow = 0;
while (rowIterator.hasNext()) {
Row row = rowIterator.next();
if (row.getRowNum() == 0) {
oldRow++;
Iterator<Cell> cellIterator = row.cellIterator();
int index = 0;
while (cellIterator.hasNext()) {
Cell next = cellIterator.next();
// 解决获取excel数据的方法与实际类型不符
next.setCellType(CellType.STRING);
String value = next.getStringCellValue();
for (int count = 0; count < splitNumber; count++) {
HSSFRow row1 = hssfWorkbooks.get(count).getSheetAt(0).getRow(0);
row1.createCell(index).setCellValue(value);
}
index++;
}
}
if (row.getRowNum() != 0) {
Iterator<Cell> cellIterator = row.cellIterator();
HSSFWorkbook hssfWorkbook = hssfWorkbooks.get(oldRow % splitNumber);
HSSFSheet sheet1 = hssfWorkbook.getSheetAt(0);
int index = 0;
HSSFRow row1 = sheet1.createRow(sheet1.getLastRowNum() + 1);
while (cellIterator.hasNext()) {
Cell next = cellIterator.next();
next.setCellType(CellType.STRING);
String value = next.getStringCellValue();
row1.createCell(index++).setCellValue(value);
}
oldRow++;
}
}
//创建拆分文件并关流
for (int i = 0; i < hssfWorkbooks.size(); i++) {
String splitFilePath = defaultDir + fileName + File.separator + fileName + i + ".xlsx";
File file = new File(splitFilePath);
file.createNewFile();
FileOutputStream outputStream = new FileOutputStream(file);
hssfWorkbooks.get(i).write(outputStream);
hssfWorkbooks.get(i).close();
}
} catch (IOException e) {
logger.error("拆分xls文件失败 ：" + e);
}
stopWatch.stop();
logger.info("xls文件拆分共花费： " + stopWatch.getTotalTimeMillis() + " ms");
return defaultDir + fileName + File.separator;
}
3、拆分xlsx类型的文件方法
（1）参数inputStream为要拆分的文件流
（2）参数fileName为拆分文件名
（3）参数splitSize为拆分后每个文件的最大行数
（4）思路：根据前端传的excel文件流转换为一个Workbook文件，创建Workbook文件目的就是能够更好的操作excel，然后获取总行数，根据传到splitSize计算出需要创建几个excel文件，然后创建需要的若干个空excel放入list集合中，遍历原始excel，遍历时有行，单元格的概念，这个poi都能够操作，原始文件第一行的所有单元格都遍历到每个拆分文件的第一行，其余行随机往拆分文件里面一行一行的写，写到最后，将拆分文件集合中的Workbook文件写入存放拆分文件的目录中，关流。
/**
* 拆分xlsx文件并返回文件夹路径
*
* @param inputStream
* @param fileName
* @param splitSize
* @return
*/
public static String getXlsxZipPath(InputStream inputStream, String fileName, int splitSize) {
StopWatch stopWatch = new StopWatch();
stopWatch.start();
Workbook workBook = null;
try {
workBook = WorkbookFactory.create(inputStream);
} catch (Exception e) {
logger.error("Load excel file error!", e);
return null;
}
try {
Sheet sheet = workBook.getSheetAt(0);
Iterator<Row> rowIterator = sheet.rowIterator();
int totalRows = sheet.getPhysicalNumberOfRows();
int splitNumber = (int) ((totalRows % splitSize == 0) ? (totalRows / splitSize) : (totalRows / splitSize + 1));
logger.info("xlsx文件总行数： {}行 拆分文件个数：{}个", totalRows, splitNumber);
List<XSSFWorkbook> xssfWorkbooks = new ArrayList<>();
//创建存放拆分文件的目录，文件夹存在，可能里面有内容，删除所有内容
File dir = new File(defaultDir + fileName);
if (dir.exists()) {
delAllFile(dir.getAbsolutePath());
}
dir.mkdirs();
//创建的拆分文件写入流并放入集合中
for (int count = 0; count < splitNumber; count++) {
XSSFWorkbook xssfWorkbook = new XSSFWorkbook();
xssfWorkbook.createSheet().createRow(0);
xssfWorkbooks.add(xssfWorkbook);
}
int oldRow = 0;
while (rowIterator.hasNext()) {
Row row = rowIterator.next();
if (row.getRowNum() == 0) {
oldRow++;
Iterator<Cell> cellIterator = row.cellIterator();
int index = 0;
while (cellIterator.hasNext()) {
Cell next = cellIterator.next();
next.setCellType(CellType.STRING);
String value = next.getStringCellValue();
for (int count = 0; count < splitNumber; count++) {
XSSFRow row1 = xssfWorkbooks.get(count).getSheetAt(0).getRow(0);
row1.createCell(index).setCellValue(value);
}
index++;
}
}
if (row.getRowNum() != 0) {
Iterator<Cell> cellIterator = row.cellIterator();
XSSFWorkbook xssfWorkbook = xssfWorkbooks.get(oldRow % splitNumber);
XSSFSheet sheet1 = xssfWorkbook.getSheetAt(0);
int index = 0;
XSSFRow row1 = sheet1.createRow(sheet1.getLastRowNum() + 1);
while (cellIterator.hasNext()) {
Cell next = cellIterator.next();
next.setCellType(CellType.STRING);
String value = next.getStringCellValue();
row1.createCell(index++).setCellValue(value);
}
oldRow++;
}
}
//创建拆分文件并关流
for (int i = 0; i < xssfWorkbooks.size(); i++) {
String splitFilePath = defaultDir + fileName + File.separator + fileName + i + ".xlsx";
File file = new File(splitFilePath);
file.createNewFile();
FileOutputStream outputStream = new FileOutputStream(file);
xssfWorkbooks.get(i).write(outputStream);
xssfWorkbooks.get(i).close();
}
} catch (IOException e) {
logger.error("拆分xlsx文件失败 ：" + e);
}
stopWatch.stop();
logger.info("xlsx文件拆分共花费： " + stopWatch.getTotalTimeMillis() + " ms");
return defaultDir + fileName + File.separator;
}
4、这两种方法具体用哪一个需要在使用之前判断传过来的文件类型。
5、上面方法需要用到的删除文件方法如下
/***
* 删除文件夹
*
*/
public static void delFolder(String folderPath) {
try {
delAllFile(folderPath); // 删除完里面所有内容
String filePath = folderPath;
filePath = filePath.toString();
File myFilePath = new File(filePath);
myFilePath.delete(); // 删除空文件夹
} catch (Exception e) {
e.printStackTrace();
}
}
/***
* 删除指定文件夹下所有文件
*
* @param path 文件夹完整绝对路径
* @return
*/
public static boolean delAllFile(String path) {
boolean flag = false;
File file = new File(path);
if (!file.exists()) {
return flag;
}
if (!file.isDirectory()) {
return flag;
}
String[] tempList = file.list();
File temp = null;
for (int i = 0; i < tempList.length; i++) {
if (path.endsWith(File.separator)) {
temp = new File(path + tempList[i]);
} else {
temp = new File(path + File.separator + tempList[i]);
}
if (temp.isFile()) {
temp.delete();
}
if (temp.isDirectory()) {
delAllFile(path + "/" + tempList[i]);// 先删除文件夹里面的文件
delFolder(path + "/" + tempList[i]);// 再删除空文件夹
flag = true;
}
}
return flag;
}
三、总结
以上就是我封装的拆分excel文件的方法，希望可以帮到你，觉得有用点个赞
本文参与 腾讯云自媒体同步曝光计划，分享自微信公众号。
原始发表：2020-04-01，如有侵权请联系 cloudcommunity@tencent.com 删除
评论
登录后参与评论
推荐阅读
腾讯云开发者
