feat(llm): 添加删除数据集时删除问题和答案的功能

- 在 DatasetAnswerMapper 和 DatasetQuestionMapper 中添加删除方法
- 在 DatasetServiceImpl 中调用这些方法以删除数据集的问题和答案
-优化数据集删除逻辑,确保相关问题和答案也被正确删除
This commit is contained in:
sunxiqing 2025-08-16 09:53:18 +08:00
parent 4c0ad24006
commit 4d40eac3af
5 changed files with 15 additions and 0 deletions

View File

@ -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);
}

View File

@ -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);
}

View File

@ -258,6 +258,10 @@ public class DatasetServiceImpl implements DatasetService {
validateDatasetExists(id);
// 校验使用
validateDatasetUse(id);
// 删除问题数据量太大直接删除
datasetQuestionMapper.deleteTheIssue(id);
// 删除答案
datasetAnswerMapper.deleteTheAnswer(id);
// 删除
datasetMapper.deleteById(id);
}

View File

@ -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 增删改查即可。

View File

@ -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 增删改查即可。