bonus
/
Bonus-Cloud-Material-Mall
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge remote-tracking branch 'origin/ah-simple-test' into ah-simple-test

This commit is contained in:
lizhenhua 2025-12-11 18:50:45 +08:00
parent 7393716069 d5125993c0
commit 0ea29f9ceb
2 changed files with 59 additions and 39 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
bonus-modules/bonus-material-mall/src/main/java/com/bonus/material/device/controller/DevMergeController.java
View File

@ -20,6 +20,7 @@ import com.bonus.material.device.domain.vo.DevInfoVo;
import com.bonus.material.device.domain.vo.DevMergeVo;
import com.bonus.material.device.service.DevInfoService;
import com.bonus.material.device.service.DevMergeService;
import com.bonus.material.utils.CenterExcelUtil;
import com.bonus.material.utils.FolderZipUtil;
import com.bonus.material.utils.ReflectUtils;
import io.swagger.annotations.Api;
@ -346,7 +347,8 @@ public class DevMergeController extends BaseController {
}
// 3. 调用工具类生成模板（自动触发下拉选生成）
ExcelUtil<EquipmentImportDTO> util = new ExcelUtil<>(EquipmentImportDTO.class);
// 使用 CenterExcelUtil 替代 ExcelUtil，确保所有单元格（包括空白行）都居中对齐
CenterExcelUtil<EquipmentImportDTO> util = new CenterExcelUtil<>(EquipmentImportDTO.class);
// importTemplateExcel：专门生成导入模板的方法（仅表头+下拉选，无数据）
util.importTemplateExcel(response, "装备信息导入模板", "装备信息导入模板");
}

94
bonus-modules/bonus-material-mall/src/main/java/com/bonus/material/utils/CenterExcelUtil.java
View File

@ -30,7 +30,7 @@ public class CenterExcelUtil<T> extends ExcelUtil<T> {
try {
// 1. 将Excel写入到字节数组（而不是直接写入响应）
ByteArrayOutputStream baos = new ByteArrayOutputStream();
HttpServletResponse tempResponse = createTempResponse(baos);
HttpServletResponse tempResponse = createTempResponse(response, baos);
// 2. 调用父类方法生成模板
super.importTemplateExcel(tempResponse, sheetName, title);
@ -54,8 +54,8 @@ public class CenterExcelUtil<T> extends ExcelUtil<T> {
/**
* 创建临时响应对象，用于捕获Excel字节数据
*/
private HttpServletResponse createTempResponse(ByteArrayOutputStream baos) throws IOException {
return new HttpServletResponseWrapper(null) {
private HttpServletResponse createTempResponse(HttpServletResponse originalResponse, ByteArrayOutputStream baos) throws IOException {
return new HttpServletResponseWrapper(originalResponse) {
@Override
public javax.servlet.ServletOutputStream getOutputStream() {
return new javax.servlet.ServletOutputStream() {
@ -86,23 +86,67 @@ public class CenterExcelUtil<T> extends ExcelUtil<T> {
Workbook workbook = WorkbookFactory.create(bais);
ByteArrayOutputStream baos = new ByteArrayOutputStream()) {
// 创建居中样式（用于空白单元格）
CellStyle centerStyle = workbook.createCellStyle();
centerStyle.setAlignment(HorizontalAlignment.CENTER);
centerStyle.setVerticalAlignment(VerticalAlignment.CENTER);
// 使用Map缓存已处理的样式，避免创建过多样式对象（POI有64000个样式的限制）
java.util.Map<Short, CellStyle> styleCache = new java.util.HashMap<>();
// 遍历所有工作表
for (int i = 0; i < workbook.getNumberOfSheets(); i++) {
Sheet sheet = workbook.getSheetAt(i);
if (sheet == null) continue;
// 遍历所有存在的行
for (int rowIndex = sheet.getFirstRowNum(); rowIndex <= sheet.getLastRowNum(); rowIndex++) {
// 获取表头行以确定列数
Row headerRow = sheet.getRow(sheet.getFirstRowNum());
if (headerRow == null) continue;
// 获取实际的列数，取表头列数和11（A-K列）的最大值
int headerColNum = headerRow.getLastCellNum();
int maxColNum = Math.max(headerColNum, 11); // 确保至少包含A-K列（0-10）
// 预创建足够多的行（例如1000行），并为所有单元格设置居中样式
// 这样用户在任何位置输入内容时都会自动居中
int maxRows = 1000; // 可以根据需要调整行数
for (int rowIndex = 0; rowIndex < maxRows; rowIndex++) {
Row row = sheet.getRow(rowIndex);
if (row == null) continue;
if (row == null) {
row = sheet.createRow(rowIndex);
}
// 遍历所有存在的单元格
for (int cellIndex = row.getFirstCellNum(); cellIndex <= row.getLastCellNum(); cellIndex++) {
Cell cell = row.getCell(cellIndex);
if (cell == null) continue;
// 为该行的所有列创建单元格并设置居中样式（A-K列，即0-10）
for (int colIndex = 0; colIndex < maxColNum; colIndex++) {
Cell cell = row.getCell(colIndex);
if (cell == null) {
cell = row.createCell(colIndex);
}
// 设置单元格居中
setCellCenterStyle(workbook, cell);
// 获取或创建居中样式
CellStyle existingStyle = cell.getCellStyle();
short styleIndex = existingStyle.getIndex();
// 检查缓存中是否已有该样式的居中版本
CellStyle centeredStyle = styleCache.get(styleIndex);
if (centeredStyle == null) {
// 创建新的居中样式
if (styleIndex == 0) {
// 默认样式，使用预创建的居中样式
centeredStyle = centerStyle;
} else {
// 自定义样式，复制并添加居中
centeredStyle = workbook.createCellStyle();
centeredStyle.cloneStyleFrom(existingStyle);
centeredStyle.setAlignment(HorizontalAlignment.CENTER);
centeredStyle.setVerticalAlignment(VerticalAlignment.CENTER);
}
// 缓存样式
styleCache.put(styleIndex, centeredStyle);
}
// 应用居中样式
cell.setCellStyle(centeredStyle);
}
}
}
@ -112,32 +156,6 @@ public class CenterExcelUtil<T> extends ExcelUtil<T> {
}
}
/**
* 设置单元格居中样式
*/
private void setCellCenterStyle(Workbook workbook, Cell cell) {
// 获取当前样式
CellStyle currentStyle = cell.getCellStyle();
// 创建新样式
CellStyle newStyle = workbook.createCellStyle();
// 复制原有样式
newStyle.cloneStyleFrom(currentStyle);
// 设置水平居中
newStyle.setAlignment(HorizontalAlignment.CENTER);
newStyle.setVerticalAlignment(VerticalAlignment.CENTER);
// 保持原有的数据格式
if (currentStyle.getDataFormatString() != null) {
newStyle.setDataFormat(Short.parseShort(currentStyle.getDataFormatString()));
}
// 应用新样式
cell.setCellStyle(newStyle);
}
/**
* 设置响应头
*/