feat(ai): 新增产品题目导入导出功能

Administrator

yesterday cbe4b354224dfc34a852c7b7f932d33bbe85ffec

 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;

@@ -215,6 +230,7 @@
                Wrappers.lambdaUpdate(AiProductQuestion.class)
                        .set(AiProductQuestion::getState, 2)
                        .set(AiProductQuestion::getUpdatedTime, new Date())
                        .eq(AiProductQuestion::getState, 0)
                        .in(AiProductQuestion::getId, idList));
        return new FebsResponse().success().message("操作成功");
    }
@@ -400,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返回的原始响应字符串