| package doumeemes.service.system.impl; | 
|   | 
| import doumeemes.dao.system.SystemMenuMapper; | 
| import doumeemes.dao.system.dto.QuerySystemMenuDTO; | 
| import doumeemes.dao.system.model.SystemMenu; | 
| import doumeemes.dao.system.vo.SystemMenuListVO; | 
| import doumeemes.service.system.SystemMenuService; | 
| import com.baomidou.mybatisplus.core.conditions.Wrapper; | 
| import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; | 
| import org.springframework.beans.factory.annotation.Autowired; | 
| import org.springframework.stereotype.Service; | 
| import org.springframework.transaction.annotation.Transactional; | 
| import org.springframework.util.CollectionUtils; | 
|   | 
| import java.util.ArrayList; | 
| import java.util.Arrays; | 
| import java.util.List; | 
| import java.util.stream.Collectors; | 
|   | 
| /** | 
|  * 系统菜单Service实现 | 
|  * @author Eva.Caesar Liu | 
|  * @date 2022/04/18 18:12 | 
|  */ | 
| @Service | 
| public class SystemMenuServiceImpl implements SystemMenuService { | 
|   | 
|     @Autowired | 
|     private SystemMenuMapper systemMenuMapper; | 
|   | 
|     @Override | 
|     public Integer create(SystemMenu systemMenu) { | 
|         systemMenuMapper.insert(systemMenu); | 
|         return systemMenu.getId(); | 
|     } | 
|   | 
|     @Override | 
|     public void deleteById(Integer id) { | 
|         SystemMenu systemMenu = new SystemMenu(); | 
|         systemMenu.setId(id); | 
|         systemMenu.setDeleted(Boolean.TRUE); | 
|         this.updateById(systemMenu); | 
|     } | 
|   | 
|     @Override | 
|     @Transactional | 
|     public void deleteByIdInBatch(List<Integer> ids) { | 
|         if (CollectionUtils.isEmpty(ids)) return; | 
|         for (Integer id : ids) { | 
|             this.deleteById(id); | 
|         } | 
|     } | 
|   | 
|     @Override | 
|     public void updateById(SystemMenu systemMenu) { | 
|         systemMenuMapper.updateById(systemMenu); | 
|     } | 
|   | 
|     @Override | 
|     @Transactional | 
|     public void updateByIdInBatch(List<SystemMenu> systemMenus) { | 
|         if (CollectionUtils.isEmpty(systemMenus)) return; | 
|         for (SystemMenu systemMenu: systemMenus) { | 
|             this.updateById(systemMenu); | 
|         } | 
|     } | 
|   | 
|     @Override | 
|     public SystemMenu findById(Integer id) { | 
|         return systemMenuMapper.selectById(id); | 
|     } | 
|   | 
|     @Override | 
|     public SystemMenu findOne(SystemMenu systemMenu) { | 
|         Wrapper<SystemMenu> wrapper = new QueryWrapper<>(systemMenu); | 
|         return systemMenuMapper.selectOne(wrapper); | 
|     } | 
|   | 
|     @Override | 
|     public List<SystemMenu> findList(SystemMenu systemMenu) { | 
|         QueryWrapper<SystemMenu> wrapper = new QueryWrapper<>(systemMenu); | 
|         wrapper.lambda().orderByAsc(SystemMenu::getSort); | 
|         return systemMenuMapper.selectList(wrapper); | 
|     } | 
|   | 
|     @Override | 
|     public List<SystemMenuListVO> findList(QuerySystemMenuDTO dto ) { | 
|         return systemMenuMapper.selectManageList(dto); | 
|     } | 
|   | 
|     @Override | 
|     public List<SystemMenu> findRootList(int type) { | 
|         QueryWrapper<SystemMenu> queryWrapper = new QueryWrapper<>(); | 
|         queryWrapper.lambda() | 
|                 .isNull(SystemMenu::getParentId) | 
|                 .eq(SystemMenu::getDeleted, Boolean.FALSE) | 
|                 .eq(SystemMenu::getType, type) | 
|                 .orderByAsc(SystemMenu::getSort); | 
|         return systemMenuMapper.selectList(queryWrapper); | 
|     } | 
|   | 
|     @Override | 
|     public List<SystemMenu> findByUserId(Integer userId,Integer type,Integer comId) { | 
|         return systemMenuMapper.selectByUserId(userId,type,comId); | 
|     } | 
|   | 
|     @Override | 
|     public List<SystemMenu> findByRoleId(Integer roleId) { | 
|         return systemMenuMapper.selectByRoleId(roleId); | 
|     } | 
|   | 
|     @Override | 
|     public long count(SystemMenu systemMenu) { | 
|         Wrapper<SystemMenu> wrapper = new QueryWrapper<>(systemMenu); | 
|         return systemMenuMapper.selectCount(wrapper); | 
|     } | 
|   | 
|     @Override | 
|     public List<Integer> findChildren(Integer menuId) { | 
|         List<Integer> pool = new ArrayList<>(); | 
|         this.fillChildren(pool, Arrays.asList(menuId)); | 
|         return pool; | 
|     } | 
|   | 
|     /** | 
|      * 获取子部门ID | 
|      * @author Eva.Caesar Liu | 
|      * @date 2022/04/18 18:12 | 
|      */ | 
|     private void fillChildren(List<Integer> pool, List<Integer> parentIds) { | 
|         QueryWrapper<SystemMenu> queryWrapper = new QueryWrapper<>(); | 
|         queryWrapper.lambda() | 
|                 .eq(SystemMenu::getDeleted, Boolean.FALSE) | 
|                 .in(SystemMenu::getParentId, parentIds); | 
|         List<SystemMenu> departments = systemMenuMapper.selectList(queryWrapper); | 
|         List<Integer> ids = departments.stream().map(SystemMenu::getId).collect(Collectors.toList()); | 
|         if (ids.size() > 0) { | 
|             pool.addAll(ids); | 
|             this.fillChildren(pool, ids); | 
|         } | 
|     } | 
| } |