Fix query scene bug

Signed-off-by: fanyinbo <[email protected]>
xtreme1-io · Nov 28, 2023 · ddbf8bc · ddbf8bc
1 parent f8088f3
commit ddbf8bc
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/backend/src/main/java/ai/basic/x1/usecase/DataInfoUseCase.java b/backend/src/main/java/ai/basic/x1/usecase/DataInfoUseCase.java
@@ -329,7 +329,7 @@ public Wrapper<DataInfo> commonDataQueryWrapper(DataInfoQueryBO queryBO) {
 
     private void setSceneFirstData(Long datasetId, List<DataInfoBO> dataInfoBOList) {
         List<Long> dataIds = null;
-        var sceneIds = dataInfoBOList.stream().filter(dataInfoBO -> SCENE.equals(dataInfoBO.getType())).map(DataInfoBO::getId).collect(Collectors.toList());
+        var sceneIds = dataInfoBOList.stream().filter(dataInfoBO -> ItemTypeEnum.SCENE.equals(dataInfoBO.getType())).map(DataInfoBO::getId).collect(Collectors.toList());
         if (CollectionUtil.isNotEmpty(sceneIds)) {
             dataIds = dataInfoDAO.getBaseMapper().selectFirstDataIdBySceneIds(datasetId, sceneIds);
         }
@@ -338,7 +338,7 @@ private void setSceneFirstData(Long datasetId, List<DataInfoBO> dataInfoBOList)
             var sceneDataInfoMap = dataInfoList.stream()
                     .collect(Collectors.toMap(DataInfo::getParentId, dataInfo -> dataInfo, (k1, k2) -> k1));
             dataInfoBOList.forEach(dataInfoBO -> {
-                if (SCENE.equals(dataInfoBO.getType())) {
+                if (ItemTypeEnum.SCENE.equals(dataInfoBO.getType())) {
                     var dataInfo = sceneDataInfoMap.get(dataInfoBO.getId());
                     if (ObjectUtil.isNotNull(dataInfo)) {
                         dataInfoBO.setContent(DefaultConverter.convert(dataInfo.getContent(), DataInfoBO.FileNodeBO.class));