weiweiw
/
Bonus-Cloud-AI-V2
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

提交代码

This commit is contained in:
jiang 2024-12-02 13:48:56 +08:00
parent f2e354b221
commit 84a2063a58
2 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
bonus-modules/bonus-ai/src/main/java/com/bonus/ai/controller/DataSetLabelsController.java
View File

@ -102,6 +102,6 @@ public class DataSetLabelsController extends BaseController
@PostMapping("/getLabelsTree/{labelId}") @PostMapping("/getLabelsTree/{labelId}")
public AjaxResult getLabelsTree(@PathVariable Long labelId){ public AjaxResult getLabelsTree(@PathVariable Long labelId){
return dataSetLabelsService.getLabelsTree(labelId); return dataSetLabelsService.getLabelsTree(labelId);
} }
} }

3
bonus-modules/bonus-ai/src/main/java/com/bonus/ai/service/Impl/DataSetLabelsServiceImpl.java
View File

@ -130,6 +130,9 @@ public class DataSetLabelsServiceImpl implements DataSetLabelsService
public AjaxResult getLabelsTree(Long labelId) { public AjaxResult getLabelsTree(Long labelId) {
DataSetLabels entity = new DataSetLabels(); DataSetLabels entity = new DataSetLabels();
List<DataSetLabels> dataSetLabels = dataSetLabelsMapper.selectDataSetLabelsList(entity); List<DataSetLabels> dataSetLabels = dataSetLabelsMapper.selectDataSetLabelsList(entity);
if (labelId==0){
return AjaxResult.success(dataSetLabels);
}
Set<Long> labelIds = getFileWithChildren(labelId); Set<Long> labelIds = getFileWithChildren(labelId);
for (Long id : labelIds){ for (Long id : labelIds){
dataSetLabels.removeIf(item -> item.getLabelId().equals(id)); dataSetLabels.removeIf(item -> item.getLabelId().equals(id));