|
@@ -1,6 +1,8 @@
|
|
|
package com.usky.system.service.impl;
|
|
|
|
|
|
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
|
|
|
+import com.baomidou.mybatisplus.core.toolkit.CollectionUtils;
|
|
|
+import com.baomidou.mybatisplus.core.toolkit.StringUtils;
|
|
|
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
|
|
|
import com.usky.common.core.exception.BusinessException;
|
|
|
import com.usky.common.mybatis.core.AbstractCrudService;
|
|
@@ -43,17 +45,30 @@ public class HceCategoryServiceImpl extends AbstractCrudService<HceCategoryMappe
|
|
|
long userId = SecurityUtils.getUserId();
|
|
|
SysUser sysUser = userService.selectUserByIdOne(userId);
|
|
|
List<HceCategoryListVO> list = new ArrayList<>();
|
|
|
- if(("00").equals(sysUser.getUserType())){
|
|
|
- if (1L == userId)
|
|
|
- {
|
|
|
- list = baseMapper.selectMenuTreeAll(hceCategoryVO.getCategoryName(),hceCategoryVO.getStartTime(),hceCategoryVO.getEndTime());
|
|
|
+ List<Integer> categoryIdList = new ArrayList<>();
|
|
|
+ if(StringUtils.isNotBlank(hceCategoryVO.getCategoryName())){
|
|
|
+ LambdaQueryWrapper<HceCategory> queryWrapper = Wrappers.lambdaQuery();
|
|
|
+ queryWrapper.like(HceCategory::getCategoryName, hceCategoryVO.getCategoryName());
|
|
|
+ List<HceCategory> list1 = this.list(queryWrapper);
|
|
|
+ if(list1.size() > 0){
|
|
|
+ for (int i = 0; i < list1.size(); i++) {
|
|
|
+ categoryIdList.add(list1.get(i).getId());
|
|
|
+ }
|
|
|
}
|
|
|
- else
|
|
|
- {
|
|
|
- list = baseMapper.selectMenuTreeByUserId(hceCategoryVO.getCategoryName(),hceCategoryVO.getStartTime(),hceCategoryVO.getEndTime(),userId);
|
|
|
+ }
|
|
|
+ if(CollectionUtils.isNotEmpty(categoryIdList)){
|
|
|
+ if(("00").equals(sysUser.getUserType())){
|
|
|
+ if (1L == userId)
|
|
|
+ {
|
|
|
+ list = baseMapper.selectMenuTreeAll(categoryIdList,hceCategoryVO.getStartTime(),hceCategoryVO.getEndTime());
|
|
|
+ }
|
|
|
+ else
|
|
|
+ {
|
|
|
+ list = baseMapper.selectMenuTreeByUserId(categoryIdList,hceCategoryVO.getStartTime(),hceCategoryVO.getEndTime(),userId);
|
|
|
+ }
|
|
|
+ }else{
|
|
|
+ list = baseMapper.selectMenuTreeByUserIdOne(categoryIdList,hceCategoryVO.getStartTime(),hceCategoryVO.getEndTime(),SecurityUtils.getTenantId());
|
|
|
}
|
|
|
- }else{
|
|
|
- list = baseMapper.selectMenuTreeByUserIdOne(hceCategoryVO.getCategoryName(),hceCategoryVO.getStartTime(),hceCategoryVO.getEndTime(),SecurityUtils.getTenantId());
|
|
|
}
|
|
|
|
|
|
List<HceCategoryListVO> finalList = list;
|
|
@@ -68,7 +83,21 @@ public class HceCategoryServiceImpl extends AbstractCrudService<HceCategoryMappe
|
|
|
|
|
|
@Override
|
|
|
public List<HceCategoryListVO> pageWhite(HceCategoryVO hceCategoryVO){
|
|
|
- List<HceCategoryListVO> list = baseMapper.selectMenuTreeAll(hceCategoryVO.getCategoryName(),hceCategoryVO.getStartTime(),hceCategoryVO.getEndTime());
|
|
|
+ List<HceCategoryListVO> list = new ArrayList<>();
|
|
|
+ List<Integer> categoryIdList = new ArrayList<>();
|
|
|
+ if(StringUtils.isNotBlank(hceCategoryVO.getCategoryName())){
|
|
|
+ LambdaQueryWrapper<HceCategory> queryWrapper = Wrappers.lambdaQuery();
|
|
|
+ queryWrapper.like(HceCategory::getCategoryName, hceCategoryVO.getCategoryName());
|
|
|
+ List<HceCategory> list2 = this.list(queryWrapper);
|
|
|
+ if(list2.size() > 0){
|
|
|
+ for (int i = 0; i < list2.size(); i++) {
|
|
|
+ categoryIdList.add(list2.get(i).getId());
|
|
|
+ }
|
|
|
+ }
|
|
|
+ }
|
|
|
+ if(CollectionUtils.isNotEmpty(categoryIdList)){
|
|
|
+ list = baseMapper.selectMenuTreeAll(categoryIdList,hceCategoryVO.getStartTime(),hceCategoryVO.getEndTime());
|
|
|
+ }
|
|
|
|
|
|
List<HceCategoryListVO> finalList = list;
|
|
|
List<HceCategoryListVO> list1 = list.stream().filter(s ->s.getParentId() == 0).map(
|