Java Apache POI: 在 Excel 工作簿中插入日期数据
这段代码使用 Java Apache POI 库在 Excel 工作簿中插入日期数据,并进行一些操作,例如复制单元格格式、获取指定列的日期数据、在日期数据中插入中间值以及将日期数据写入指定列。
这段代码有一些问题,以下是对代码的分析和改进建议:
-
在复制单元格格式的方法
copyCellStyle中,应该使用createCellStyle创建新的单元格样式,而不是直接复制源单元格的样式。因为源单元格的样式是与源工作簿相关联的,如果直接复制样式,可能会导致样式无法正确应用到目标工作簿中。 -
在获取指定列的日期数据的方法
getColumnValues中,如果单元格的类型不是数字类型,会导致getCellType方法返回的值不是CellType.NUMERIC,从而无法正确获取日期数据。应该使用cell.getDateCellValue()方法来获取日期数据。 -
在插入日期的方法
insertDates中,修改了插入条件为大于 10 天,但没有修改相应的注释,导致注释与代码不一致。 -
在写入日期数据的方法
writeColumnValues中,创建了一个SimpleDateFormat对象,但没有将其赋值给一个变量或者使用它来格式化日期数据。应该将格式化后的日期字符串赋值给单元格的值。 -
在写入日期数据的方法
writeColumnValues中,创建了一个新的单元格样式,但没有将其应用到单元格中。应该使用cell.setCellStyle(style)方法将样式应用到单元格中。
以下是对代码的改进版本:
package Data_Recovery;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.text.SimpleDateFormat;
import java.util.ArrayList;
import java.util.Date;
import java.util.List;
import org.apache.poi.ss.usermodel.*;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;
public class EM03 {
public static void main(String[] args) {
try {
// 读取 input-2.xlsx 文件
FileInputStream file = new FileInputStream('input-2.xlsx');
Workbook workbook = new XSSFWorkbook(file);
Sheet sheet = workbook.getSheet('P1');
// 检查是否存在 P2 工作表,如果存在则删除
if (workbook.getSheet('P2') != null) {
workbook.removeSheetAt(workbook.getSheetIndex('P2'));
}
// 创建新的 P2 工作表
Sheet newSheet = workbook.createSheet('P2');
// 复制 P1 工作表的单元格格式到 P2 工作表
copyCellStyle(sheet, newSheet);
// 获取 P1 工作表第一列的时间日期数据
List<Date> dates = getColumnValues(sheet);
// 进行日期插入操作
List<Date> newDates = insertDates(dates);
// 将插入后的日期写入 P2 工作表
writeColumnValues(newSheet, newDates);
// 保存结果到 input-2.xlsx 文件
FileOutputStream outFile = new FileOutputStream('input-2.xlsx');
workbook.write(outFile);
outFile.close();
System.out.println('操作完成');
} catch (Exception e) {
e.printStackTrace();
}
}
// 复制单元格格式
private static void copyCellStyle(Sheet sourceSheet, Sheet targetSheet) {
for (int rowIndex = 0; rowIndex <= sourceSheet.getLastRowNum(); rowIndex++) {
Row sourceRow = sourceSheet.getRow(rowIndex);
Row targetRow = targetSheet.createRow(rowIndex);
for (int columnIndex = 0; columnIndex < sourceRow.getLastCellNum(); columnIndex++) {
Cell sourceCell = sourceRow.getCell(columnIndex);
Cell targetCell = targetRow.createCell(columnIndex);
if (sourceCell != null) {
CellStyle style = sourceCell.getCellStyle();
CellStyle newStyle = targetSheet.getWorkbook().createCellStyle();
newStyle.cloneStyleFrom(style); // 克隆样式
targetCell.setCellStyle(newStyle);
}
}
}
}
// 获取指定列的日期数据
private static List<Date> getColumnValues(Sheet sheet) {
List<Date> columnValues = new ArrayList<>();
for (int rowIndex = 0; rowIndex <= sheet.getLastRowNum(); rowIndex++) {
Row row = sheet.getRow(rowIndex);
if (row != null) {
Cell cell = row.getCell(0);
if (cell != null && cell.getCellType() == CellType.NUMERIC) {
columnValues.add(cell.getDateCellValue());
}
}
}
return columnValues;
}
// 在日期数据中插入中间值,直至相邻日期间隔小于 10 天
private static List<Date> insertDates(List<Date> dates) {
List<Date> newDates = new ArrayList<>(dates);
int originalSize = newDates.size();
int maxSize = originalSize * 2; // 设置插入的日期数量阈值
int insertCount = 0; // 插入计数器
for (int i = 0; i < newDates.size() - 1; i++) {
Date currentDate = newDates.get(i);
Date nextDate = newDates.get(i + 1);
long interval = (nextDate.getTime() - currentDate.getTime()) / (24 * 60 * 60 * 1000);
if (interval > 10) { // 修改条件为大于 10 天
while (interval > 10 && newDates.size() < maxSize && insertCount < 100) { // 修改条件为大于 10 天,并添加日期数量限制条件和插入计数器
Date middleDate = new Date((currentDate.getTime() + nextDate.getTime()) / 2);
if (!newDates.contains(middleDate)) {
newDates.add(i + 1, middleDate);
insertCount++;
}
interval = (nextDate.getTime() - middleDate.getTime()) / (24 * 60 * 60 * 1000);
nextDate = newDates.get(i + 2); // 更新相邻日期
}
}
}
return newDates;
}
// 将日期数据写入指定列
private static void writeColumnValues(Sheet sheet, List<Date> dates) {
SimpleDateFormat dateFormat = new SimpleDateFormat('yyyy-MM-dd hh:mm'); // 创建 SimpleDateFormat 对象并将其赋值给一个变量
for (int rowIndex = 0; rowIndex < dates.size(); rowIndex++) {
Row row = sheet.getRow(rowIndex);
if (row == null) {
row = sheet.createRow(rowIndex);
}
Cell cell = row.createCell(0);
cell.setCellValue(dateFormat.format(dates.get(rowIndex))); // 将格式化后的日期字符串赋值给单元格的值
CellStyle style = sheet.getWorkbook().createCellStyle();
style.setDataFormat(sheet.getWorkbook().createDataFormat().getFormat('yyyy-MM-dd hh:mm'));
cell.setCellStyle(style); // 将样式应用到单元格中
}
}
}
以上代码对原代码进行了改进,修复了之前提到的问题。
此外,还建议添加一些错误处理机制,例如在读取文件或写入文件时添加异常处理,以提高代码的健壮性。
希望这些改进建议能帮助您更好地理解和使用 Java Apache POI 库。
原文地址: https://www.cveoy.top/t/topic/fTDP 著作权归作者所有。请勿转载和采集!