diff --git a/electrmangnetic/src/main/java/com/electromagnetic/industry/software/manage/pojo/other/FileInfoVO.java b/electrmangnetic/src/main/java/com/electromagnetic/industry/software/manage/pojo/other/FileInfoVO.java index 6d20d0e..cf6d6c9 100644 --- a/electrmangnetic/src/main/java/com/electromagnetic/industry/software/manage/pojo/other/FileInfoVO.java +++ b/electrmangnetic/src/main/java/com/electromagnetic/industry/software/manage/pojo/other/FileInfoVO.java @@ -49,7 +49,7 @@ public class FileInfoVO { private List labels; - private Integer prjDir; + private String dataOwn; // 是否是收藏数据,1是,0否 private Integer isFavorite; diff --git a/electrmangnetic/src/main/java/com/electromagnetic/industry/software/manage/service/serviceimpl/EdFileInfoServiceImpl.java b/electrmangnetic/src/main/java/com/electromagnetic/industry/software/manage/service/serviceimpl/EdFileInfoServiceImpl.java index 3c9e8db..dc6cce4 100644 --- a/electrmangnetic/src/main/java/com/electromagnetic/industry/software/manage/service/serviceimpl/EdFileInfoServiceImpl.java +++ b/electrmangnetic/src/main/java/com/electromagnetic/industry/software/manage/service/serviceimpl/EdFileInfoServiceImpl.java @@ -16,6 +16,7 @@ import cn.hutool.crypto.symmetric.AES; import cn.hutool.json.JSONUtil; import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper; import com.baomidou.mybatisplus.core.metadata.IPage; +import com.baomidou.mybatisplus.core.toolkit.StringUtils; import com.baomidou.mybatisplus.core.toolkit.Wrappers; import com.baomidou.mybatisplus.extension.plugins.pagination.Page; import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; @@ -1439,10 +1440,22 @@ public class EdFileInfoServiceImpl extends ServiceImpl findFavorite(String userId, FileInfoQueryDTO fileInfoQueryDTO) { - // 查询该用户收藏的文件id - LambdaQueryWrapper queryWrapper = new LambdaQueryWrapper<>(); - queryWrapper.select(EdFileFavorite::getFileId).eq(EdFileFavorite::getUserId, userId); - List fileIds = fileFavoriteService.listObjs(queryWrapper, Object::toString); + + List fileIds; + + // 查询收藏文件夹下的文件 + String parentId = fileInfoQueryDTO.getParentId(); + if (StringUtils.isNotBlank(parentId)) { + LambdaQueryWrapper queryWrapper = new LambdaQueryWrapper<>(); + queryWrapper.eq(EdFileInfo::getParentId, parentId) + .eq(EdFileInfo::getSaveStatus, EleDataSaveStatusEnum.SUCCESS.code) + .eq(EdFileInfo::getEffectFlag, EffectFlagEnum.EFFECT.code); + fileIds = this.baseMapper.selectList(queryWrapper).stream().map(EdFileInfo::getFileId).collect(Collectors.toList()); + } else { + LambdaQueryWrapper queryWrapper = new LambdaQueryWrapper<>(); + queryWrapper.select(EdFileFavorite::getFileId).eq(EdFileFavorite::getUserId, userId); + fileIds = fileFavoriteService.listObjs(queryWrapper, Object::toString); + } // 创建分页对象 Page page = new Page<>(fileInfoQueryDTO.getPageNum(), fileInfoQueryDTO.getPageSize()); @@ -1450,7 +1463,7 @@ public class EdFileInfoServiceImpl extends ServiceImpl { e.setFileSizeShow(EleCommonUtil.convertFileSize(e.getFileSize())); e.setIsFavorite(isFavorite(UserThreadLocal.getUserId(), e.getId()) ? 1 : 0); - e.setIsPersonal( (e.getPrjDir()== DataOwnEnum.USER_FILE.code && Objects.equals(e.getCreatedBy(), UserThreadLocal.getUserId())) ? 1 : 0); + e.setIsPersonal( (String.valueOf(DataOwnEnum.USER_FILE.code).equals(e.getDataOwn()) && Objects.equals(e.getCreatedBy(), UserThreadLocal.getUserId())) ? 1 : 0); // 设置权限 Map permissions = (e.getIsPersonal() == 1) ? permissionService.getPersonalPermission() diff --git a/electrmangnetic/src/main/java/com/electromagnetic/industry/software/manage/service/serviceimpl/UserServiceImpl.java b/electrmangnetic/src/main/java/com/electromagnetic/industry/software/manage/service/serviceimpl/UserServiceImpl.java index 236e162..b789d89 100644 --- a/electrmangnetic/src/main/java/com/electromagnetic/industry/software/manage/service/serviceimpl/UserServiceImpl.java +++ b/electrmangnetic/src/main/java/com/electromagnetic/industry/software/manage/service/serviceimpl/UserServiceImpl.java @@ -364,7 +364,6 @@ public class UserServiceImpl implements UserService { User user = userMapper.getSingleUser(userId); Assert.notNull(user, StrFormatter.format("用户不存在,ID为 {}", userId)); String decodeOldPwd = AESUtils.decrypt(oldInputPassword, UserConstants.SECRET_KEY); - Assert.isTrue(matchPassword(user, decodeOldPwd), StrFormatter.format("旧密码错误,ID为 {}", userId)); - return ElectromagneticResultUtil.success(true); + return ElectromagneticResultUtil.success(matchPassword(user, decodeOldPwd)); } } diff --git a/electrmangnetic/src/main/resources/sqlmapper/EdFileInfoMapper.xml b/electrmangnetic/src/main/resources/sqlmapper/EdFileInfoMapper.xml index 7c8ab8e..237a9e6 100644 --- a/electrmangnetic/src/main/resources/sqlmapper/EdFileInfoMapper.xml +++ b/electrmangnetic/src/main/resources/sqlmapper/EdFileInfoMapper.xml @@ -13,7 +13,7 @@ - +