From c600eb961aee6d178271d334beac7dd19f8fc289 Mon Sep 17 00:00:00 2001
From: “djh” <“3298565835@qq.com”>
Date: 星期五, 17 十月 2025 09:13:22 +0800
Subject: [PATCH] 修改
---
multi-system/src/main/java/com/gkhy/exam/system/service/impl/CompanyIndustryTemplateServiceImpl.java | 138 +++++++++++++++++++++++++++++++++++++++++++--
1 files changed, 131 insertions(+), 7 deletions(-)
diff --git a/multi-system/src/main/java/com/gkhy/exam/system/service/impl/CompanyIndustryTemplateServiceImpl.java b/multi-system/src/main/java/com/gkhy/exam/system/service/impl/CompanyIndustryTemplateServiceImpl.java
index 4521dc5..1faff8d 100644
--- a/multi-system/src/main/java/com/gkhy/exam/system/service/impl/CompanyIndustryTemplateServiceImpl.java
+++ b/multi-system/src/main/java/com/gkhy/exam/system/service/impl/CompanyIndustryTemplateServiceImpl.java
@@ -1,5 +1,7 @@
package com.gkhy.exam.system.service.impl;
+import cn.hutool.core.collection.ListUtil;
+import com.alibaba.fastjson2.JSONObject;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.gkhy.exam.common.api.CommonPage;
import com.gkhy.exam.common.api.CommonResult;
@@ -7,18 +9,27 @@
import com.gkhy.exam.common.exception.ApiException;
import com.gkhy.exam.common.utils.PageUtils;
import com.gkhy.exam.common.utils.SecurityUtils;
-import com.gkhy.exam.system.domain.CompanyIndustryTemplate;
-import com.gkhy.exam.system.domain.CompanyRoster;
-import com.gkhy.exam.system.domain.SysCompany;
+import com.gkhy.exam.common.utils.StringUtils;
+import com.gkhy.exam.system.domain.*;
import com.gkhy.exam.system.mapper.CompanyIndustryTemplateMapper;
import com.gkhy.exam.system.mapper.CompanyRosterMapper;
+import com.gkhy.exam.system.mapper.SysIndustryTypeMapper;
import com.gkhy.exam.system.service.CompanyIndustryTemplateService;
import com.gkhy.exam.system.service.SysCompanyService;
+import com.gkhy.exam.system.service.SysIndustryTypeService;
+import org.apache.poi.ss.usermodel.*;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
+import org.springframework.web.multipart.MultipartFile;
+import java.io.IOException;
import java.time.LocalDateTime;
+import java.util.ArrayList;
+import java.util.HashMap;
import java.util.List;
+import java.util.Map;
+
+import static cn.hutool.poi.excel.cell.CellUtil.getCellValue;
@Service
public class CompanyIndustryTemplateServiceImpl extends ServiceImpl<CompanyIndustryTemplateMapper, CompanyIndustryTemplate> implements CompanyIndustryTemplateService {
@@ -28,11 +39,14 @@
@Autowired
private SysCompanyService sysCompanyService;
+ @Autowired
+ private SysIndustryTypeMapper sysIndustryTypeMapper;
+
@Override
- public CommonPage selectCompanyIndustryTemplateList(Integer companyId) {
- boolean admin = SecurityUtils.isAdmin(SecurityUtils.getUserId());
+ public CommonPage selectCompanyIndustryTemplateList(CompanyIndustryTemplate companyId) {
+ boolean admin = SecurityUtils.adminUser();
if (!admin){
if (companyId==null){
throw new ApiException("非管理员操作，查询条件不可为空");
@@ -47,7 +61,7 @@
public CommonResult insertCompanyIndustryTemplate(CompanyIndustryTemplate companyIndustryTemplate) {
LoginUserDetails loginUser = SecurityUtils.getLoginUser();
- SysCompany sysCompany = sysCompanyService.selectCompanyById(SecurityUtils.getCompanyId());
+ SysCompany sysCompany = sysCompanyService.selectCompanyById(companyIndustryTemplate.getCompanyId()==null?SecurityUtils.getCompanyId():companyIndustryTemplate.getCompanyId());
companyIndustryTemplate.setCompanyName(sysCompany.getName());
companyIndustryTemplate.setCreateBy(loginUser.getUsername());
companyIndustryTemplate.setCreateTime(LocalDateTime.now());
@@ -61,7 +75,7 @@
@Override
public CommonResult updateCompanyIndustryTemplate(CompanyIndustryTemplate companyIndustryTemplate) {
LoginUserDetails loginUser = SecurityUtils.getLoginUser();
- SysCompany sysCompany = sysCompanyService.selectCompanyById(SecurityUtils.getCompanyId());
+ SysCompany sysCompany = sysCompanyService.selectCompanyById(companyIndustryTemplate.getCompanyId()==null?SecurityUtils.getCompanyId():companyIndustryTemplate.getCompanyId());
companyIndustryTemplate.setCompanyName(sysCompany.getName());
companyIndustryTemplate.setUpdateBy(loginUser.getUsername());
companyIndustryTemplate.setUpdateTime(LocalDateTime.now());
@@ -86,4 +100,114 @@
}
return CommonResult.failed();
}
+
+ @Override
+ public CommonResult uploadQuestion(MultipartFile file) throws IOException {
+ List<CompanyIndustryType> companyIndustryTypes = sysIndustryTypeMapper.selectIndustryTypeList();
+ List<SysCompany> list = sysCompanyService.selectCompanyLists();
+ Workbook workbook = WorkbookFactory.create(file.getInputStream());
+ Sheet sheet = workbook.getSheetAt(0);
+
+// 1. 创建列名-索引映射表
+ Map<String, Integer> columnIndexMap = new HashMap<>();
+ Row headerRow = sheet.getRow(0);
+ for (Cell cell : headerRow) {
+ String headerName = cell.getStringCellValue().trim();
+ columnIndexMap.put(headerName, cell.getColumnIndex());
+ }
+
+// 验证必要列是否存在
+ String[] requiredHeaders = {"章节", "文件记录", "分类", "行业", "企业名称"};
+ List<String> missingHeaders = new ArrayList<>();
+ for (String header : requiredHeaders) {
+ if (!columnIndexMap.containsKey(header)) {
+ missingHeaders.add(header);
+ }
+ }
+ if (!missingHeaders.isEmpty()) {
+ workbook.close();
+ return CommonResult.failed("缺少必要表头: " + String.join(", ", missingHeaders));
+ }
+ List<CompanyIndustryTemplate> companyIndustryTemplates = new ArrayList<>();
+ LoginUserDetails loginUser = SecurityUtils.getLoginUser();
+ DataFormatter dataFormatter = new DataFormatter();
+ StringBuilder errorBuilder = new StringBuilder();
+ // 2. 获取各列的索引
+ int chapterIndex = columnIndexMap.get("章节");
+ int templateNameIndex = columnIndexMap.get("文件记录");
+ int typeIndex = columnIndexMap.get("分类");
+ int industryTypeIndex = columnIndexMap.get("行业");
+ int companyNameIndex = columnIndexMap.get("企业名称");
+ // 3. 从第二行开始处理数据
+ for (int i = 1; i <= sheet.getLastRowNum(); i++) {
+ Row row = sheet.getRow(i);
+ if (row == null) continue;
+
+ CompanyIndustryTemplate template = new CompanyIndustryTemplate();
+ template.setCreateBy(loginUser.getUsername());
+ template.setCreateTime(LocalDateTime.now());
+
+ // 4. 使用动态索引获取单元格
+ template.setChapter(getCellValue(row, chapterIndex, dataFormatter));
+ template.setTemplateName(getCellValue(row, templateNameIndex, dataFormatter));
+ template.setType(getCellValue(row, typeIndex, dataFormatter));
+
+ // 5. 处理行业类型映射
+ String industryName = getCellValue(row, industryTypeIndex, dataFormatter);
+ boolean industryFound = false;
+ for (CompanyIndustryType type : companyIndustryTypes) {
+ if (industryName.equals(type.getName())) {
+ template.setIndustryType(type.getId());
+ industryFound = true;
+ break;
+ }
+ }
+ if (!industryFound) {
+ errorBuilder.append("第").append(i + 1).append("行: 未找到行业[").append(industryName).append("]; ");
+ }
+
+ // 6. 处理企业映射
+ String companyName = getCellValue(row, companyNameIndex, dataFormatter);
+ boolean companyFound = false;
+ for (SysCompany company : list) {
+ if (companyName.equals(company.getName())) {
+ template.setCompanyId(company.getId());
+ companyFound = true;
+ break;
+ }
+ }
+ if (!companyFound) {
+ errorBuilder.append("第").append(i + 1).append("行: 未找到企业[").append(companyName).append("]; ");
+ }
+ if (SecurityUtils.getCompanyId()!=null){
+ template.setCompanyId(SecurityUtils.getCompanyId());
+ }
+
+ companyIndustryTemplates.add(template);
+ }
+
+ // 7. 错误处理
+ if (errorBuilder.length() > 0) {
+ workbook.close();
+ return CommonResult.failed(errorBuilder.toString());
+ }
+
+ List<List<CompanyIndustryTemplate>> split = ListUtil.split(companyIndustryTemplates, 50);
+ for (List<CompanyIndustryTemplate> industryTemplates : split) {
+ int affectedRows = companyIndustryTemplateMapper.insertIndustrys(industryTemplates);
+ }
+
+ workbook.close();
+
+ return CommonResult.success();
+ }
+ // 安全获取单元格值
+ private String getCellValue(Row row, int columnIndex, DataFormatter formatter) {
+ if (columnIndex < 0) return "";
+ Cell cell = row.getCell(columnIndex, Row.MissingCellPolicy.CREATE_NULL_AS_BLANK);
+ return formatter.formatCellValue(cell).trim();
+ }
+
+
+
}
--
Gitblit v1.9.2