From 4d40eac3afbd474a6c0ac6a13c04311124a8f931 Mon Sep 17 00:00:00 2001
From: sunxiqing <2240398334@qq.com>
Date: Sat, 16 Aug 2025 09:53:18 +0800
Subject: [PATCH] =?UTF-8?q?feat(llm):=20=E6=B7=BB=E5=8A=A0=E5=88=A0?=
 =?UTF-8?q?=E9=99=A4=E6=95=B0=E6=8D=AE=E9=9B=86=E6=97=B6=E5=88=A0=E9=99=A4?=
 =?UTF-8?q?=E9=97=AE=E9=A2=98=E5=92=8C=E7=AD=94=E6=A1=88=E7=9A=84=E5=8A=9F?=
 =?UTF-8?q?=E8=83=BD?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

- 在 DatasetAnswerMapper 和 DatasetQuestionMapper 中添加删除方法
- 在 DatasetServiceImpl 中调用这些方法以删除数据集的问题和答案
-优化数据集删除逻辑，确保相关问题和答案也被正确删除
---
 .../module/llm/dal/mysql/dataset/DatasetAnswerMapper.java     | 2 ++
 .../module/llm/dal/mysql/dataset/DatasetQuestionMapper.java   | 2 ++
 .../yudao/module/llm/service/dataset/DatasetServiceImpl.java  | 4 ++++
 .../src/main/resources/mapper/dataset/DatasetAnswerMapper.xml | 3 +++
 .../main/resources/mapper/dataset/DatasetQuestionMapper.xml   | 4 ++++
 5 files changed, 15 insertions(+)

diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/dal/mysql/dataset/DatasetAnswerMapper.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/dal/mysql/dataset/DatasetAnswerMapper.java
index 63b0e4d3f..ed48605fa 100644
--- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/dal/mysql/dataset/DatasetAnswerMapper.java
+++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/dal/mysql/dataset/DatasetAnswerMapper.java
@@ -33,4 +33,6 @@ public interface DatasetAnswerMapper extends BaseMapperX<DatasetAnswerDO> {
     void deleteTrue(@Param("datasetPostId") Long datasetPostId);
 
     List<DatasetAnswerRespVO> getAnswersToYourQuestions(@Param("collected") List<Long> collected);
+
+    void deleteTheAnswer(@Param("id") Long id);
 }
diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/dal/mysql/dataset/DatasetQuestionMapper.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/dal/mysql/dataset/DatasetQuestionMapper.java
index f47b62cf2..27a004916 100644
--- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/dal/mysql/dataset/DatasetQuestionMapper.java
+++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/dal/mysql/dataset/DatasetQuestionMapper.java
@@ -35,4 +35,6 @@ public interface DatasetQuestionMapper extends BaseMapperX<DatasetQuestionDO> {
     void deleteTrue(@Param("datasetPostId") Long datasetPostId);
 
     List<DatasetQuestionRespVO> getAListOfIssues(@Param("datasetId") Long datasetId);
+
+    void deleteTheIssue(@Param("id") Long id);
 }
diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/dataset/DatasetServiceImpl.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/dataset/DatasetServiceImpl.java
index 055a6c2f4..8bc0d4ddc 100644
--- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/dataset/DatasetServiceImpl.java
+++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/dataset/DatasetServiceImpl.java
@@ -258,6 +258,10 @@ public class DatasetServiceImpl implements DatasetService {
         validateDatasetExists(id);
         // 校验使用
         validateDatasetUse(id);
+        // 删除问题（数据量太大直接删除）
+        datasetQuestionMapper.deleteTheIssue(id);
+        // 删除答案
+        datasetAnswerMapper.deleteTheAnswer(id);
         // 删除
         datasetMapper.deleteById(id);
     }
diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/resources/mapper/dataset/DatasetAnswerMapper.xml b/yudao-module-llm/yudao-module-llm-biz/src/main/resources/mapper/dataset/DatasetAnswerMapper.xml
index 11c667c57..c0aa2cd97 100644
--- a/yudao-module-llm/yudao-module-llm-biz/src/main/resources/mapper/dataset/DatasetAnswerMapper.xml
+++ b/yudao-module-llm/yudao-module-llm-biz/src/main/resources/mapper/dataset/DatasetAnswerMapper.xml
@@ -1,6 +1,9 @@
 <?xml version="1.0" encoding="UTF-8"?>
 <!DOCTYPE mapper PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN" "http://mybatis.org/dtd/mybatis-3-mapper.dtd">
 <mapper namespace="cn.iocoder.yudao.module.llm.dal.mysql.dataset.DatasetAnswerMapper">
+    <delete id="deleteTheAnswer">
+        DELETE FROM llm_dataset_answer WHERE dataset_id = #{id}
+    </delete>
 
     <!--
         一般情况下，尽可能使用 Mapper 进行 CRUD 增删改查即可。
diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/resources/mapper/dataset/DatasetQuestionMapper.xml b/yudao-module-llm/yudao-module-llm-biz/src/main/resources/mapper/dataset/DatasetQuestionMapper.xml
index 7548f4a06..3452188e3 100644
--- a/yudao-module-llm/yudao-module-llm-biz/src/main/resources/mapper/dataset/DatasetQuestionMapper.xml
+++ b/yudao-module-llm/yudao-module-llm-biz/src/main/resources/mapper/dataset/DatasetQuestionMapper.xml
@@ -1,6 +1,10 @@
 <?xml version="1.0" encoding="UTF-8"?>
 <!DOCTYPE mapper PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN" "http://mybatis.org/dtd/mybatis-3-mapper.dtd">
 <mapper namespace="cn.iocoder.yudao.module.llm.dal.mysql.dataset.DatasetQuestionMapper">
+    <delete id="deleteTheIssue">
+        DELETE FROM llm_dataset_question
+        WHERE dataset_id = #{id}
+    </delete>
 
     <!--
         一般情况下，尽可能使用 Mapper 进行 CRUD 增删改查即可。