| | |
| | | import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; |
| | | import com.ycl.dto.UmsMenuNode; |
| | | import com.ycl.entity.user.UmsMenu; |
| | | import com.ycl.entity.user.vo.UmsMenuVo; |
| | | import com.ycl.mapper.user.AdminMenuRelationMapper; |
| | | import com.ycl.mapper.user.UmsMenuMapper; |
| | | import com.ycl.service.user.UmsMenuService; |
| | | import org.springframework.beans.BeanUtils; |
| | | import org.springframework.stereotype.Service; |
| | | |
| | | import javax.annotation.Resource; |
| | | import java.util.Date; |
| | | import java.util.List; |
| | | import java.util.Objects; |
| | | import java.util.stream.Collectors; |
| | | |
| | | /** |
| | |
| | | @Service |
| | | public class UmsMenuServiceImpl extends ServiceImpl<UmsMenuMapper, UmsMenu>implements UmsMenuService { |
| | | |
| | | @Resource |
| | | AdminMenuRelationMapper adminMenuRelationMapper; |
| | | @Resource |
| | | UmsMenuMapper umsMenuMapper; |
| | | |
| | | @Override |
| | | public boolean create(UmsMenu umsMenu) { |
| | | umsMenu.setCreateTime(new Date()); |
| | | updateLevel(umsMenu); |
| | | return save(umsMenu); |
| | | } |
| | | |
| | | |
| | | /** |
| | | * 修改菜单层级 |
| | |
| | | return updateById(umsMenu); |
| | | } |
| | | |
| | | @Override |
| | | public List<UmsMenuVo> ListTreeMyself(Long id) { |
| | | List<UmsMenuVo> fatherUmsMenuVoList = umsMenuMapper.adminMenuRelation(id); |
| | | List<UmsMenuVo> umsMenuVoList = fatherUmsMenuVoList.stream() |
| | | .filter(item -> item.getParentId() == 0L) |
| | | .peek(item -> item.setMenuVoArrayList(getChildren(item, fatherUmsMenuVoList))) |
| | | .collect(Collectors.toList()); |
| | | return umsMenuVoList; |
| | | } |
| | | |
| | | public List<UmsMenuVo> getChildren(UmsMenuVo root, List<UmsMenuVo> allMenus) { |
| | | return allMenus.stream() |
| | | .filter(menu -> Objects.equals(menu.getParentId(), root.getId())) |
| | | .peek(menu -> menu.setMenuVoArrayList(getChildren(menu, allMenus))) |
| | | .collect(Collectors.toList()); |
| | | } |
| | | |
| | | /** |
| | | * 将UmsMenu转化为UmsMenuNode并设置children属性 |
| | | */ |