From 6b652d0e9269156936a1d6425829e104b7e680b5 Mon Sep 17 00:00:00 2001
From: heheng <475597332@qq.com>
Date: 星期五, 14 十一月 2025 13:21:48 +0800
Subject: [PATCH] 功能修改

---
 multi-system/src/main/java/com/gkhy/exam/system/service/impl/ExternalKnowledgeServiceImpl.java |   75 ++++++++++++++++++++++++++++++++++++-
 1 files changed, 73 insertions(+), 2 deletions(-)

diff --git a/multi-system/src/main/java/com/gkhy/exam/system/service/impl/ExternalKnowledgeServiceImpl.java b/multi-system/src/main/java/com/gkhy/exam/system/service/impl/ExternalKnowledgeServiceImpl.java
index f86b5b0..4ae57a7 100644
--- a/multi-system/src/main/java/com/gkhy/exam/system/service/impl/ExternalKnowledgeServiceImpl.java
+++ b/multi-system/src/main/java/com/gkhy/exam/system/service/impl/ExternalKnowledgeServiceImpl.java
@@ -1,19 +1,29 @@
 package com.gkhy.exam.system.service.impl;
 
+import com.alibaba.excel.EasyExcel;
+import com.alibaba.excel.ExcelReader;
+import com.alibaba.excel.context.AnalysisContext;
+import com.alibaba.excel.read.listener.ReadListener;
+import com.alibaba.excel.read.metadata.ReadSheet;
 import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper;
 import com.gkhy.exam.common.api.CommonPage;
 import com.gkhy.exam.common.api.CommonResult;
+import com.gkhy.exam.common.api.ImportResult;
 import com.gkhy.exam.common.constant.UserConstant;
 import com.gkhy.exam.common.utils.PageUtils;
 import com.gkhy.exam.common.utils.SecurityUtils;
+import com.gkhy.exam.system.domain.DTO.KnowledgeImportDTO;
 import com.gkhy.exam.system.domain.ExternalKnowledge;
 import com.gkhy.exam.system.mapper.ExternalKnowledgeMapper;
 import com.gkhy.exam.system.service.ExternalKnowledgeService;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
+import org.springframework.transaction.annotation.Transactional;
+import org.springframework.web.multipart.MultipartFile;
 
 import java.time.LocalDateTime;
+import java.util.ArrayList;
 import java.util.List;
 
 /**
@@ -41,12 +51,12 @@
     @Override
     public CommonResult saveExternalKnowledge(ExternalKnowledge externalKnowledge) {
         int i = 0;
-        if (externalKnowledge.getId() == null){
+        if (externalKnowledge.getId() == null) {
             externalKnowledge.setCreateBy(SecurityUtils.getUsername());
             externalKnowledge.setCreateTime(LocalDateTime.now());
             i = externalKnowledgeMapper.insert(externalKnowledge);
 
-        }else {
+        } else {
             externalKnowledge.setUpdateBy(SecurityUtils.getUsername());
             externalKnowledge.setUpdateTime(LocalDateTime.now());
             i = externalKnowledgeMapper.updateById(externalKnowledge);
@@ -63,4 +73,65 @@
                         .set(ExternalKnowledge::getUpdateBy, SecurityUtils.getUsername()));
         return CommonResult.success();
     }
+
+    @Override
+    @Transactional
+    public ImportResult importExternalKnowledge(Long companyId, MultipartFile file) {
+        ImportResult result = new ImportResult();
+        result.setSuccessCount(0);
+        result.setFailCount(0);
+        result.setErrorMessages(new ArrayList<>());
+
+
+        try {
+            // 使用EasyExcel读取文件
+            ReadListener<KnowledgeImportDTO> listener = new ReadListener<KnowledgeImportDTO>() {
+                @Override
+                public void invoke(KnowledgeImportDTO dto, AnalysisContext context) {
+                    try {
+                        // 转换为实体对象
+                        ExternalKnowledge externalKnowledge = new ExternalKnowledge();
+                        externalKnowledge.setName(dto.getName());
+                        externalKnowledge.setNumber(dto.getNumber());
+                        switch (dto.getKnowledgeType()) {
+                            case "管理":
+                                externalKnowledge.setKnowledgeType(1);
+                                break;
+                            case "技术":
+                                externalKnowledge.setKnowledgeType(2);
+                                break;
+                            case "质量服务标准":
+                                externalKnowledge.setKnowledgeType(3);
+                                break;
+                            default:
+                                externalKnowledge.setKnowledgeType(4);
+                                break;
+                        }
+                        externalKnowledge.setCompanyId(companyId);
+                        externalKnowledgeMapper.insert(externalKnowledge);
+                        result.setSuccessCount(result.getSuccessCount() + 1);
+                    } catch (Exception e) {
+                        result.getErrorMessages().add("行" + context.readRowHolder().getRowIndex() + "导入失败:" + e.getMessage());
+                        result.setFailCount(result.getFailCount() + 1);
+                    }
+                }
+
+
+                @Override
+                public void doAfterAllAnalysed(AnalysisContext context) {
+                    // 导入完成后的处理
+                }
+            };
+
+            // 创建读取器
+            ExcelReader excelReader = EasyExcel.read(file.getInputStream(), KnowledgeImportDTO.class, listener).build();
+            ReadSheet readSheet = EasyExcel.readSheet(0).build();
+            excelReader.read(readSheet);
+
+        } catch (Exception e) {
+            result.getErrorMessages().add("导入失败:" + e.getMessage());
+        }
+
+        return result;
+    }
 }

--
Gitblit v1.9.2