From cbe4b354224dfc34a852c7b7f932d33bbe85ffec Mon Sep 17 00:00:00 2001
From: Administrator <15274802129@163.com>
Date: Tue, 03 Feb 2026 15:19:39 +0800
Subject: [PATCH] feat(ai): 新增产品题目导入导出功能
---
src/main/java/cc/mrbird/febs/ai/service/AiProductQuestionService.java | 7 +
src/main/java/cc/mrbird/febs/ai/entity/AiProductQuestion.java | 3
src/main/java/cc/mrbird/febs/ai/service/impl/AiProductQuestionServiceImpl.java | 179 +++++++++++++++++++++++++++++++++++
src/main/java/cc/mrbird/febs/ai/controller/productQuestion/AiProductQuestionController.java | 33 ++++++
src/main/resources/templates/febs/views/modules/ai/productQuestion/list.html | 30 +++++
5 files changed, 250 insertions(+), 2 deletions(-)
diff --git a/src/main/java/cc/mrbird/febs/ai/controller/productQuestion/AiProductQuestionController.java b/src/main/java/cc/mrbird/febs/ai/controller/productQuestion/AiProductQuestionController.java
index 67031b3..075ec0a 100644
--- a/src/main/java/cc/mrbird/febs/ai/controller/productQuestion/AiProductQuestionController.java
+++ b/src/main/java/cc/mrbird/febs/ai/controller/productQuestion/AiProductQuestionController.java
@@ -1,6 +1,7 @@
package cc.mrbird.febs.ai.controller.productQuestion;
import cc.mrbird.febs.ai.entity.AiProductQuestion;
+import cc.mrbird.febs.ai.entity.AiProductQuestionItem;
import cc.mrbird.febs.ai.entity.AiProductQuestionJob;
import cc.mrbird.febs.ai.req.AiProductQuestionAiDto;
import cc.mrbird.febs.ai.service.AiProductQuestionService;
@@ -8,13 +9,33 @@
import cc.mrbird.febs.common.controller.BaseController;
import cc.mrbird.febs.common.entity.FebsResponse;
import cc.mrbird.febs.common.entity.QueryRequest;
+import cc.mrbird.febs.common.utils.excl.ExcelSheetPO;
+import cc.mrbird.febs.common.utils.excl.ExcelUtil;
+import cc.mrbird.febs.common.utils.excl.ExcelVersion;
+import cc.mrbird.febs.common.utils.excl.ResponseHeadUtil;
+import cc.mrbird.febs.mall.dto.DeliverGoodsDto;
+import cc.mrbird.febs.mall.entity.MallOrderInfo;
+import cc.mrbird.febs.mall.entity.MallOrderItem;
+import cn.hutool.core.collection.CollUtil;
+import cn.hutool.core.date.DateUtil;
+import cn.hutool.core.util.ObjectUtil;
+import cn.hutool.core.util.StrUtil;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.validation.annotation.Validated;
import org.springframework.web.bind.annotation.*;
+import org.springframework.web.multipart.MultipartFile;
+import javax.servlet.http.HttpServletResponse;
import javax.validation.Valid;
import javax.validation.constraints.NotNull;
+import java.io.File;
+import java.io.IOException;
+import java.io.OutputStream;
+import java.net.URLEncoder;
+import java.util.ArrayList;
+import java.util.Date;
+import java.util.List;
import java.util.Map;
/**
@@ -113,4 +134,16 @@
String companyId = getCurrentUserCompanyId();
return new FebsResponse().success().data(aiProductQuestionService.questionTree(companyId));
}
+
+ @GetMapping("exportProductQuestion")
+ @ControllerEndpoint(operation = "导出", exceptionMessage = "操作失败")
+ public FebsResponse exportProductQuestion(AiProductQuestion dto, HttpServletResponse response) throws IOException {
+ aiProductQuestionService.exportProductQuestion(dto, response);
+ return null;
+ }
+ @PostMapping(value = "/importProductQuestion")
+ @ControllerEndpoint(operation = "导入", exceptionMessage = "操作失败")
+ public FebsResponse importDeliver(@RequestBody MultipartFile file){
+ return aiProductQuestionService.importDeliver(file);
+ }
}
diff --git a/src/main/java/cc/mrbird/febs/ai/entity/AiProductQuestion.java b/src/main/java/cc/mrbird/febs/ai/entity/AiProductQuestion.java
index 0660101..dec6b31 100644
--- a/src/main/java/cc/mrbird/febs/ai/entity/AiProductQuestion.java
+++ b/src/main/java/cc/mrbird/febs/ai/entity/AiProductQuestion.java
@@ -44,4 +44,7 @@
@TableField(exist = false)
private List<AiProductQuestionItem> aiProductQuestionItems;
+
+ @TableField(exist = false)
+ private String ids;
}
diff --git a/src/main/java/cc/mrbird/febs/ai/service/AiProductQuestionService.java b/src/main/java/cc/mrbird/febs/ai/service/AiProductQuestionService.java
index bc75b00..7da398f 100644
--- a/src/main/java/cc/mrbird/febs/ai/service/AiProductQuestionService.java
+++ b/src/main/java/cc/mrbird/febs/ai/service/AiProductQuestionService.java
@@ -9,6 +9,9 @@
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.extension.service.IService;
+import org.springframework.web.multipart.MultipartFile;
+
+import javax.servlet.http.HttpServletResponse;
import java.util.List;
/**
@@ -51,4 +54,8 @@
void getAddQuestion(String id);
+
+ void exportProductQuestion(AiProductQuestion dto, HttpServletResponse response);
+
+ FebsResponse importDeliver(MultipartFile file);
}
diff --git a/src/main/java/cc/mrbird/febs/ai/service/impl/AiProductQuestionServiceImpl.java b/src/main/java/cc/mrbird/febs/ai/service/impl/AiProductQuestionServiceImpl.java
index c952b28..939d015 100644
--- a/src/main/java/cc/mrbird/febs/ai/service/impl/AiProductQuestionServiceImpl.java
+++ b/src/main/java/cc/mrbird/febs/ai/service/impl/AiProductQuestionServiceImpl.java
@@ -11,9 +11,17 @@
import cc.mrbird.febs.common.entity.FebsResponse;
import cc.mrbird.febs.common.entity.QueryRequest;
import cc.mrbird.febs.common.exception.FebsException;
+import cc.mrbird.febs.common.utils.excl.ExcelSheetPO;
+import cc.mrbird.febs.common.utils.excl.ExcelUtil;
+import cc.mrbird.febs.common.utils.excl.ExcelVersion;
+import cc.mrbird.febs.common.utils.excl.ResponseHeadUtil;
+import cc.mrbird.febs.mall.dto.DeliverGoodsDto;
import cc.mrbird.febs.mall.entity.MallMember;
+import cc.mrbird.febs.mall.entity.MallOrderInfo;
+import cc.mrbird.febs.mall.entity.MallOrderItem;
import cc.mrbird.febs.rabbit.producter.AgentProducer;
import cn.hutool.core.collection.CollUtil;
+import cn.hutool.core.date.DateUtil;
import cn.hutool.core.util.ObjectUtil;
import cn.hutool.core.util.StrUtil;
import cn.hutool.crypto.SecureUtil;
@@ -33,7 +41,14 @@
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
+import org.springframework.web.multipart.MultipartFile;
+import javax.servlet.http.HttpServletResponse;
+import java.io.File;
+import java.io.IOException;
+import java.io.OutputStream;
+import java.net.URLEncoder;
+import java.util.ArrayList;
import java.util.Date;
import java.util.List;
@@ -401,6 +416,170 @@
}
}
+ @Override
+ public void exportProductQuestion(AiProductQuestion dto, HttpServletResponse response) {
+ try {
+ List<ExcelSheetPO> res = new ArrayList<>();
+ ExcelSheetPO orderSheet = new ExcelSheetPO();
+ String title = "单选题目列表";
+ orderSheet.setSheetName(title);
+ orderSheet.setTitle(title);
+ String[] header = {"答案ID", "题目ID", "题目", "答案", "正确答案", "解析"};
+ orderSheet.setHeaders(header);
+
+ QueryRequest request = new QueryRequest();
+ request.setPageNum(1);
+ request.setPageSize(9999);
+ String questionIds = dto.getIds();
+ List<String> ids = StrUtil.splitTrim(questionIds, ",");
+ List<AiProductQuestionItem> aiProductQuestionItems = aiProductQuestionItemService.getListByQuery(
+ Wrappers.lambdaQuery(AiProductQuestionItem.class)
+ .in(AiProductQuestionItem::getProductQuestionId, ids)
+ .orderByAsc(AiProductQuestionItem::getProductQuestionId)
+ );
+
+ List<List<Object>> list = new ArrayList<>();
+ if (CollUtil.isNotEmpty(aiProductQuestionItems)) {
+ for (AiProductQuestionItem item : aiProductQuestionItems) {
+ List<Object> temp = new ArrayList<>();
+ temp.add(item.getId());
+ temp.add(item.getProductQuestionId());
+ temp.add(item.getTitle());
+ temp.add(item.getAnswer());
+ temp.add(item.getCorrectAnswer() == 1 ? "是" : "否");
+ temp.add(item.getAnswerAnalysis());
+ list.add(temp);
+ }
+ }
+ orderSheet.setDataList(list);
+ res.add(orderSheet);
+
+ // 设置响应头
+ response = ResponseHeadUtil.setExcelHead(response);
+ String fileName = title + DateUtil.format(new Date(), "yyyyMMddHHmmss") + ".xlsx";
+ response.setHeader("Content-Disposition",
+ "attachment;filename=" + URLEncoder.encode(fileName, "UTF-8"));
+
+ // 写入 Excel 文件
+ try (OutputStream os = response.getOutputStream()) {
+ ExcelUtil.createWorkbookAtOutStream(ExcelVersion.V2007, res, os, true);
+ }
+ } catch (Exception e) {
+ log.error("导出产品题目失败", e);
+ throw new RuntimeException("导出失败，请稍后重试");
+ }
+ }
+
+ @Override
+ @Transactional
+ public FebsResponse importDeliver(MultipartFile file) {
+ try {
+ if (file.isEmpty()) {
+ return new FebsResponse().fail();
+ }
+
+ String fileName = file.getOriginalFilename();
+ String dirPath = "/home/javaweb/webresource/ai/import/";
+
+ File saveFile = new File(new File(dirPath).getAbsolutePath() + File.separator + fileName);
+ if (!saveFile.exists()) {
+ if (!saveFile.getParentFile().exists()) {
+ saveFile.getParentFile().mkdirs();
+ }
+ }
+ file.transferTo(saveFile);
+
+ List<ExcelSheetPO> data = ExcelUtil.readExcel(saveFile, null, null);
+ if (CollUtil.isEmpty(data)) {
+ return new FebsResponse().fail();
+ }
+
+ List<List<Object>> dataList = data.get(0).getDataList();
+ // String[] header = {"答案ID", "题目ID", "题目", "答案", "正确答案", "解析"};
+ int questionItemIdIndex = -1;
+ int questionIdIndex = -1;
+ int titleIndex = -1;
+ int answerIndex = -1;
+ int correctAnswerIndex = -1;
+ int answerAnalysisIndex = -1;
+ for (int i = 1; i < dataList.size(); i++) {
+ List<Object> objects = dataList.get(i);
+
+ String questionItemId = "";
+ String questionId = "";
+ String title = "";
+ String answer = "";
+ Integer correctAnswer = 0;
+ String answerAnalysis = "";
+ for (int j = 0; j < objects.size(); j++) {
+ Object obj = objects.get(j);
+ if ("答案ID".equals(obj)) {
+ questionItemIdIndex = j;
+ }
+ if ("题目ID".equals(obj)) {
+ questionIdIndex = j;
+ }
+ if ("题目".equals(obj)) {
+ titleIndex = j;
+ }
+ if ("答案".equals(obj)) {
+ answerIndex = j;
+ }
+ if ("正确答案".equals(obj)) {
+ correctAnswerIndex = j;
+ }
+ if ("解析".equals(obj)) {
+ answerAnalysisIndex = j;
+ }
+ if (j == questionItemIdIndex) {
+ questionItemId = (String) objects.get(j);
+ }
+ if (j == questionIdIndex) {
+ questionId = (String) objects.get(j);
+ }
+ if (j == titleIndex) {
+ title = (String) objects.get(j);
+ }
+ if (j == answerIndex) {
+ answer = (String) objects.get(j);
+ }
+ if (j == correctAnswerIndex) {
+ correctAnswer = "是".equals((String) objects.get(j)) ? 1 : 0;
+ }
+ if (j == answerAnalysisIndex) {
+ answerAnalysis = (String) objects.get(j);
+ }
+ }
+
+ if (
+ StrUtil.isNotBlank(questionItemId)
+ && StrUtil.isNotBlank(questionId)
+ && StrUtil.isNotBlank(title)
+ && StrUtil.isNotBlank(answer)
+ ){
+ aiProductQuestionItemService.getBaseMapper().update(null,
+ Wrappers.lambdaUpdate(AiProductQuestionItem.class)
+ .set(AiProductQuestionItem::getTitle, title)
+ .set(AiProductQuestionItem::getAnswer, answer)
+ .set(AiProductQuestionItem::getCorrectAnswer, correctAnswer)
+ .set(AiProductQuestionItem::getAnswerAnalysis, answerAnalysis)
+ .set(AiProductQuestionItem::getUpdatedTime, new Date())
+ .eq(AiProductQuestionItem::getId, questionItemId)
+ .eq(AiProductQuestionItem::getProductQuestionId, questionId));
+ this.update(null,
+ Wrappers.lambdaUpdate(AiProductQuestion.class)
+ .set(AiProductQuestion::getTitle, title)
+ .eq(AiProductQuestion::getId, questionId));
+ }
+ }
+ return new FebsResponse().success();
+ } catch (IOException e) {
+ e.printStackTrace();
+ throw new RuntimeException("导入失败，请稍后重试");
+ }
+ }
+
+
/**
* 解析AI返回的包含问题和答案的JSON字符串
* @param aiResponse AI返回的原始响应字符串
diff --git a/src/main/resources/templates/febs/views/modules/ai/productQuestion/list.html b/src/main/resources/templates/febs/views/modules/ai/productQuestion/list.html
index 246eb71..1c627a3 100644
--- a/src/main/resources/templates/febs/views/modules/ai/productQuestion/list.html
+++ b/src/main/resources/templates/febs/views/modules/ai/productQuestion/list.html
@@ -55,7 +55,9 @@
<button class="layui-btn layui-btn-sm layui-btn-primary febs-button-green-plain" shiro:hasPermission="productQuestionList:aiAdd" lay-event="productQuestionAiAdd">AI新增</button>
<button class="layui-btn layui-btn-sm layui-btn-primary febs-button-green-plain" shiro:hasPermission="productQuestionList:aiAdd" lay-event="productQuestionStateOpen">启用</button>
<button class="layui-btn layui-btn-sm layui-btn-primary febs-button-green-plain" shiro:hasPermission="productQuestionList:aiAdd" lay-event="productQuestionStateClose">禁用</button>
- <button class="layui-btn layui-btn-sm layui-btn-primary febs-button-green-plain" shiro:hasPermission="productQuestionList:aiAdd" lay-event="productQuestionDelete">删除</button>
+ <button class="layui-btn layui-btn-danger layui-btn-primary febs-button-green-plain" shiro:hasPermission="productQuestionList:aiAdd" lay-event="productQuestionDelete">删除</button>
+ <button class="layui-btn layui-btn-sm layui-btn-primary febs-button-blue-plain" shiro:hasPermission="productQuestionList:aiAdd" lay-event="exportProductQuestion">导出</button>
+ <button class="layui-btn layui-btn-sm layui-btn-primary febs-button-blue-plain" shiro:hasPermission="productQuestionList:aiAdd" lay-event="importProductQuestion">导入</button>
</div>
</script>
@@ -93,11 +95,12 @@
<!-- 表格操作栏 end -->
<script data-th-inline="none" type="text/javascript">
// 引入组件并初始化
- layui.use([ 'jquery', 'form', 'table', 'febs', 'xmSelect'], function () {
+ layui.use([ 'jquery', 'form', 'table', 'febs', 'xmSelect', 'upload'], function () {
var $ = layui.jquery,
febs = layui.febs,
form = layui.form,
table = layui.table,
+ upload = layui.upload,
$view = $('#febs-productQuestion'),
$query = $view.find('#query'),
$reset = $view.find('#reset'),
@@ -252,6 +255,29 @@
productQuestionDelete(ids.join(','));
});
}
+ if (layEvent == 'exportProductQuestion') {
+ var checkData = table.checkStatus('productQuestionTable').data;
+ if (checkData.length <= 0) {
+ febs.alert.warn('请选择');
+ return;
+ }
+ var ids = [];
+ layui.each(checkData, function (key, item) {
+ ids.push(item.id)
+ });
+ window.location.href = ctx + "admin/productQuestion/exportProductQuestion?ids="+ids;
+ }
+ });
+
+ upload.render({
+ elem: '#importProductQuestion'
+ ,url: 'admin/productQuestion/importProductQuestion' //此处配置你自己的上传接口即可
+ ,accept: 'file' //普通文件
+ ,done: function(res){
+ console.log("123");
+ febs.alert.success('操作成功');
+ $query.click();
+ }
});
--
Gitblit v1.9.1