From d3492864ec22a30e9fcc99b2b174cf6eb9531399 Mon Sep 17 00:00:00 2001
From: Administrator <15274802129@163.com>
Date: Mon, 02 Feb 2026 14:54:09 +0800
Subject: [PATCH] feat(rabbit): 添加AI题目生成功能
---
src/main/java/cc/mrbird/febs/ai/service/impl/AiProductQuestionServiceImpl.java | 209 ++++++++++++++++++++++++++++++++++++++++++----------
1 files changed, 168 insertions(+), 41 deletions(-)
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 24d05ec..7bd223d 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
@@ -1,6 +1,7 @@
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.service.AiProductQuestionItemService;
@@ -9,16 +10,25 @@
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.mall.entity.MallMember;
+import cc.mrbird.febs.rabbit.producter.AgentProducer;
import cn.hutool.core.collection.CollUtil;
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;
@@ -39,8 +49,10 @@
public class AiProductQuestionServiceImpl extends ServiceImpl<AiProductQuestionMapper, AiProductQuestion> implements AiProductQuestionService {
private final AiProductQuestionMapper aiProductQuestionMapper;
+ private final AiProductQuestionJobMapper aiProductQuestionJobMapper;
private final AiProductQuestionItemService aiProductQuestionItemService;
private final AiService aiService;
+ private final AgentProducer agentProducer;
@Override
public AiProductQuestion getById(String id) {
@@ -52,8 +64,30 @@
public IPage<AiProductQuestion> listInPage(AiProductQuestion dto, QueryRequest request) {
Page<AiProductQuestion> page = new Page<>(request.getPageNum(), request.getPageSize());
LambdaQueryWrapper<AiProductQuestion> 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<AiProductQuestion> pages = aiProductQuestionMapper.selectPage(page, query);
+ return pages;
+ }
+
+ @Override
+ public IPage<AiProductQuestionJob> listJobInPage(AiProductQuestionJob dto, QueryRequest request) {
+ Page<AiProductQuestionJob> page = new Page<>(request.getPageNum(), request.getPageSize());
+ LambdaQueryWrapper<AiProductQuestionJob> 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<AiProductQuestionJob> pages = aiProductQuestionJobMapper.selectPage(page, query);
return pages;
}
@@ -96,6 +130,7 @@
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);
@@ -122,7 +157,6 @@
if (ObjectUtil.isNotNull( entity)){
this.update(null,
Wrappers.lambdaUpdate(AiProductQuestion.class)
- .set(AiProductQuestion::getCompanyId, dto.getCompanyId())
.set(AiProductQuestion::getProductCategoryId, dto.getProductCategoryId())
.set(AiProductQuestion::getTitle, dto.getTitle())
.set(AiProductQuestion::getDifficulty, dto.getDifficulty())
@@ -144,6 +178,20 @@
}
@Override
+ public FebsResponse stateUpdate(String ids, Integer type) {
+ if (StrUtil.isEmpty(ids)) {
+ throw new FebsException("参数错误");
+ }
+ List<String> 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 delete(String id) {
AiProductQuestion entity = this.getById(id);
if(ObjectUtil.isNotNull( entity)){
@@ -158,9 +206,13 @@
}
@Override
- public List<AiProductQuestion> questionTree() {
+ public List<AiProductQuestion> questionTree(String companyId) {
+ LambdaQueryWrapper<AiProductQuestion> query = Wrappers.lambdaQuery(AiProductQuestion.class);
+ if (StrUtil.isNotEmpty(companyId)){
+ query.eq(AiProductQuestion::getCompanyId, companyId);
+ }
- return aiProductQuestionMapper.selectList(null);
+ return aiProductQuestionMapper.selectList(query);
}
@Override
@@ -174,6 +226,7 @@
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<JSONObject> questionList = parseAiQuestionResponse(questionAndAnswerStr);
@@ -186,6 +239,7 @@
AiProductQuestion aiProductQuestion = new AiProductQuestion();
aiProductQuestion.setId(UUID.getSimpleUUIDString());
+ aiProductQuestion.setCompanyId(dto.getCompanyId());
aiProductQuestion.setProductCategoryId(productCategoryId);
aiProductQuestion.setTitle(title);
aiProductQuestion.setDifficulty(difficulty);
@@ -201,6 +255,7 @@
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"));
@@ -211,6 +266,115 @@
}
}
return new FebsResponse().success().message("操作成功");
+ }
+
+ @Override
+ public FebsResponse aiAddV2(AiProductQuestionAiDto dto) {
+ String companyId = dto.getCompanyId();
+ String productCategoryId = dto.getProductCategoryId();
+ 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) {
+ 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){
+ aiProductQuestionJob.setState(2);
+ 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 >= 10){
+ i = 10;
+ }
+ 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);
+ // 解析AI返回的结果
+ List<JSONObject> 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.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.setCreatedTime(createdTime);
+ aiProductQuestionItemService.getBaseMapper().insert(aiProductQuestionItem);
+ }
+ }
+ aiProductQuestionJobMapper.update(null,
+ Wrappers.lambdaUpdate(AiProductQuestionJob.class)
+ .set(AiProductQuestionJob::getQuestionDoneCnt, questionDoneCnt + questionList.size())
+ .set(AiProductQuestionJob::getUpdatedTime, new Date())
+ .set(AiProductQuestionJob::getState, 1)
+ .eq(AiProductQuestionJob::getId, id));
+ }
+ AiProductQuestionJob aiProductQuestionJobDone = aiProductQuestionJobMapper.selectById(id);
+ if (2 != aiProductQuestionJobDone.getState()){
+ agentProducer.sendAddQuestionJob(id);
+ }
}
/**
@@ -230,6 +394,7 @@
String jsonContent = output.replace("<start>", "")
.replace("</start>", "")
.replace("\\n", "")
+ .replace("\\n", "")
.replace("\\\"", "\"")
.trim();
@@ -246,44 +411,6 @@
log.error("解析AI问题响应失败: ", e);
throw new RuntimeException("解析AI响应失败", e);
}
- }
-
- public static void main(String[] args) {
- String questionStr = "{\"output\":\"<start>\\n{\\n \\\"question_list\\\": [\\n {\\n \\\"title\\\": \\\"1. 如何通过菲欧曼品牌背景和价值塑造透明质酸产品的高端形象？\\\",\\n \\\"answer_list\\\": [\\n {\\n \\\"answer\\\": \\\"强调菲欧曼拥有45年历史，是法国第一家医学美容实验室，产品行销全球60多个国家，树立其国际权威形象。\\\",\\n \\\"type\\\": \\\"2\\\",\\n \\\"analysis\\\": \\\"该答案全面结合品牌历史、科研投入及市场影响力，系统塑造高端定位，符合标准答案要求。\\\"\\n },\\n {\\n \\\"answer\\\": \\\"通过讲述菲欧曼产品获得欧盟CE、中国NMPA等国际认证，强调其安全性和合规性。\\\",\\n \\\"type\\\": \\\"1\\\",\\n \\\"analysis\\\": \\\"\\\"\\n },\\n {\\n \\\"answer\\\": \\\"通过明星使用、医生推荐等口碑效应，提升菲欧曼在消费者心中的高端认知。\\\",\\n \\\"type\\\": \\\"1\\\",\\n \\\"analysis\\\": \\\"\\\"\\n }\\n ]\\n }\\n ]\\n}\\n</start>\"}";
- // 第一步：解析外层JSON
- JSONObject outerJson = JSONUtil.parseObj(questionStr);
-
- // 第二步：提取output字段
- String output = outerJson.getStr("output");
-
- // 第三步：去除<start>和</start>标签
- String jsonContent = output.replace("<start>", "").replace("</start>", "").trim();
-
- // 第四步：解析内部的JSON
- JSONObject innerJson = JSONUtil.parseObj(jsonContent);
-
- // 第五步：提取question_list
- JSONArray questionList = innerJson.getJSONArray("question_list");
-
- // 遍历问题列表
- for (int i = 0; i < questionList.size(); i++) {
- JSONObject question = questionList.getJSONObject(i);
-
- System.out.println("问题标题: " + question.getStr("title"));
-
- JSONArray answerList = question.getJSONArray("answer_list");
- System.out.println("答案列表:");
-
- // 遍历答案列表
- 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"));
- System.out.println();
- }
- }
-
}
}
--
Gitblit v1.9.1