diff --git a/bonus-admin/src/main/java/com/bonus/web/service/data/SampleService.java b/bonus-admin/src/main/java/com/bonus/web/service/data/SampleService.java
index e0c207b..b43b571 100644
--- a/bonus-admin/src/main/java/com/bonus/web/service/data/SampleService.java
+++ b/bonus-admin/src/main/java/com/bonus/web/service/data/SampleService.java
@@ -116,10 +116,14 @@ public class SampleService {
return AjaxResult.error(validResult);
}
// 1.查询样本库数据是否包含样本数据
- int result = diSampleService.hasLabelData(dto);
- if (result > 0) {
+ int result1 = diSampleService.hasLabelData(dto);
+ if (result1 > 0) {
return AjaxResult.error("样本库包含样本数据无法删除");
}
+ int result2 = diSampleService.hasDatasetData(dto);
+ if (result2 > 0) {
+ return AjaxResult.error("数据集包含样本数据无法删除");
+ }
// 2.删除样本库数据
diSampleService.operData(dto, 3);
return AjaxResult.success();
diff --git a/bonus-data/src/main/java/com/bonus/data/mapper/DISampleMapper.java b/bonus-data/src/main/java/com/bonus/data/mapper/DISampleMapper.java
index 1a78b58..73f70e8 100644
--- a/bonus-data/src/main/java/com/bonus/data/mapper/DISampleMapper.java
+++ b/bonus-data/src/main/java/com/bonus/data/mapper/DISampleMapper.java
@@ -58,6 +58,15 @@ public interface DISampleMapper {
*/
int hasLabelData(SampleLibraryDto dto);
+ /**
+ * 查询数据集数据是否包含样本数据
+ * @param dto
+ * @return int
+ * @author lhdhy
+ * @date 2025/12/22 11:19
+ */
+ int hasDatasetData(SampleLibraryDto dto);
+
/**
* 查询样本库详情
* @param dto
diff --git a/bonus-data/src/main/java/com/bonus/data/service/DISampleService.java b/bonus-data/src/main/java/com/bonus/data/service/DISampleService.java
index 7dba3e3..c32959b 100644
--- a/bonus-data/src/main/java/com/bonus/data/service/DISampleService.java
+++ b/bonus-data/src/main/java/com/bonus/data/service/DISampleService.java
@@ -56,6 +56,15 @@ public interface DISampleService {
*/
int hasLabelData(SampleLibraryDto dto);
+ /**
+ * 查询数据集数据是否包含样本数据
+ * @param dto
+ * @return int
+ * @author cwchen
+ * @date 2025/12/22 11:18
+ */
+ int hasDatasetData(SampleLibraryDto dto);
+
/**
* 查询样本库详情
* @param dto
diff --git a/bonus-data/src/main/java/com/bonus/data/service/impl/DSampleServiceImpl.java b/bonus-data/src/main/java/com/bonus/data/service/impl/DSampleServiceImpl.java
index 7865b2c..08a45b2 100644
--- a/bonus-data/src/main/java/com/bonus/data/service/impl/DSampleServiceImpl.java
+++ b/bonus-data/src/main/java/com/bonus/data/service/impl/DSampleServiceImpl.java
@@ -48,6 +48,11 @@ public class DSampleServiceImpl implements DISampleService {
return diSampleMapper.hasLabelData(dto);
}
+ @Override
+ public int hasDatasetData(SampleLibraryDto dto) {
+ return diSampleMapper.hasDatasetData(dto);
+ }
+
@Override
public SampleLibraryVo getSampleDetail(SampleLibraryDto dto) {
return diSampleMapper.getSampleDetail(dto);
diff --git a/bonus-data/src/main/resources/mapper/DILabelingTaskMapper.xml b/bonus-data/src/main/resources/mapper/DILabelingTaskMapper.xml
index a0efe89..22c27df 100644
--- a/bonus-data/src/main/resources/mapper/DILabelingTaskMapper.xml
+++ b/bonus-data/src/main/resources/mapper/DILabelingTaskMapper.xml
@@ -859,7 +859,7 @@
diff --git a/bonus-data/src/main/resources/mapper/DSampleMapper.xml b/bonus-data/src/main/resources/mapper/DSampleMapper.xml
index a26c2a5..00ac760 100644
--- a/bonus-data/src/main/resources/mapper/DSampleMapper.xml
+++ b/bonus-data/src/main/resources/mapper/DSampleMapper.xml
@@ -100,6 +100,11 @@
+
+
+