package cc.mrbird.febs.ai.service.impl; import cc.mrbird.febs.ai.entity.*; import cc.mrbird.febs.ai.mapper.AiProductQuestionJobMapper; import cc.mrbird.febs.ai.mapper.AiProductQuestionMapper; import cc.mrbird.febs.ai.req.AiProductQuestionAiDto; import cc.mrbird.febs.ai.res.AdminQuestionInfoVo; import cc.mrbird.febs.ai.service.AiProductCategoryService; import cc.mrbird.febs.ai.service.AiProductQuestionItemService; import cc.mrbird.febs.ai.service.AiProductQuestionService; import cc.mrbird.febs.ai.service.AiService; import cc.mrbird.febs.ai.util.UUID; 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.mall.vo.AdminAgentMemberVo; 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; import cn.hutool.json.JSONArray; import cn.hutool.json.JSONObject; import cn.hutool.json.JSONUtil; import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper; import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper; import com.baomidou.mybatisplus.core.metadata.IPage; import com.baomidou.mybatisplus.core.toolkit.Wrappers; import com.baomidou.mybatisplus.extension.plugins.pagination.Page; import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; import com.fasterxml.jackson.databind.JsonNode; import com.fasterxml.jackson.databind.ObjectMapper; import com.fasterxml.jackson.databind.SerializationFeature; import lombok.RequiredArgsConstructor; 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.math.BigDecimal; import java.net.URLEncoder; import java.util.*; import java.util.stream.Collectors; /** * AI产品题目 Service实现类 * * @author yourname * @date 2025-07-29 */ @Slf4j @Service @RequiredArgsConstructor public class AiProductQuestionServiceImpl extends ServiceImpl implements AiProductQuestionService { private final AiProductQuestionMapper aiProductQuestionMapper; private final AiProductQuestionJobMapper aiProductQuestionJobMapper; private final AiProductQuestionItemService aiProductQuestionItemService; private final AiProductCategoryService aiProductCategoryService; private final AiService aiService; private final AgentProducer agentProducer; @Override public AiProductQuestion getById(String id) { return aiProductQuestionMapper.selectById(id); } @Override public IPage listInPage(AiProductQuestion dto, QueryRequest request) { Page page = new Page<>(request.getPageNum(), request.getPageSize()); LambdaQueryWrapper query = Wrappers.lambdaQuery(AiProductQuestion.class); if (StrUtil.isNotEmpty(dto.getCompanyId())){ query.eq(AiProductQuestion::getCompanyId, dto.getCompanyId()); } if (StrUtil.isNotEmpty(dto.getProductCategoryId())){ query.eq(AiProductQuestion::getProductCategoryId, dto.getProductCategoryId()); } query.ne(AiProductQuestion::getState, 2); query.orderByDesc(AiProductQuestion::getCreatedTime); Page pages = aiProductQuestionMapper.selectPage(page, query); List records = pages.getRecords(); if (CollUtil.isNotEmpty( records)){ //stream流操作records，获取全部的productCategoryId的set集合 Set productCategoryIds = records.stream().map(AiProductQuestion::getProductCategoryId).collect(Collectors.toSet()); if(CollUtil.isNotEmpty( productCategoryIds)){ Map map = aiProductCategoryService.selectMapByIds(productCategoryIds); for (AiProductQuestion record : records){ AiProductCategory orDefault = map.getOrDefault(record.getProductCategoryId(), null); if(ObjectUtil.isNotNull(orDefault)){ record.setProductCategoryName(orDefault.getName()); } } } } return pages; } @Override public IPage listJobInPage(AiProductQuestionJob dto, QueryRequest request) { Page page = new Page<>(request.getPageNum(), request.getPageSize()); LambdaQueryWrapper query = Wrappers.lambdaQuery(AiProductQuestionJob.class); if (StrUtil.isNotEmpty(dto.getCompanyId())){ query.eq(AiProductQuestionJob::getCompanyId, dto.getCompanyId()); } if (StrUtil.isNotEmpty(dto.getProductCategoryId())){ query.eq(AiProductQuestionJob::getProductCategoryId, dto.getProductCategoryId()); } query.orderByDesc(AiProductQuestionJob::getCreatedTime); Page pages = aiProductQuestionJobMapper.selectPage(page, query); return pages; } @Override public FebsResponse changeState(String id, Integer state) { AiProductQuestion entity = this.getById(id); if(ObjectUtil.isNotEmpty(entity)){ this.update(null, Wrappers.lambdaUpdate(AiProductQuestion.class) .set(AiProductQuestion::getState, state) .set(AiProductQuestion::getUpdatedTime, new Date()) .eq(AiProductQuestion::getId, id)); } return new FebsResponse().success().message("操作成功"); } @Override public FebsResponse add(AiProductQuestion dto) { List aiProductQuestionItems = dto.getAiProductQuestionItems(); if(CollUtil.isEmpty(aiProductQuestionItems)){ return new FebsResponse().fail().message("请添加答案选项"); } //stream流操作aiProductQuestionItems，返回aiProductQuestionItems是否存在中correctAnswer为1的元素 boolean hasCorrectAnswer = aiProductQuestionItems.stream().anyMatch(item -> item != null && item.getCorrectAnswer() != null && item.getCorrectAnswer() == 1); if (!hasCorrectAnswer) { return new FebsResponse().fail().message("请添加一项为正确的答案"); } AiProductQuestion entity = new AiProductQuestion(); entity.setId(UUID.getSimpleUUIDString()); entity.setCompanyId(dto.getCompanyId()); entity.setProductCategoryId(dto.getProductCategoryId()); entity.setTitle(dto.getTitle()); entity.setDifficulty(dto.getDifficulty()); entity.setCreatedTime(new Date()); this.save(entity); for (AiProductQuestionItem item : dto.getAiProductQuestionItems()){ item.setId(UUID.getSimpleUUIDString()); item.setProductQuestionId(entity.getId()); item.setCompanyId(dto.getCompanyId()); item.setTitle(entity.getTitle()); item.setCreatedTime(new Date()); aiProductQuestionItemService.getBaseMapper().insert(item); } return new FebsResponse().success().message("操作成功"); } @Override public FebsResponse update(AiProductQuestion dto) { String id = dto.getId(); AiProductQuestion entity = this.getById(id); List aiProductQuestionItems = dto.getAiProductQuestionItems(); if(CollUtil.isEmpty(aiProductQuestionItems)){ return new FebsResponse().fail().message("请添加答案选项"); } //stream流操作aiProductQuestionItems，返回aiProductQuestionItems是否存在中correctAnswer为1的元素 boolean hasCorrectAnswer = aiProductQuestionItems.stream().anyMatch(item -> item != null && item.getCorrectAnswer() != null && item.getCorrectAnswer() == 1); if (!hasCorrectAnswer) { return new FebsResponse().fail().message("请添加一项为正确的答案"); } if (ObjectUtil.isNotNull( entity)){ this.update(null, Wrappers.lambdaUpdate(AiProductQuestion.class) .set(AiProductQuestion::getProductCategoryId, dto.getProductCategoryId()) .set(AiProductQuestion::getTitle, dto.getTitle()) .set(AiProductQuestion::getDifficulty, dto.getDifficulty()) .set(AiProductQuestion::getUpdatedTime, new Date()) .eq(AiProductQuestion::getId, id) ); aiProductQuestionItemService.deleteByQuestionId(id); for (AiProductQuestionItem item : dto.getAiProductQuestionItems()){ item.setId(UUID.getSimpleUUIDString()); item.setProductQuestionId(entity.getId()); item.setTitle(entity.getTitle()); item.setCreatedTime(new Date()); aiProductQuestionItemService.getBaseMapper().insert(item); } } return new FebsResponse().success().message("操作成功"); } @Override public FebsResponse stateUpdate(String ids, Integer type) { if (StrUtil.isEmpty(ids)) { throw new FebsException("参数错误"); } List idList = StrUtil.split(ids, ','); this.update(null, Wrappers.lambdaUpdate(AiProductQuestion.class) .set(AiProductQuestion::getState, type) .set(AiProductQuestion::getUpdatedTime, new Date()) .in(AiProductQuestion::getId, idList)); return new FebsResponse().success().message("操作成功"); } @Override public FebsResponse jobStateUpdate(String ids) { if (StrUtil.isEmpty(ids)) { throw new FebsException("参数错误"); } List idList = StrUtil.split(ids, ','); aiProductQuestionJobMapper.update(null, Wrappers.lambdaUpdate(AiProductQuestionJob.class) .set(AiProductQuestionJob::getState, 2) .set(AiProductQuestionJob::getUpdatedTime, new Date()) .in(AiProductQuestionJob::getId, idList)); return new FebsResponse().success().message("操作成功"); } @Override public FebsResponse delete(String id) { AiProductQuestion entity = this.getById(id); if(ObjectUtil.isNotNull( entity)){ this.update(null, Wrappers.lambdaUpdate(AiProductQuestion.class) .set(AiProductQuestion::getState, 2) .set(AiProductQuestion::getUpdatedTime, new Date()) .eq(AiProductQuestion::getId, id)); } return new FebsResponse().success().message("操作成功"); } @Override public FebsResponse productQuestionDelete(String ids) { if (StrUtil.isEmpty(ids)) { throw new FebsException("参数错误"); } List idList = StrUtil.split(ids, ','); this.update(null, 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("操作成功"); } @Override public List questionTree(String companyId) { LambdaQueryWrapper query = Wrappers.lambdaQuery(AiProductQuestion.class); if (StrUtil.isNotEmpty(companyId)){ query.eq(AiProductQuestion::getCompanyId, companyId); } return aiProductQuestionMapper.selectList(query); } @Override public List productQuestionTree(LambdaQueryWrapper aiProductQuestionLambdaQueryWrapper) { return aiProductQuestionMapper.selectList(aiProductQuestionLambdaQueryWrapper); } @Override public FebsResponse aiAdd(AiProductQuestionAiDto dto) { String jsonFormat = "{\"question_list\":[{\"title\": \"消费者对透明质酸的主要担忧是什么？\",\"answer_list\":[{\"answer\": \"消费者担心透明质酸维持时间短，效果不明显。\",\"type\": 1,\"analysis\": \"\"},{\"answer\": \"消费者担心透明质酸注射后可能出现移位、凹陷、馒化、僵硬以及炎症反应。\",\"type\": 1,\"analysis\": \"\"},{\"answer\": \"消费者的主要担忧包括效果不明显、维持时间短、注射后出现移位、凹陷、馒化、僵硬以及炎症反应。他们希望既达到理想效果，又避免这些副作用。\",\"type\": 2,\"analysis\": \"标准答案涵盖了消费者对透明质酸的所有主要担忧，既包括效果问题，也涵盖副作用风险，全面反映消费者心理需求。\"}]}]} "; dto.setJsonFormat(jsonFormat); String questionAndAnswerStr = aiService.llmInvokeNonStreaming(dto); System.out.println(questionAndAnswerStr); // 解析AI返回的结果 List questionList = parseAiQuestionResponse(questionAndAnswerStr); String productCategoryId = dto.getProductCategoryId(); Integer difficulty = dto.getDifficulty(); Date createdTime = new Date(); // 处理解析后的问题列表 for (JSONObject questionObj : questionList) { String title = questionObj.getStr("title"); AiProductQuestion aiProductQuestion = new AiProductQuestion(); aiProductQuestion.setId(UUID.getSimpleUUIDString()); aiProductQuestion.setCompanyId(dto.getCompanyId()); aiProductQuestion.setProductCategoryId(productCategoryId); aiProductQuestion.setTitle(title); aiProductQuestion.setDifficulty(difficulty); aiProductQuestion.setCreatedTime(createdTime); this.save(aiProductQuestion); JSONArray answerList = questionObj.getJSONArray("answer_list"); System.out.println("问题: " + title); for (int i = 0; i < answerList.size(); i++) { JSONObject answer = answerList.getJSONObject(i); System.out.println("答案" + (i+1) + ": " + answer.getStr("answer")); System.out.println("类型: " + answer.getStr("type")); System.out.println("分析: " + answer.getStr("analysis")); AiProductQuestionItem aiProductQuestionItem = new AiProductQuestionItem(); aiProductQuestionItem.setId(UUID.getSimpleUUIDString()); aiProductQuestionItem.setCompanyId(dto.getCompanyId()); aiProductQuestionItem.setProductQuestionId(aiProductQuestion.getId()); aiProductQuestionItem.setTitle(aiProductQuestion.getTitle()); aiProductQuestionItem.setAnswer(answer.getStr("answer")); aiProductQuestionItem.setCorrectAnswer(answer.getInt("type") == 2 ? 1 : 0); aiProductQuestionItem.setAnswerAnalysis(answer.getStr("analysis")); aiProductQuestionItem.setCreatedTime(createdTime); aiProductQuestionItemService.getBaseMapper().insert(aiProductQuestionItem); } } return new FebsResponse().success().message("操作成功"); } @Override public FebsResponse aiAddV2(AiProductQuestionAiDto dto) { String companyId = dto.getCompanyId(); String productCategoryId = dto.getProductCategoryId(); if (StrUtil.isEmpty(productCategoryId)){ return new FebsResponse().fail().message("请选择产品分类"); } Integer questionCnt = dto.getQuestionCnt(); String query = dto.getQuery(); Integer difficulty = dto.getDifficulty(); String promptAiSystem = dto.getPromptAiSystem(); Date createdTime = new Date(); AiProductQuestionJob aiProductQuestionJob = new AiProductQuestionJob(); aiProductQuestionJob.setId(UUID.getSimpleUUIDString()); aiProductQuestionJob.setCompanyId(companyId); aiProductQuestionJob.setProductCategoryId(productCategoryId); aiProductQuestionJob.setTitle(query); aiProductQuestionJob.setPromptAiSystem(promptAiSystem); aiProductQuestionJob.setQuestionCnt(questionCnt); aiProductQuestionJob.setDifficulty(difficulty); aiProductQuestionJob.setCreatedTime(createdTime); aiProductQuestionJobMapper.insert(aiProductQuestionJob); agentProducer.sendAddQuestionJob(aiProductQuestionJob.getId()); return new FebsResponse().success().message("操作成功"); } @Override public void getAddQuestion(String id) { try { AiProductQuestionJob aiProductQuestionJob = aiProductQuestionJobMapper.selectById(id); if (ObjectUtil.isEmpty(aiProductQuestionJob)){ return; } Integer state = aiProductQuestionJob.getState(); if (2 == state){ return; } Integer questionCnt = aiProductQuestionJob.getQuestionCnt(); Integer questionDoneCnt = aiProductQuestionJob.getQuestionDoneCnt(); if (questionCnt <= questionDoneCnt){ aiProductQuestionJobMapper.update(null, Wrappers.lambdaUpdate(AiProductQuestionJob.class) .set(AiProductQuestionJob::getState, 2) .set(AiProductQuestionJob::getUpdatedTime, new Date()) .eq(AiProductQuestionJob::getId, id)); return; } // int i = questionCnt - questionDoneCnt; // if (i >= 5){ // i = 5; // } int i = questionCnt - questionDoneCnt; AiProductQuestionAiDto dto = new AiProductQuestionAiDto(); String jsonFormat = "{\"question_list\":[{\"title\": \"消费者对透明质酸的主要担忧是什么？\",\"answer_list\":[{\"answer\": \"消费者担心透明质酸维持时间短，效果不明显。\",\"type\": 1,\"analysis\": \"\"},{\"answer\": \"消费者担心透明质酸注射后可能出现移位、凹陷、馒化、僵硬以及炎症反应。\",\"type\": 1,\"analysis\": \"\"},{\"answer\": \"消费者的主要担忧包括效果不明显、维持时间短、注射后出现移位、凹陷、馒化、僵硬以及炎症反应。他们希望既达到理想效果，又避免这些副作用。\",\"type\": 2,\"analysis\": \"标准答案涵盖了消费者对透明质酸的所有主要担忧，既包括效果问题，也涵盖副作用风险，全面反映消费者心理需求。\"}]}]} "; dto.setJsonFormat(jsonFormat); dto.setCompanyId(aiProductQuestionJob.getCompanyId()); dto.setProductCategoryId(aiProductQuestionJob.getProductCategoryId()); dto.setQuery(aiProductQuestionJob.getTitle()); dto.setPromptAiSystem(aiProductQuestionJob.getPromptAiSystem()); dto.setQuestionCnt(i); dto.setDifficulty(aiProductQuestionJob.getDifficulty()); String questionAndAnswerStr = aiService.llmInvokeNonStreaming(dto); log.info("问题答案: " + questionAndAnswerStr); // 解析AI返回的结果 List questionList = parseAiQuestionResponse(questionAndAnswerStr); if (CollUtil.isNotEmpty(questionList)){ String productCategoryId = dto.getProductCategoryId(); Integer difficulty = dto.getDifficulty(); Date createdTime = new Date(); // 处理解析后的问题列表 for (JSONObject questionObj : questionList) { String title = questionObj.getStr("title"); AiProductQuestion aiProductQuestion = new AiProductQuestion(); aiProductQuestion.setId(UUID.getSimpleUUIDString()); aiProductQuestion.setCompanyId(dto.getCompanyId()); aiProductQuestion.setProductCategoryId(productCategoryId); aiProductQuestion.setTitle(title); aiProductQuestion.setDifficulty(difficulty); aiProductQuestion.setJobId(id); aiProductQuestion.setCreatedTime(createdTime); this.save(aiProductQuestion); JSONArray answerList = questionObj.getJSONArray("answer_list"); for (int j = 0; j < answerList.size(); j++) { JSONObject answer = answerList.getJSONObject(j); System.out.println("答案" + (j+1) + ": " + answer.getStr("answer")); System.out.println("类型: " + answer.getStr("type")); System.out.println("分析: " + answer.getStr("analysis")); AiProductQuestionItem aiProductQuestionItem = new AiProductQuestionItem(); aiProductQuestionItem.setId(UUID.getSimpleUUIDString()); aiProductQuestionItem.setCompanyId(dto.getCompanyId()); aiProductQuestionItem.setProductQuestionId(aiProductQuestion.getId()); aiProductQuestionItem.setTitle(aiProductQuestion.getTitle()); aiProductQuestionItem.setAnswer(answer.getStr("answer")); aiProductQuestionItem.setCorrectAnswer(answer.getInt("type") == 2 ? 1 : 0); aiProductQuestionItem.setAnswerAnalysis(answer.getStr("analysis")); aiProductQuestionItem.setJobId( id); aiProductQuestionItem.setCreatedTime(createdTime); aiProductQuestionItemService.getBaseMapper().insert(aiProductQuestionItem); } } aiProductQuestionJobMapper.update(null, Wrappers.lambdaUpdate(AiProductQuestionJob.class) .set(AiProductQuestionJob::getQuestionDoneCnt, questionDoneCnt + questionList.size()) .set(AiProductQuestionJob::getUpdatedTime, new Date()) .eq(AiProductQuestionJob::getId, id)); } AiProductQuestionJob aiProductQuestionJobDone = aiProductQuestionJobMapper.selectById(id); if (2 != aiProductQuestionJobDone.getState()){ agentProducer.sendAddQuestionJob(id); } } catch (Exception e) { agentProducer.sendAddQuestionJob(id); log.error("知识库异常", e); } } @Override public void exportProductQuestion(AiProductQuestion dto, HttpServletResponse response) { try { List 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 ids = StrUtil.splitTrim(questionIds, ","); List aiProductQuestionItems = aiProductQuestionItemService.getListByQuery( Wrappers.lambdaQuery(AiProductQuestionItem.class) .in(AiProductQuestionItem::getProductQuestionId, ids) .orderByAsc(AiProductQuestionItem::getProductQuestionId) ); List> list = new ArrayList<>(); if (CollUtil.isNotEmpty(aiProductQuestionItems)) { for (AiProductQuestionItem item : aiProductQuestionItems) { List 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 data = ExcelUtil.readExcel(saveFile, null, null); if (CollUtil.isEmpty(data)) { return new FebsResponse().fail(); } List> 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 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返回的原始响应字符串 * @return 解析后的问题列表 */ public static List parseAiQuestionResponse(String aiResponse) { try { // 解析外层JSON JSONObject outerJson = JSONUtil.parseObj(aiResponse); // 提取output字段 String output = outerJson.getStr("output"); log.info("原始输出内容: {}", output); // 去除和标签，并清理多余字符 String jsonContent = output.replace("", "") .replace("", "") .replace("\\n", "") .replace("\\\"", "\"") .trim(); log.info("清理后JSON内容: {}", jsonContent); // 解析内部JSON JSONObject innerJson = JSONUtil.parseObj(jsonContent); // 提取question_list JSONArray questionList = innerJson.getJSONArray("question_list"); // 转换为List返回 return questionList.toList(JSONObject.class); } catch (Exception e) { log.error("解析AI问题响应失败: ", e); throw new RuntimeException("解析AI响应失败", e); } } public static void main(String[] args) { String s = "{\"output\":\"{\\\"question_list\\\":[{\\\"title\\\":\\\"根据知识库内容，调整T区的三大作用分别是什么？请逐条准确列举。\\\",\\\"answer_list\\\":[{\\\"answer\\\":\\\"1.提升折叠度；2.调整面部比例；3.提升骨性支撑力\\\",\\\"type\\\":2,\\\"analysis\\\":\\\"原文明确列出：'1.提升折叠度(增加正面、侧面转折落差，减少扁平感)；2.调整面部比例(通过T区的纵向与横线的调整，优化三庭五眼、四高三底)；3.提升骨性支撑力(为软组织提供坚实的\\\\\\\"地基\\\\\\\"抗衰并提升高级的骨感)'\\\"},{\\\"answer\\\":\\\"1.提升折叠度；2.调整三庭五眼；3.增强软组织弹性\\\",\\\"type\\\":1,\\\"analysis\\\":\\\"\\\"},{\\\"answer\\\":\\\"1.提升立体感；2.优化面部比例；3.强化韧带支撑\\\",\\\"type\\\":1,\\\"analysis\\\":\\\"\\\"}]}]}\"}"; List questionList = parseAiQuestionResponse(s); System.out.println(questionList); } @Override public FebsResponse updateLabel(String ids, String label) { if (StrUtil.isEmpty(ids)) { throw new FebsException("参数错误"); } List idList = StrUtil.split(ids, ','); this.update(null, Wrappers.lambdaUpdate(AiProductQuestion.class) .set(AiProductQuestion::getLabel, label) .set(AiProductQuestion::getUpdatedTime, new Date()) .in(AiProductQuestion::getId, idList)); return new FebsResponse().success().message("操作成功"); } @Override public void exportNewProductQuestion(AiProductQuestion dto, HttpServletResponse response) { try { List res = new ArrayList<>(); ExcelSheetPO orderSheet = new ExcelSheetPO(); String title = "单选题目列表"; orderSheet.setSheetName(title); // 确保不设置标题行 orderSheet.setTitle(title); // 明确设置为null String[] header = {"题目", "答案", "正确答案", "解析"}; orderSheet.setHeaders(header); // 获取示例数据 AiProductQuestion aiProductQuestion = aiProductQuestionMapper.selectOne( Wrappers.lambdaQuery(AiProductQuestion.class) .orderByAsc(AiProductQuestion::getCreatedTime) .last("limit 1") ); List aiProductQuestionItems = new ArrayList<>(); if (aiProductQuestion != null) { aiProductQuestionItems = aiProductQuestionItemService.getListByQuery( Wrappers.lambdaQuery(AiProductQuestionItem.class) .eq(AiProductQuestionItem::getProductQuestionId, aiProductQuestion.getId()) .orderByAsc(AiProductQuestionItem::getProductQuestionId) ); } List> list = new ArrayList<>(); if (CollUtil.isNotEmpty(aiProductQuestionItems)) { String label = "(示例, 新增请删除整行内容)"; for (AiProductQuestionItem item : aiProductQuestionItems) { List temp = new ArrayList<>(); temp.add(label + item.getTitle()); temp.add(item.getAnswer()); temp.add(item.getCorrectAnswer() == 1 ? "是" : "否"); temp.add(item.getAnswerAnalysis()); list.add(temp); } } else { // 如果没有数据，添加一个示例行 List exampleRow = new ArrayList<>(); exampleRow.add("(示例, 新增请删除整行内容)我今年32岁, 最近感觉脸部开始有松弛现象, 尤其是下颌线不清晰了。朋友推荐我做MFU™pro美拉, 说是'无创提拉', 这个项目真的有效吗? 适合我这个年龄段吗?"); exampleRow.add("这个项目否"); exampleRow.add("是"); exampleRow.add("根据文档内容, MFU™pro美拉推荐年龄为25+, 适用于轻中度松弛人群, 具有即刻紧致与长期提拉效果, 适合早期抗衰需求"); list.add(exampleRow); } 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 public FebsResponse importNewProductQuestion(MultipartFile file, String categoryId, String companyId, Integer difficulty) { try { if (file.isEmpty()) { return new FebsResponse().fail().message("文件不能为空"); } 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 data = ExcelUtil.readExcel(saveFile, null, null); if (CollUtil.isEmpty(data)) { return new FebsResponse().fail().message("Excel文件内容为空"); } List> dataList = data.get(0).getDataList(); if (CollUtil.isEmpty(dataList) || dataList.size() < 3) { return new FebsResponse().fail().message("Excel数据行数不足"); } // 第一行是文件名称，第二行是标题栏，从第三行开始才是数据 // 从第二行获取标题栏，确定列索引 List headerRow = dataList.get(1); // 第二行是标题栏 int titleIndex = -1; int answerIndex = -1; int correctAnswerIndex = -1; int answerAnalysisIndex = -1; // 解析标题栏，确定各字段的列索引 for (int j = 0; j < headerRow.size(); j++) { Object headerCell = headerRow.get(j); if (headerCell != null) { String headerText = headerCell.toString().trim(); switch (headerText) { case "题目": titleIndex = j; break; case "答案": answerIndex = j; break; case "正确答案": correctAnswerIndex = j; break; case "解析": answerAnalysisIndex = j; break; } } } // 验证必要的列是否存在 if (titleIndex == -1 || answerIndex == -1) { return new FebsResponse().fail().message("Excel缺少必要的列（题目、答案）"); } // 用于存储题目和选项 Map questionMap = new HashMap<>(); Map> itemsMap = new HashMap<>(); // 从第三行开始处理数据（索引2） for (int i = 2; i < dataList.size(); i++) { List rowData = dataList.get(i); // 跳过空行 if (CollUtil.isEmpty(rowData)) { continue; } String title = ""; String answer = ""; Integer correctAnswer = 0; String answerAnalysis = ""; // 提取各字段的值 if (titleIndex < rowData.size() && rowData.get(titleIndex) != null) { title = rowData.get(titleIndex).toString().trim(); } if (answerIndex < rowData.size() && rowData.get(answerIndex) != null) { answer = rowData.get(answerIndex).toString().trim(); } if (correctAnswerIndex != -1 && correctAnswerIndex < rowData.size() && rowData.get(correctAnswerIndex) != null) { String correctAnswerStr = rowData.get(correctAnswerIndex).toString().trim(); // 支持多种表示正确的方式 correctAnswer = ("是".equals(correctAnswerStr) || "1".equals(correctAnswerStr) || "正确".equals(correctAnswerStr) || "对".equals(correctAnswerStr)) ? 1 : 0; } if (answerAnalysisIndex != -1 && answerAnalysisIndex < rowData.size() && rowData.get(answerAnalysisIndex) != null) { answerAnalysis = rowData.get(answerAnalysisIndex).toString().trim(); } // 如果题目为空，跳过该行 if (StrUtil.isBlank(title)) { continue; } // 使用题目内容作为key，确保同一个题目只创建一次 String questionKey = title; // 如果这个题目还没有创建过，创建题目 if (!questionMap.containsKey(questionKey)) { AiProductQuestion question = new AiProductQuestion(); question.setId(UUID.getSimpleUUIDString()); question.setCompanyId(companyId); question.setProductCategoryId(categoryId); question.setTitle(title); question.setDifficulty(difficulty); // 默认中等难度 question.setState(0); // 默认启用 question.setCreatedTime(new Date()); questionMap.put(questionKey, question); itemsMap.put(questionKey, new ArrayList<>()); } // 添加选项（只要答案不为空） if (StrUtil.isNotBlank(answer)) { AiProductQuestionItem item = new AiProductQuestionItem(); item.setId(UUID.getSimpleUUIDString()); item.setProductQuestionId(questionMap.get(questionKey).getId()); item.setCompanyId(companyId); item.setTitle(title); // 选项内容 item.setAnswer(answer); // 选项内容 item.setCorrectAnswer(correctAnswer); item.setAnswerAnalysis(answerAnalysis); item.setCreatedTime(new Date()); itemsMap.get(questionKey).add(item); } } // 保存所有题目和选项 int successCount = 0; for (Map.Entry entry : questionMap.entrySet()) { String questionKey = entry.getKey(); AiProductQuestion question = entry.getValue(); List items = itemsMap.get(questionKey); if (CollUtil.isNotEmpty(items)) { saveQuestionWithItems(question, items); successCount++; } } return new FebsResponse().success().message(String.format("成功导入%d个题目，共%d个选项", successCount, itemsMap.values().stream().mapToInt(List::size).sum())); } catch (IOException e) { log.error("导入Excel文件失败", e); throw new RuntimeException("导入失败，请稍后重试"); } } @Override public IPage jobChild(QueryRequest request, AiProductQuestionJob dto) { Page page = new Page<>(request.getPageNum(), request.getPageSize()); return aiProductQuestionMapper.selectJobChildPage(page, dto); } private void saveQuestionWithItems(AiProductQuestion question, List items) { // 保存题目（新增一次） this.save(question); // 保存选项（有几条数据就增加几条） for (AiProductQuestionItem item : items) { aiProductQuestionItemService.save(item); } } }