|
|
@@ -0,0 +1,967 @@
|
|
|
+package shop.alien.store.service.impl;
|
|
|
+
|
|
|
+import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
|
|
|
+import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
|
|
|
+import lombok.RequiredArgsConstructor;
|
|
|
+import lombok.extern.slf4j.Slf4j;
|
|
|
+import org.apache.commons.lang3.StringUtils;
|
|
|
+import org.apache.poi.ss.usermodel.*;
|
|
|
+import org.apache.poi.xssf.usermodel.XSSFWorkbook;
|
|
|
+import org.springframework.stereotype.Service;
|
|
|
+import org.springframework.transaction.annotation.Transactional;
|
|
|
+import org.springframework.web.multipart.MultipartFile;
|
|
|
+import shop.alien.entity.result.R;
|
|
|
+import shop.alien.entity.store.StoreDictionary;
|
|
|
+import shop.alien.entity.store.excelVo.BusinessSectionExcelVo;
|
|
|
+import shop.alien.entity.store.excelVo.util.ExcelHeader;
|
|
|
+import shop.alien.mapper.StoreDictionaryMapper;
|
|
|
+import shop.alien.store.service.PlatformBusinessSectionService;
|
|
|
+
|
|
|
+import javax.servlet.http.HttpServletResponse;
|
|
|
+import java.io.IOException;
|
|
|
+import java.io.InputStream;
|
|
|
+import java.io.OutputStream;
|
|
|
+import java.lang.reflect.Field;
|
|
|
+import java.net.URLEncoder;
|
|
|
+import java.util.ArrayList;
|
|
|
+import java.util.Date;
|
|
|
+import java.util.HashMap;
|
|
|
+import java.util.List;
|
|
|
+import java.util.Map;
|
|
|
+
|
|
|
+/**
|
|
|
+ * 平台经营版块管理ServiceImpl
|
|
|
+ *
|
|
|
+ * @author ssk
|
|
|
+ * @version 1.0
|
|
|
+ * @date 2025/01/01
|
|
|
+ */
|
|
|
+@Slf4j
|
|
|
+@Service
|
|
|
+@RequiredArgsConstructor
|
|
|
+public class PlatformBusinessSectionServiceImpl extends ServiceImpl<StoreDictionaryMapper, StoreDictionary> implements PlatformBusinessSectionService {
|
|
|
+
|
|
|
+ private final StoreDictionaryMapper storeDictionaryMapper;
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 查询经营种类（三级树形结构）
|
|
|
+ *
|
|
|
+ * @return 经营种类树形结构列表
|
|
|
+ */
|
|
|
+ @Override
|
|
|
+ public List<StoreDictionary> queryBusinessSectionTree() {
|
|
|
+ // 查询所有经营种类数据
|
|
|
+ LambdaQueryWrapper<StoreDictionary> queryWrapper = new LambdaQueryWrapper<>();
|
|
|
+ queryWrapper.eq(StoreDictionary::getTypeName, "business_section");
|
|
|
+ queryWrapper.eq(StoreDictionary::getDeleteFlag, 0);
|
|
|
+ queryWrapper.orderByAsc(StoreDictionary::getDictId);
|
|
|
+ List<StoreDictionary> storeDictionaryList = storeDictionaryMapper.selectList(queryWrapper);
|
|
|
+
|
|
|
+ // 构建三级树形结构
|
|
|
+ return buildTreeOptimized(storeDictionaryList);
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 构建树形结构（优化版）
|
|
|
+ *
|
|
|
+ * @param flatList 扁平列表
|
|
|
+ * @return 树形结构列表
|
|
|
+ */
|
|
|
+ private List<StoreDictionary> buildTreeOptimized(List<StoreDictionary> flatList) {
|
|
|
+ if (flatList == null || flatList.isEmpty()) {
|
|
|
+ return new ArrayList<>();
|
|
|
+ }
|
|
|
+
|
|
|
+ // 创建三个存储结构
|
|
|
+ Map<Integer, StoreDictionary> nodeMap = new HashMap<>(); // ID到节点的映射
|
|
|
+ Map<Integer, List<StoreDictionary>> parentChildMap = new HashMap<>(); // 父ID到子节点列表的映射
|
|
|
+ List<StoreDictionary> result = new ArrayList<>(); // 结果列表
|
|
|
+
|
|
|
+ // 填充nodeMap和parentChildMap
|
|
|
+ for (StoreDictionary entity : flatList) {
|
|
|
+ Integer id = entity.getId();
|
|
|
+ Integer parentId = entity.getParentId();
|
|
|
+
|
|
|
+ // 存入节点映射
|
|
|
+ nodeMap.put(id, entity);
|
|
|
+
|
|
|
+ // 初始化子节点列表
|
|
|
+ entity.setStoreDictionaryList(new ArrayList<>());
|
|
|
+
|
|
|
+ // 如果是根节点（parentId为null或0），直接添加到结果
|
|
|
+ if (parentId == null || parentId == 0) {
|
|
|
+ result.add(entity);
|
|
|
+ } else {
|
|
|
+ // 否则，记录父子关系
|
|
|
+ parentChildMap.computeIfAbsent(parentId, k -> new ArrayList<>()).add(entity);
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ // 建立父子关系
|
|
|
+ for (StoreDictionary entity : flatList) {
|
|
|
+ Integer id = entity.getId();
|
|
|
+ if (parentChildMap.containsKey(id)) {
|
|
|
+ entity.getStoreDictionaryList().addAll(parentChildMap.get(id));
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ return result;
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 新增经营种类（支持一级、二级、三级）
|
|
|
+ *
|
|
|
+ * @param storeDictionary 经营种类信息
|
|
|
+ * @return 新增结果
|
|
|
+ */
|
|
|
+ @Override
|
|
|
+ public boolean addBusinessSection(StoreDictionary storeDictionary) {
|
|
|
+ // 参数校验
|
|
|
+ if (storeDictionary == null || StringUtils.isBlank(storeDictionary.getDictDetail())) {
|
|
|
+ throw new IllegalArgumentException("经营种类描述不能为空");
|
|
|
+ }
|
|
|
+
|
|
|
+ // 设置固定字段
|
|
|
+ storeDictionary.setTypeName("business_section");
|
|
|
+// storeDictionary.setTypeDetail("经营板块");
|
|
|
+ storeDictionary.setDeleteFlag(0);
|
|
|
+ storeDictionary.setCreatedTime(new Date());
|
|
|
+
|
|
|
+ // 处理 parent_id：如果为 null，设置为 0（一级）
|
|
|
+ Integer parentId = storeDictionary.getParentId();
|
|
|
+ if (parentId == null) {
|
|
|
+ parentId = 0;
|
|
|
+// storeDictionary.setParentId(0);
|
|
|
+ }
|
|
|
+
|
|
|
+ // 如果是二级或三级，验证父节点是否存在
|
|
|
+ if (parentId != 0) {
|
|
|
+ StoreDictionary parent = storeDictionaryMapper.selectById(parentId);
|
|
|
+ if (parent == null || !"business_section".equals(parent.getTypeName()) || parent.getDeleteFlag() == 1) {
|
|
|
+ throw new IllegalArgumentException("父节点不存在或已删除");
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ // 生成 dict_id：查询同级（相同 parent_id）的最大 dict_id，然后 +1
|
|
|
+ String maxDictId = getMaxDictIdByParentId(parentId);
|
|
|
+ int nextDictId = (maxDictId == null ? 0 : Integer.parseInt(maxDictId)) + 1;
|
|
|
+ storeDictionary.setDictId(String.valueOf(nextDictId));
|
|
|
+
|
|
|
+ // 保存
|
|
|
+ return this.save(storeDictionary);
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 修改经营种类（支持一级、二级、三级）
|
|
|
+ *
|
|
|
+ * @param storeDictionary 经营种类信息
|
|
|
+ * @return 修改结果
|
|
|
+ */
|
|
|
+ @Override
|
|
|
+ public boolean updateBusinessSection(StoreDictionary storeDictionary) {
|
|
|
+ // 参数校验
|
|
|
+ if (storeDictionary == null || storeDictionary.getId() == null) {
|
|
|
+ throw new IllegalArgumentException("ID不能为空");
|
|
|
+ }
|
|
|
+ if (StringUtils.isBlank(storeDictionary.getDictDetail())) {
|
|
|
+ throw new IllegalArgumentException("经营种类描述不能为空");
|
|
|
+ }
|
|
|
+
|
|
|
+ // 查询原记录
|
|
|
+ StoreDictionary existing = storeDictionaryMapper.selectById(storeDictionary.getId());
|
|
|
+ if (existing == null) {
|
|
|
+ throw new IllegalArgumentException("记录不存在");
|
|
|
+ }
|
|
|
+ if (!"business_section".equals(existing.getTypeName())) {
|
|
|
+ throw new IllegalArgumentException("该记录不是经营种类类型");
|
|
|
+ }
|
|
|
+ if (existing.getDeleteFlag() == 1) {
|
|
|
+ throw new IllegalArgumentException("该记录已删除");
|
|
|
+ }
|
|
|
+
|
|
|
+ // 判断是否是一级或二级节点
|
|
|
+ Integer existingParentId = existing.getParentId();
|
|
|
+ boolean isFirstLevel = (existingParentId == null || existingParentId == 0);
|
|
|
+ boolean isSecondLevel = false;
|
|
|
+
|
|
|
+ // 如果不是一级节点，判断是否是二级节点（父节点是一级节点）
|
|
|
+ if (!isFirstLevel) {
|
|
|
+ StoreDictionary parent = storeDictionaryMapper.selectById(existingParentId);
|
|
|
+ if (parent != null) {
|
|
|
+ Integer grandParentId = parent.getParentId();
|
|
|
+ isSecondLevel = (grandParentId == null || grandParentId == 0);
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ // 处理 parent_id：如果为 null，设置为 0（一级）
|
|
|
+ Integer parentId = storeDictionary.getParentId();
|
|
|
+ if (parentId == null) {
|
|
|
+ parentId = 0;
|
|
|
+ }
|
|
|
+
|
|
|
+ // 标准化parentId用于比较（null转为0）
|
|
|
+ Integer normalizedParentId = (parentId == null) ? 0 : parentId;
|
|
|
+ Integer normalizedExistingParentId = (existingParentId == null) ? 0 : existingParentId;
|
|
|
+ String oldDictId = existing.getDictId();
|
|
|
+ String newDictId = storeDictionary.getDictId();
|
|
|
+ boolean parentIdChanged = !normalizedParentId.equals(normalizedExistingParentId);
|
|
|
+
|
|
|
+ // 判断是否修改了关键字段（parent_id）
|
|
|
+ boolean keyFieldChanged = parentIdChanged;
|
|
|
+
|
|
|
+ // 如果是一级或二级节点，且修改了关键字段（parent_id），检查是否有未删除的子节点
|
|
|
+ if ((isFirstLevel || isSecondLevel) && keyFieldChanged) {
|
|
|
+ boolean hasChildren = hasUndeletedChildren(existing.getId());
|
|
|
+ if (hasChildren) {
|
|
|
+ throw new IllegalArgumentException("该节点存在未删除的子节点，不允许修改关键字段（父节点）");
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ // 如果修改了 parentId，需要验证新的父节点是否存在
|
|
|
+ if (parentIdChanged) {
|
|
|
+ if (parentId != 0) {
|
|
|
+ StoreDictionary parent = storeDictionaryMapper.selectById(parentId);
|
|
|
+ if (parent == null || !"business_section".equals(parent.getTypeName()) || parent.getDeleteFlag() == 1) {
|
|
|
+ throw new IllegalArgumentException("父节点不存在或已删除");
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ // 如果用户指定了新位置（newDictId不为空），使用用户指定的位置
|
|
|
+ // 否则，使用新节点下的最大+1
|
|
|
+ if (StringUtils.isBlank(newDictId)) {
|
|
|
+ String maxDictId = getMaxDictIdByParentId(parentId);
|
|
|
+ int nextDictId = (maxDictId == null ? 0 : Integer.parseInt(maxDictId)) + 1;
|
|
|
+ newDictId = String.valueOf(nextDictId);
|
|
|
+ storeDictionary.setDictId(newDictId);
|
|
|
+ }
|
|
|
+
|
|
|
+ // 原节点下，原位置之后的记录需要前移（dictId - 1）
|
|
|
+ adjustSortOrderForMoveOut(existing.getId(), normalizedExistingParentId, oldDictId);
|
|
|
+
|
|
|
+ // 新节点下，如果指定了新位置，需要调整新节点下的排序
|
|
|
+ if (StringUtils.isNotBlank(newDictId)) {
|
|
|
+ try {
|
|
|
+ int newOrder = Integer.parseInt(newDictId);
|
|
|
+ String maxDictId = getMaxDictIdByParentId(parentId);
|
|
|
+ int maxOrder = (maxDictId == null ? 0 : Integer.parseInt(maxDictId));
|
|
|
+ // 如果新位置不是最大+1，需要调整新节点下的排序
|
|
|
+ if (newOrder <= maxOrder) {
|
|
|
+ adjustSortOrderForMoveIn(existing.getId(), normalizedParentId, newDictId);
|
|
|
+ }
|
|
|
+ } catch (NumberFormatException e) {
|
|
|
+ log.warn("无法解析新dictId: {}", newDictId, e);
|
|
|
+ }
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ // 设置固定字段
|
|
|
+ storeDictionary.setTypeName("business_section");
|
|
|
+ // 如果没有指定删除标记，保持原有值；如果指定了，使用新值
|
|
|
+ Integer oldHidden = existing.getHidden();
|
|
|
+ Integer newHidden = storeDictionary.getHidden() == null ? oldHidden : storeDictionary.getHidden();
|
|
|
+ storeDictionary.setHidden(newHidden);
|
|
|
+ storeDictionary.setDeleteFlag(existing.getDeleteFlag()); // 保持原有删除标记
|
|
|
+ storeDictionary.setCreatedTime(existing.getCreatedTime()); // 保持原有创建时间
|
|
|
+ storeDictionary.setCreatedUserId(existing.getCreatedUserId()); // 保持原有创建人
|
|
|
+ storeDictionary.setUpdatedTime(new Date()); // 更新修改时间
|
|
|
+
|
|
|
+ // 处理排序逻辑：如果 parentId 没有改变，但 dictId 改变了，需要调整同级其他记录的排序
|
|
|
+ if (!parentIdChanged) {
|
|
|
+ // 如果 dictId 为空，保持原有值
|
|
|
+ if (StringUtils.isBlank(newDictId)) {
|
|
|
+ storeDictionary.setDictId(oldDictId);
|
|
|
+ } else if (!newDictId.equals(oldDictId)) {
|
|
|
+ // 调整同级其他记录的排序
|
|
|
+ adjustSortOrder(existing.getId(), normalizedParentId, oldDictId, newDictId);
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ // 更新当前节点
|
|
|
+ boolean updateResult = this.updateById(storeDictionary);
|
|
|
+
|
|
|
+ // 如果是一级或二级节点，且hidden值发生了变化，同步更新所有子节点的hidden值
|
|
|
+ boolean hiddenChanged = (oldHidden == null && newHidden != null) ||
|
|
|
+ (oldHidden != null && !oldHidden.equals(newHidden));
|
|
|
+ if (updateResult && (isFirstLevel || isSecondLevel) && hiddenChanged && newHidden != null) {
|
|
|
+ updateChildrenHidden(existing.getId(), newHidden);
|
|
|
+ }
|
|
|
+
|
|
|
+ return updateResult;
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 判断节点是否有未删除的子节点
|
|
|
+ *
|
|
|
+ * @param parentId 父节点ID
|
|
|
+ * @return 如果有未删除的子节点返回 true，否则返回 false
|
|
|
+ */
|
|
|
+ private boolean hasUndeletedChildren(Integer parentId) {
|
|
|
+ LambdaQueryWrapper<StoreDictionary> queryWrapper = new LambdaQueryWrapper<>();
|
|
|
+ queryWrapper.eq(StoreDictionary::getTypeName, "business_section");
|
|
|
+ queryWrapper.eq(StoreDictionary::getParentId, parentId);
|
|
|
+ queryWrapper.eq(StoreDictionary::getDeleteFlag, 0);
|
|
|
+ queryWrapper.last("LIMIT 1");
|
|
|
+
|
|
|
+ StoreDictionary child = storeDictionaryMapper.selectOne(queryWrapper);
|
|
|
+ return child != null;
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 递归更新子节点的显示/隐藏状态
|
|
|
+ * 当一级或二级节点的hidden值改变时，同步更新其所有子节点的hidden值
|
|
|
+ *
|
|
|
+ * @param parentId 父节点ID
|
|
|
+ * @param hidden 新的hidden值（0:不隐藏, 1:隐藏）
|
|
|
+ */
|
|
|
+ private void updateChildrenHidden(Integer parentId, Integer hidden) {
|
|
|
+ if (parentId == null || hidden == null) {
|
|
|
+ return;
|
|
|
+ }
|
|
|
+
|
|
|
+ // 查询所有未删除的子节点
|
|
|
+ LambdaQueryWrapper<StoreDictionary> queryWrapper = new LambdaQueryWrapper<>();
|
|
|
+ queryWrapper.eq(StoreDictionary::getTypeName, "business_section");
|
|
|
+ queryWrapper.eq(StoreDictionary::getParentId, parentId);
|
|
|
+ queryWrapper.eq(StoreDictionary::getDeleteFlag, 0);
|
|
|
+
|
|
|
+ List<StoreDictionary> children = storeDictionaryMapper.selectList(queryWrapper);
|
|
|
+
|
|
|
+ if (children == null || children.isEmpty()) {
|
|
|
+ return;
|
|
|
+ }
|
|
|
+
|
|
|
+ // 批量更新子节点的hidden值
|
|
|
+ Date updateTime = new Date();
|
|
|
+ for (StoreDictionary child : children) {
|
|
|
+ child.setHidden(hidden);
|
|
|
+ child.setUpdatedTime(updateTime);
|
|
|
+ storeDictionaryMapper.updateById(child);
|
|
|
+
|
|
|
+ // 递归更新子节点的子节点（二级节点的子节点是三级节点）
|
|
|
+ updateChildrenHidden(child.getId(), hidden);
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 根据 parent_id 获取同级最大 dict_id
|
|
|
+ *
|
|
|
+ * @param parentId 父节点ID
|
|
|
+ * @return 最大 dict_id，如果不存在则返回 null
|
|
|
+ */
|
|
|
+ private String getMaxDictIdByParentId(Integer parentId) {
|
|
|
+ LambdaQueryWrapper<StoreDictionary> queryWrapper = new LambdaQueryWrapper<>();
|
|
|
+ queryWrapper.eq(StoreDictionary::getTypeName, "business_section");
|
|
|
+ queryWrapper.eq(StoreDictionary::getDeleteFlag, 0);
|
|
|
+
|
|
|
+ if (parentId == null || parentId == 0) {
|
|
|
+ queryWrapper.and(wrapper -> wrapper.isNull(StoreDictionary::getParentId)
|
|
|
+ .or().eq(StoreDictionary::getParentId, 0));
|
|
|
+ } else {
|
|
|
+ queryWrapper.eq(StoreDictionary::getParentId, parentId);
|
|
|
+ }
|
|
|
+
|
|
|
+ queryWrapper.orderByDesc(StoreDictionary::getDictId);
|
|
|
+ queryWrapper.last("LIMIT 1");
|
|
|
+
|
|
|
+ StoreDictionary maxDict = storeDictionaryMapper.selectOne(queryWrapper);
|
|
|
+ return maxDict != null ? maxDict.getDictId() : null;
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 调整排序：当记录的dictId改变时，调整同级其他记录的排序
|
|
|
+ * 上升则顺推：如果新dictId < 原dictId，将原dictId到新dictId之间的记录的dictId都+1
|
|
|
+ * 下降则顺升：如果新dictId > 原dictId，将原dictId到新dictId之间的记录的dictId都-1
|
|
|
+ *
|
|
|
+ * @param currentId 当前记录ID
|
|
|
+ * @param parentId 父节点ID
|
|
|
+ * @param oldDictId 原dictId
|
|
|
+ * @param newDictId 新dictId
|
|
|
+ */
|
|
|
+ private void adjustSortOrder(Integer currentId, Integer parentId, String oldDictId, String newDictId) {
|
|
|
+ try {
|
|
|
+ int oldOrder = Integer.parseInt(oldDictId);
|
|
|
+ int newOrder = Integer.parseInt(newDictId);
|
|
|
+
|
|
|
+ // 如果排序没有改变，直接返回
|
|
|
+ if (oldOrder == newOrder) {
|
|
|
+ return;
|
|
|
+ }
|
|
|
+
|
|
|
+ // 查询同级所有记录（排除当前记录）
|
|
|
+ LambdaQueryWrapper<StoreDictionary> queryWrapper = new LambdaQueryWrapper<>();
|
|
|
+ queryWrapper.eq(StoreDictionary::getTypeName, "business_section");
|
|
|
+ queryWrapper.eq(StoreDictionary::getDeleteFlag, 0);
|
|
|
+ queryWrapper.ne(StoreDictionary::getId, currentId);
|
|
|
+
|
|
|
+ if (parentId == null || parentId == 0) {
|
|
|
+ queryWrapper.and(wrapper -> wrapper.isNull(StoreDictionary::getParentId)
|
|
|
+ .or().eq(StoreDictionary::getParentId, 0));
|
|
|
+ } else {
|
|
|
+ queryWrapper.eq(StoreDictionary::getParentId, parentId);
|
|
|
+ }
|
|
|
+
|
|
|
+ List<StoreDictionary> siblings = storeDictionaryMapper.selectList(queryWrapper);
|
|
|
+
|
|
|
+ if (siblings == null || siblings.isEmpty()) {
|
|
|
+ return;
|
|
|
+ }
|
|
|
+
|
|
|
+ // 上升则顺推：新dictId < 原dictId
|
|
|
+ if (newOrder < oldOrder) {
|
|
|
+ // 将新dictId到原dictId之间的记录的dictId都+1
|
|
|
+ for (StoreDictionary sibling : siblings) {
|
|
|
+ try {
|
|
|
+ int siblingOrder = Integer.parseInt(sibling.getDictId());
|
|
|
+ if (siblingOrder >= newOrder && siblingOrder < oldOrder) {
|
|
|
+ sibling.setDictId(String.valueOf(siblingOrder + 1));
|
|
|
+ sibling.setUpdatedTime(new Date());
|
|
|
+ storeDictionaryMapper.updateById(sibling);
|
|
|
+ }
|
|
|
+ } catch (NumberFormatException e) {
|
|
|
+ log.warn("无法解析dictId: {}", sibling.getDictId(), e);
|
|
|
+ }
|
|
|
+ }
|
|
|
+ }
|
|
|
+ // 下降则顺升：新dictId > 原dictId
|
|
|
+ else if (newOrder > oldOrder) {
|
|
|
+ // 将原dictId到新dictId之间的记录的dictId都-1
|
|
|
+ for (StoreDictionary sibling : siblings) {
|
|
|
+ try {
|
|
|
+ int siblingOrder = Integer.parseInt(sibling.getDictId());
|
|
|
+ if (siblingOrder > oldOrder && siblingOrder <= newOrder) {
|
|
|
+ sibling.setDictId(String.valueOf(siblingOrder - 1));
|
|
|
+ sibling.setUpdatedTime(new Date());
|
|
|
+ storeDictionaryMapper.updateById(sibling);
|
|
|
+ }
|
|
|
+ } catch (NumberFormatException e) {
|
|
|
+ log.warn("无法解析dictId: {}", sibling.getDictId(), e);
|
|
|
+ }
|
|
|
+ }
|
|
|
+ }
|
|
|
+ } catch (NumberFormatException e) {
|
|
|
+ log.error("调整排序失败，dictId格式错误: oldDictId={}, newDictId={}", oldDictId, newDictId, e);
|
|
|
+ } catch (Exception e) {
|
|
|
+ log.error("调整排序失败", e);
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 处理从原节点移出时的排序调整：原位置之后的记录需要前移（dictId - 1）
|
|
|
+ *
|
|
|
+ * @param currentId 当前记录ID
|
|
|
+ * @param oldParentId 原父节点ID
|
|
|
+ * @param oldDictId 原dictId
|
|
|
+ */
|
|
|
+ private void adjustSortOrderForMoveOut(Integer currentId, Integer oldParentId, String oldDictId) {
|
|
|
+ try {
|
|
|
+ int oldOrder = Integer.parseInt(oldDictId);
|
|
|
+
|
|
|
+ // 查询原节点同级所有记录（排除当前记录）
|
|
|
+ LambdaQueryWrapper<StoreDictionary> queryWrapper = new LambdaQueryWrapper<>();
|
|
|
+ queryWrapper.eq(StoreDictionary::getTypeName, "business_section");
|
|
|
+ queryWrapper.eq(StoreDictionary::getDeleteFlag, 0);
|
|
|
+ queryWrapper.ne(StoreDictionary::getId, currentId);
|
|
|
+
|
|
|
+ if (oldParentId == null || oldParentId == 0) {
|
|
|
+ queryWrapper.and(wrapper -> wrapper.isNull(StoreDictionary::getParentId)
|
|
|
+ .or().eq(StoreDictionary::getParentId, 0));
|
|
|
+ } else {
|
|
|
+ queryWrapper.eq(StoreDictionary::getParentId, oldParentId);
|
|
|
+ }
|
|
|
+
|
|
|
+ List<StoreDictionary> siblings = storeDictionaryMapper.selectList(queryWrapper);
|
|
|
+
|
|
|
+ if (siblings == null || siblings.isEmpty()) {
|
|
|
+ return;
|
|
|
+ }
|
|
|
+
|
|
|
+ // 原位置之后的记录需要前移（dictId - 1）
|
|
|
+ for (StoreDictionary sibling : siblings) {
|
|
|
+ try {
|
|
|
+ int siblingOrder = Integer.parseInt(sibling.getDictId());
|
|
|
+ if (siblingOrder > oldOrder) {
|
|
|
+ sibling.setDictId(String.valueOf(siblingOrder - 1));
|
|
|
+ sibling.setUpdatedTime(new Date());
|
|
|
+ storeDictionaryMapper.updateById(sibling);
|
|
|
+ }
|
|
|
+ } catch (NumberFormatException e) {
|
|
|
+ log.warn("无法解析dictId: {}", sibling.getDictId(), e);
|
|
|
+ }
|
|
|
+ }
|
|
|
+ } catch (NumberFormatException e) {
|
|
|
+ log.error("移出节点排序调整失败，dictId格式错误: oldDictId={}", oldDictId, e);
|
|
|
+ } catch (Exception e) {
|
|
|
+ log.error("移出节点排序调整失败", e);
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 处理移入新节点时的排序调整：新位置之后的记录需要后移（dictId + 1）
|
|
|
+ *
|
|
|
+ * @param currentId 当前记录ID
|
|
|
+ * @param newParentId 新父节点ID
|
|
|
+ * @param newDictId 新dictId
|
|
|
+ */
|
|
|
+ private void adjustSortOrderForMoveIn(Integer currentId, Integer newParentId, String newDictId) {
|
|
|
+ try {
|
|
|
+ int newOrder = Integer.parseInt(newDictId);
|
|
|
+
|
|
|
+ // 查询新节点同级所有记录（排除当前记录）
|
|
|
+ LambdaQueryWrapper<StoreDictionary> queryWrapper = new LambdaQueryWrapper<>();
|
|
|
+ queryWrapper.eq(StoreDictionary::getTypeName, "business_section");
|
|
|
+ queryWrapper.eq(StoreDictionary::getDeleteFlag, 0);
|
|
|
+ queryWrapper.ne(StoreDictionary::getId, currentId);
|
|
|
+
|
|
|
+ if (newParentId == null || newParentId == 0) {
|
|
|
+ queryWrapper.and(wrapper -> wrapper.isNull(StoreDictionary::getParentId)
|
|
|
+ .or().eq(StoreDictionary::getParentId, 0));
|
|
|
+ } else {
|
|
|
+ queryWrapper.eq(StoreDictionary::getParentId, newParentId);
|
|
|
+ }
|
|
|
+
|
|
|
+ List<StoreDictionary> siblings = storeDictionaryMapper.selectList(queryWrapper);
|
|
|
+
|
|
|
+ if (siblings == null || siblings.isEmpty()) {
|
|
|
+ return;
|
|
|
+ }
|
|
|
+
|
|
|
+ // 新位置之后的记录需要后移（dictId + 1）
|
|
|
+ for (StoreDictionary sibling : siblings) {
|
|
|
+ try {
|
|
|
+ int siblingOrder = Integer.parseInt(sibling.getDictId());
|
|
|
+ if (siblingOrder >= newOrder) {
|
|
|
+ sibling.setDictId(String.valueOf(siblingOrder + 1));
|
|
|
+ sibling.setUpdatedTime(new Date());
|
|
|
+ storeDictionaryMapper.updateById(sibling);
|
|
|
+ }
|
|
|
+ } catch (NumberFormatException e) {
|
|
|
+ log.warn("无法解析dictId: {}", sibling.getDictId(), e);
|
|
|
+ }
|
|
|
+ }
|
|
|
+ } catch (NumberFormatException e) {
|
|
|
+ log.error("移入节点排序调整失败，dictId格式错误: newDictId={}", newDictId, e);
|
|
|
+ } catch (Exception e) {
|
|
|
+ log.error("移入节点排序调整失败", e);
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 批量导入经营版块
|
|
|
+ *
|
|
|
+ * @param file Excel文件
|
|
|
+ * @return 导入结果
|
|
|
+ */
|
|
|
+ @Override
|
|
|
+ @Transactional(rollbackFor = Exception.class)
|
|
|
+ public R<String> importBusinessSection(MultipartFile file) {
|
|
|
+ log.info("PlatformBusinessSectionServiceImpl.importBusinessSection fileName={}",
|
|
|
+ file != null ? file.getOriginalFilename() : "null");
|
|
|
+ try {
|
|
|
+ if (file == null || file.isEmpty()) {
|
|
|
+ return R.fail("文件不能为空");
|
|
|
+ }
|
|
|
+
|
|
|
+ String fileName = file.getOriginalFilename();
|
|
|
+ if (fileName == null || (!fileName.endsWith(".xlsx") && !fileName.endsWith(".xls"))) {
|
|
|
+ return R.fail("文件格式不正确，请上传Excel文件");
|
|
|
+ }
|
|
|
+
|
|
|
+ List<String> errorMessages = new ArrayList<>();
|
|
|
+ int successCount = 0;
|
|
|
+ int totalCount = 0;
|
|
|
+
|
|
|
+ // 使用POI读取Excel
|
|
|
+ try (InputStream inputStream = file.getInputStream();
|
|
|
+ Workbook workbook = new XSSFWorkbook(inputStream)) {
|
|
|
+ Sheet sheet = workbook.getSheetAt(0);
|
|
|
+
|
|
|
+ // 获取表头
|
|
|
+ Row headerRow = sheet.getRow(0);
|
|
|
+ if (headerRow == null) {
|
|
|
+ return R.fail("Excel文件格式不正确，缺少表头");
|
|
|
+ }
|
|
|
+
|
|
|
+ // 构建字段映射（表头名称 -> 列索引）
|
|
|
+ Map<String, Integer> headerMap = new HashMap<>();
|
|
|
+ Field[] fields = BusinessSectionExcelVo.class.getDeclaredFields();
|
|
|
+ for (int i = 0; i < headerRow.getLastCellNum(); i++) {
|
|
|
+ Cell cell = headerRow.getCell(i);
|
|
|
+ if (cell != null) {
|
|
|
+ String headerName = getCellValueAsString(cell);
|
|
|
+ if (StringUtils.isNotBlank(headerName)) {
|
|
|
+ headerMap.put(headerName.trim(), i);
|
|
|
+ }
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ // 读取数据行
|
|
|
+ for (int rowIndex = 1; rowIndex <= sheet.getLastRowNum(); rowIndex++) {
|
|
|
+ Row row = sheet.getRow(rowIndex);
|
|
|
+ if (row == null) {
|
|
|
+ continue;
|
|
|
+ }
|
|
|
+
|
|
|
+ // 检查是否为空行
|
|
|
+ boolean isEmptyRow = true;
|
|
|
+ for (int i = 0; i < row.getLastCellNum(); i++) {
|
|
|
+ Cell cell = row.getCell(i);
|
|
|
+ if (cell != null && cell.getCellType() != CellType.BLANK) {
|
|
|
+ String cellValue = getCellValueAsString(cell);
|
|
|
+ if (StringUtils.isNotBlank(cellValue)) {
|
|
|
+ isEmptyRow = false;
|
|
|
+ break;
|
|
|
+ }
|
|
|
+ }
|
|
|
+ }
|
|
|
+ if (isEmptyRow) {
|
|
|
+ continue;
|
|
|
+ }
|
|
|
+
|
|
|
+ totalCount++;
|
|
|
+ BusinessSectionExcelVo excelVo = new BusinessSectionExcelVo();
|
|
|
+
|
|
|
+ // 读取每个字段
|
|
|
+ for (Field field : fields) {
|
|
|
+ if (!field.isAnnotationPresent(ExcelHeader.class)) {
|
|
|
+ continue;
|
|
|
+ }
|
|
|
+ ExcelHeader excelHeader = field.getAnnotation(ExcelHeader.class);
|
|
|
+ String headerName = excelHeader.value();
|
|
|
+ Integer colIndex = headerMap.get(headerName);
|
|
|
+ if (colIndex == null) {
|
|
|
+ continue;
|
|
|
+ }
|
|
|
+
|
|
|
+ Cell cell = row.getCell(colIndex);
|
|
|
+ if (cell == null) {
|
|
|
+ continue;
|
|
|
+ }
|
|
|
+
|
|
|
+ field.setAccessible(true);
|
|
|
+ try {
|
|
|
+ String cellValue = getCellValueAsString(cell);
|
|
|
+ if (StringUtils.isNotBlank(cellValue)) {
|
|
|
+ field.set(excelVo, cellValue.trim());
|
|
|
+ }
|
|
|
+ } catch (Exception e) {
|
|
|
+ log.warn("读取字段{}失败：{}", headerName, e.getMessage());
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ // 处理导入数据
|
|
|
+ try {
|
|
|
+ processImportData(excelVo, rowIndex + 1, errorMessages);
|
|
|
+ successCount++;
|
|
|
+ } catch (Exception e) {
|
|
|
+ errorMessages.add(String.format("第%d行：%s", rowIndex + 1, e.getMessage()));
|
|
|
+ log.error("导入第{}行数据失败", rowIndex + 1, e);
|
|
|
+ }
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ // 构建返回消息
|
|
|
+ StringBuilder message = new StringBuilder();
|
|
|
+ message.append(String.format("导入完成：成功%d条，失败%d条", successCount, totalCount - successCount));
|
|
|
+ if (!errorMessages.isEmpty()) {
|
|
|
+ message.append("\n失败详情：\n");
|
|
|
+ int maxErrors = Math.min(errorMessages.size(), 10); // 最多显示10条错误
|
|
|
+ for (int i = 0; i < maxErrors; i++) {
|
|
|
+ message.append(errorMessages.get(i)).append("\n");
|
|
|
+ }
|
|
|
+ if (errorMessages.size() > 10) {
|
|
|
+ message.append(String.format("...还有%d条错误未显示", errorMessages.size() - 10));
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ if (successCount == 0) {
|
|
|
+ return R.fail(message.toString());
|
|
|
+ } else if (totalCount - successCount > 0) {
|
|
|
+ return R.success(message.toString());
|
|
|
+ } else {
|
|
|
+ return R.success(message.toString());
|
|
|
+ }
|
|
|
+ } catch (Exception e) {
|
|
|
+ log.error("导入经营版块失败", e);
|
|
|
+ return R.fail("导入失败：" + e.getMessage());
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 处理导入数据
|
|
|
+ *
|
|
|
+ * @param excelVo Excel数据对象
|
|
|
+ * @param rowIndex 行号
|
|
|
+ * @param errorMessages 错误消息列表
|
|
|
+ */
|
|
|
+ private void processImportData(BusinessSectionExcelVo excelVo, int rowIndex, List<String> errorMessages) {
|
|
|
+ // 校验必填字段
|
|
|
+ if (StringUtils.isBlank(excelVo.getFirstLevelName())) {
|
|
|
+ throw new IllegalArgumentException("一级分类名称不能为空");
|
|
|
+ }
|
|
|
+
|
|
|
+ String firstLevelName = excelVo.getFirstLevelName().trim();
|
|
|
+ String secondLevelName = StringUtils.isNotBlank(excelVo.getSecondLevelName())
|
|
|
+ ? excelVo.getSecondLevelName().trim() : null;
|
|
|
+ String thirdLevelName = StringUtils.isNotBlank(excelVo.getThirdLevelName())
|
|
|
+ ? excelVo.getThirdLevelName().trim() : null;
|
|
|
+
|
|
|
+ // 确定要创建的级别
|
|
|
+ if (StringUtils.isNotBlank(thirdLevelName)) {
|
|
|
+ // 创建三级分类
|
|
|
+ createOrUpdateLevel(firstLevelName, secondLevelName, thirdLevelName, 3, excelVo.getHidden(), errorMessages, rowIndex);
|
|
|
+ } else if (StringUtils.isNotBlank(secondLevelName)) {
|
|
|
+ // 创建二级分类
|
|
|
+ createOrUpdateLevel(firstLevelName, secondLevelName, null, 2, excelVo.getHidden(), errorMessages, rowIndex);
|
|
|
+ } else {
|
|
|
+ // 创建一级分类
|
|
|
+ createOrUpdateLevel(firstLevelName, null, null, 1, excelVo.getHidden(), errorMessages, rowIndex);
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 创建或更新分类级别
|
|
|
+ *
|
|
|
+ * @param firstLevelName 一级分类名称
|
|
|
+ * @param secondLevelName 二级分类名称
|
|
|
+ * @param thirdLevelName 三级分类名称
|
|
|
+ * @param level 级别（1、2、3）
|
|
|
+ * @param hidden 状态
|
|
|
+ * @param errorMessages 错误消息列表
|
|
|
+ * @param rowIndex 行号
|
|
|
+ */
|
|
|
+ private void createOrUpdateLevel(String firstLevelName, String secondLevelName, String thirdLevelName,
|
|
|
+ int level, String hidden, List<String> errorMessages, int rowIndex) {
|
|
|
+ try {
|
|
|
+ // 查找或创建一级分类
|
|
|
+ StoreDictionary firstLevel = findOrCreateLevel(firstLevelName, 0, 0, hidden);
|
|
|
+ if (firstLevel == null) {
|
|
|
+ throw new IllegalArgumentException("创建一级分类失败");
|
|
|
+ }
|
|
|
+ if (level >= 2 && StringUtils.isNotBlank(secondLevelName)) {
|
|
|
+ // 查找或创建二级分类
|
|
|
+ StoreDictionary secondLevel = findOrCreateLevel(secondLevelName, firstLevel.getId(), 1, hidden);
|
|
|
+ if (secondLevel == null) {
|
|
|
+ throw new IllegalArgumentException("创建二级分类失败");
|
|
|
+ }
|
|
|
+
|
|
|
+ if (level >= 3 && StringUtils.isNotBlank(thirdLevelName)) {
|
|
|
+ // 查找或创建三级分类
|
|
|
+ StoreDictionary thirdLevel = findOrCreateLevel(thirdLevelName, secondLevel.getId(), 2, hidden);
|
|
|
+ if (thirdLevel == null) {
|
|
|
+ throw new IllegalArgumentException("创建三级分类失败");
|
|
|
+ }
|
|
|
+ }
|
|
|
+ }
|
|
|
+ } catch (Exception e) {
|
|
|
+ throw new IllegalArgumentException(e.getMessage());
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 查找或创建分类
|
|
|
+ *
|
|
|
+ * @param dictDetail 分类名称
|
|
|
+ * @param parentId 父节点ID
|
|
|
+ * @param expectedLevel 期望的级别（用于校验）
|
|
|
+ * @param hidden 状态
|
|
|
+ * @return 分类对象
|
|
|
+ */
|
|
|
+ private StoreDictionary findOrCreateLevel(String dictDetail, Integer parentId, int expectedLevel, String hidden) {
|
|
|
+ // 查找是否存在
|
|
|
+ LambdaQueryWrapper<StoreDictionary> queryWrapper = new LambdaQueryWrapper<>();
|
|
|
+ queryWrapper.eq(StoreDictionary::getTypeName, "business_section");
|
|
|
+ queryWrapper.eq(StoreDictionary::getDeleteFlag, 0);
|
|
|
+ queryWrapper.eq(StoreDictionary::getDictDetail, dictDetail);
|
|
|
+
|
|
|
+ if (parentId == null || parentId == 0) {
|
|
|
+ queryWrapper.and(wrapper -> wrapper.isNull(StoreDictionary::getParentId)
|
|
|
+ .or().eq(StoreDictionary::getParentId, 0));
|
|
|
+ } else {
|
|
|
+ queryWrapper.eq(StoreDictionary::getParentId, parentId);
|
|
|
+ }
|
|
|
+
|
|
|
+ StoreDictionary existing = storeDictionaryMapper.selectOne(queryWrapper);
|
|
|
+
|
|
|
+ if (existing != null) {
|
|
|
+ return existing;
|
|
|
+ }
|
|
|
+
|
|
|
+ // 创建新分类
|
|
|
+ StoreDictionary newDict = new StoreDictionary();
|
|
|
+ newDict.setTypeName("business_section");
|
|
|
+ newDict.setDictDetail(dictDetail);
|
|
|
+ // 一级为经营版块 ，二级为经营种类，三级为分类
|
|
|
+ if (expectedLevel == 0) {
|
|
|
+ newDict.setTypeDetail("经营板块");
|
|
|
+ } else if (expectedLevel == 1) {
|
|
|
+ newDict.setTypeDetail("经营种类");
|
|
|
+ } else if (expectedLevel == 2) {
|
|
|
+ newDict.setTypeDetail("分类");
|
|
|
+ }
|
|
|
+ newDict.setParentId(parentId == null ? 0 : parentId);
|
|
|
+ newDict.setHidden(StringUtils.isNotBlank(hidden) && hidden.equals("隐藏") ? 1 : 0);
|
|
|
+ newDict.setDeleteFlag(0);
|
|
|
+ newDict.setCreatedTime(new Date());
|
|
|
+
|
|
|
+ // 生成 dict_id
|
|
|
+ String maxDictId = getMaxDictIdByParentId(newDict.getParentId());
|
|
|
+ /*int nextDictId;
|
|
|
+ if (StringUtils.isNotBlank(sortOrder) && expectedLevel == 1) {
|
|
|
+ try {
|
|
|
+ nextDictId = Integer.parseInt(sortOrder);
|
|
|
+ // 如果指定的排序已存在，需要调整
|
|
|
+ String existingDictId = getDictIdByParentIdAndOrder(newDict.getParentId(), nextDictId);
|
|
|
+ if (existingDictId != null) {
|
|
|
+ // 使用最大+1
|
|
|
+ nextDictId = (maxDictId == null ? 0 : Integer.parseInt(maxDictId)) + 1;
|
|
|
+ }
|
|
|
+ } catch (NumberFormatException e) {
|
|
|
+ nextDictId = (maxDictId == null ? 0 : Integer.parseInt(maxDictId)) + 1;
|
|
|
+ }
|
|
|
+ } else {
|
|
|
+ nextDictId = (maxDictId == null ? 0 : Integer.parseInt(maxDictId)) + 1;
|
|
|
+ }*/
|
|
|
+ newDict.setDictId(String.valueOf(maxDictId == null ? 1 : Integer.parseInt(maxDictId) + 1));
|
|
|
+
|
|
|
+ boolean saved = this.save(newDict);
|
|
|
+ return saved ? newDict : null;
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 根据parentId和排序获取dictId
|
|
|
+ *
|
|
|
+ * @param parentId 父节点ID
|
|
|
+ * @param order 排序
|
|
|
+ * @return dictId
|
|
|
+ */
|
|
|
+ private String getDictIdByParentIdAndOrder(Integer parentId, int order) {
|
|
|
+ LambdaQueryWrapper<StoreDictionary> queryWrapper = new LambdaQueryWrapper<>();
|
|
|
+ queryWrapper.eq(StoreDictionary::getTypeName, "business_section");
|
|
|
+ queryWrapper.eq(StoreDictionary::getDeleteFlag, 0);
|
|
|
+ queryWrapper.eq(StoreDictionary::getDictId, String.valueOf(order));
|
|
|
+
|
|
|
+ if (parentId == null || parentId == 0) {
|
|
|
+ queryWrapper.and(wrapper -> wrapper.isNull(StoreDictionary::getParentId)
|
|
|
+ .or().eq(StoreDictionary::getParentId, 0));
|
|
|
+ } else {
|
|
|
+ queryWrapper.eq(StoreDictionary::getParentId, parentId);
|
|
|
+ }
|
|
|
+
|
|
|
+ StoreDictionary dict = storeDictionaryMapper.selectOne(queryWrapper);
|
|
|
+ return dict != null ? dict.getDictId() : null;
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 获取单元格值（字符串格式）
|
|
|
+ *
|
|
|
+ * @param cell 单元格
|
|
|
+ * @return 字符串值
|
|
|
+ */
|
|
|
+ private String getCellValueAsString(Cell cell) {
|
|
|
+ if (cell == null) {
|
|
|
+ return null;
|
|
|
+ }
|
|
|
+
|
|
|
+ switch (cell.getCellType()) {
|
|
|
+ case STRING:
|
|
|
+ return cell.getStringCellValue();
|
|
|
+ case NUMERIC:
|
|
|
+ if (DateUtil.isCellDateFormatted(cell)) {
|
|
|
+ return cell.getDateCellValue().toString();
|
|
|
+ } else {
|
|
|
+ // 处理数字，避免科学计数法
|
|
|
+ double numericValue = cell.getNumericCellValue();
|
|
|
+ if (numericValue == (long) numericValue) {
|
|
|
+ return String.valueOf((long) numericValue);
|
|
|
+ } else {
|
|
|
+ return String.valueOf(numericValue);
|
|
|
+ }
|
|
|
+ }
|
|
|
+ case BOOLEAN:
|
|
|
+ return String.valueOf(cell.getBooleanCellValue());
|
|
|
+ case FORMULA:
|
|
|
+ return cell.getCellFormula();
|
|
|
+ default:
|
|
|
+ return null;
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 下载经营版块导入模板
|
|
|
+ *
|
|
|
+ * @param response HTTP响应
|
|
|
+ * @throws IOException IO异常
|
|
|
+ */
|
|
|
+ @Override
|
|
|
+ public void downloadTemplate(HttpServletResponse response) throws IOException {
|
|
|
+ log.info("PlatformBusinessSectionServiceImpl.downloadTemplate");
|
|
|
+
|
|
|
+ XSSFWorkbook workbook = new XSSFWorkbook();
|
|
|
+ try {
|
|
|
+ Sheet sheet = workbook.createSheet("经营版块导入模板");
|
|
|
+
|
|
|
+ // 获取带有 @ExcelHeader 注解的字段
|
|
|
+ Field[] fields = BusinessSectionExcelVo.class.getDeclaredFields();
|
|
|
+ List<Field> excelFields = new ArrayList<>();
|
|
|
+ for (Field field : fields) {
|
|
|
+ if (field.isAnnotationPresent(ExcelHeader.class)) {
|
|
|
+ excelFields.add(field);
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ // 创建表头样式
|
|
|
+ Font headerFont = workbook.createFont();
|
|
|
+ headerFont.setBold(true);
|
|
|
+ headerFont.setFontHeightInPoints((short) 12);
|
|
|
+ CellStyle headerCellStyle = workbook.createCellStyle();
|
|
|
+ headerCellStyle.setFont(headerFont);
|
|
|
+ headerCellStyle.setAlignment(HorizontalAlignment.CENTER);
|
|
|
+ headerCellStyle.setVerticalAlignment(VerticalAlignment.CENTER);
|
|
|
+ headerCellStyle.setFillForegroundColor(IndexedColors.GREY_25_PERCENT.getIndex());
|
|
|
+ headerCellStyle.setFillPattern(FillPatternType.SOLID_FOREGROUND);
|
|
|
+ headerCellStyle.setBorderBottom(BorderStyle.THIN);
|
|
|
+ headerCellStyle.setBorderTop(BorderStyle.THIN);
|
|
|
+ headerCellStyle.setBorderLeft(BorderStyle.THIN);
|
|
|
+ headerCellStyle.setBorderRight(BorderStyle.THIN);
|
|
|
+
|
|
|
+ // 创建表头
|
|
|
+ Row headerRow = sheet.createRow(0);
|
|
|
+ for (int i = 0; i < excelFields.size(); i++) {
|
|
|
+ Field field = excelFields.get(i);
|
|
|
+ ExcelHeader excelHeader = field.getAnnotation(ExcelHeader.class);
|
|
|
+ Cell cell = headerRow.createCell(i);
|
|
|
+ cell.setCellValue(excelHeader.value());
|
|
|
+ cell.setCellStyle(headerCellStyle);
|
|
|
+ }
|
|
|
+
|
|
|
+ // 设置列宽
|
|
|
+ sheet.setColumnWidth(0, 25 * 256); // 一级分类名称
|
|
|
+ sheet.setColumnWidth(1, 25 * 256); // 二级分类名称
|
|
|
+ sheet.setColumnWidth(2, 25 * 256); // 三级分类名称
|
|
|
+ sheet.setColumnWidth(3, 15 * 256); // 状态（隐藏/显示）
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+ // 设置响应头（必须在获取输出流之前设置）
|
|
|
+ response.setContentType("application/vnd.openxmlformats-officedocument.spreadsheetml.sheet");
|
|
|
+ response.setCharacterEncoding("utf-8");
|
|
|
+ String fileName = URLEncoder.encode("经营版块导入模板", "UTF-8").replaceAll("\\+", "%20");
|
|
|
+ response.setHeader("Content-disposition", "attachment;filename*=utf-8''" + fileName + ".xlsx");
|
|
|
+
|
|
|
+ // 输出到响应流
|
|
|
+ OutputStream outputStream = response.getOutputStream();
|
|
|
+ workbook.write(outputStream);
|
|
|
+ outputStream.flush();
|
|
|
+ } finally {
|
|
|
+ // 确保 workbook 被正确关闭
|
|
|
+ if (workbook != null) {
|
|
|
+ try {
|
|
|
+ workbook.close();
|
|
|
+ } catch (IOException e) {
|
|
|
+ log.error("关闭workbook失败", e);
|
|
|
+ }
|
|
|
+ }
|
|
|
+ }
|
|
|
+ }
|
|
|
+}
|
|
|
+
|
penghao