基金分析平台 springboot - mybatis-plus .zip

package com.github.fashionbrot.funds.service; import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; import com.baomidou.mybatisplus.core.toolkit.CollectionUtils; import com.github.fashionbrot.funds.annotation.IsMenu; import com.github.fashionbrot.funds.annotation.MarsPermission; import com.github.fashionbrot.funds.dao.MenuDao; import com.github.fashionbrot.funds.dao.MenuRoleRelationDao; import com.github.fashionbrot.funds.entity.Menu; import com.github.fashionbrot.funds.entity.MenuRoleRelation; import com.github.fashionbrot.funds.enums.RespCode; import com.github.fashionbrot.funds.exception.MarsException; import com.github.fashionbrot.funds.model.LoginModel; import com.github.fashionbrot.funds.vo.PageDataVo; import com.github.fashionbrot.funds.vo.RespVo; import com.github.pagehelper.Page; import com.github.pagehelper.PageHelper; import lombok.extern.slf4j.Slf4j; import org.apache.commons.lang3.StringUtils; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.stereotype.Service; import org.springframework.transaction.annotation.Transactional; import org.springframework.web.method.HandlerMethod; import javax.servlet.http.HttpServletRequest; import java.lang.reflect.Method; import java.util.*; import java.util.concurrent.ConcurrentHashMap; /** * @author fashionbrot * @version 0.1.0 * @date 2019/12/8 22:45 */ @Service @Slf4j public class MenuService { @Autowired private MenuDao menuDao; @Autowired private MenuRoleRelationDao menuRoleRelationDao; @Autowired private UserInfoService userInfoService; @Autowired private UserLoginService userLoginService; private static Map<Long, List<Menu>> MENU_LIST_MAP = new ConcurrentHashMap<Long, List<Menu>>(); /** * true 有权限 false 无权限 * * @return */ public boolean checkPermissionUrl(Object handler, HttpServletRequest request) { LoginModel model = userLoginService.getLogin(); if (handler instanceof HandlerMethod) { //如果是超级管理员 if (model!=null && model.isSuperAdmin()){ return true; } String requestUrl = request.getRequestURI(); HandlerMethod handlerMethod = (HandlerMethod) handler; Method method = handlerMethod.getMethod(); // 判断接口是否需要登录 IsMenu methodAnnotation = method.getAnnotation(IsMenu.class); if (methodAnnotation != null && methodAnnotation.checkMenuUrlPermission() ) { List<Menu> menuBarList = getMenus(model); if (CollectionUtils.isNotEmpty(menuBarList)) { for (Menu m : menuBarList) { //验证菜单是否有权限 if (m.getMenuLevel() == 2 && equRequestUrl(m.getMenuUrl(), requestUrl)) { return true; } } } return false; } MarsPermission permission= method.getAnnotation(MarsPermission.class); if(permission!=null) { String permissionCode =permission!=null?permission.value():null; ; List<Menu> menuBarList = getMenus(model); if (CollectionUtils.isNotEmpty(menuBarList)) { for (Menu m : menuBarList) { ////验证按钮是否有权限 if (m.getMenuLevel()==3 && StringUtils.isNotBlank(permissionCode)&& permissionCode.equals(m.getCode())){ return true; } } } return false; } return true; } return false; } /** * 清除菜单缓存 */ public void clearMenuList() { MENU_LIST_MAP.clear(); } public void add(Menu appInfo) { if (menuDao.add(appInfo) != 1) { throw new MarsException(RespCode.SAVE_ERROR); } } public void update(Menu appInfo) { Integer result = menuDao.update(appInfo); if (result == 1) { clearMenuList(); } else { throw new MarsException(RespCode.UPDATE_ERROR); } } public void deleteById(Long id) { if (menuDao.deleteById(id) != 1) { throw new MarsException(RespCode.DELETE_ERROR); } } public Menu queryById(Long id) { return menuDao.queryById(id); } public List<Menu> queryAll() { QueryWrapper<Menu> queryWrapper=new QueryWrapper(); queryWrapper.groupBy("priority "); List<Menu> menuBarList = menuDao.queryAll(queryWrapper); if (CollectionUtils.isNotEmpty(menuBarList)) { for (Menu m : menuBarList) { if (m.getMenuLevel() != 1) { m.setParentMenuName(parentMenuName(menuBarList, m.getParentMenuId())); } } } return menuBarList; } public PageDataVo queryAll(Integer page, Integer pageSize) { QueryWrapper<Menu> queryWrapper=new QueryWrapper(); queryWrapper.orderByAsc("priority "); Page<Object> objects = PageHelper.startPage(page, pageSize); List<Menu> menuBarList = menuDao.queryAll(queryWrapper); if (CollectionUtils.isNotEmpty(menuBarList)) { for (Menu m : menuBarList) { if (m.getMenuLevel() != 1) { m.setParentMenuName(parentMenuName(menuBarList, m.getParentMenuId())); } } } PageDataVo<Menu> pageData = new PageDataVo<Menu>(); pageData.setData(menuBarList); pageData.setITotalDisplayRecords(objects.getTotal()); return pageData; } private String parentMenuName(List<Menu> menuBarList, Long parentMenuId) { if (CollectionUtils.isNotEmpty(menuBarList)) { for (Menu m : menuBarList) { if (m.getMenuLevel() == 1 && Objects.equals(m.getId(), parentMenuId)) { return m.getMenuName(); }else if (m.getMenuLevel() == 2 && Objects.equals(m.getId(), parentMenuId)) { return m.getMenuName(); } } } return ""; } public List<Menu> queryMenuLevel(Menu menuBar) { return menuDao.queryAll(new QueryWrapper<Menu>().eq("menu_level", menuBar.getMenuLevel())); } public void loadMenuList(Object handler, HttpServletRequest request) { if (handler instanceof HandlerMethod) { LoginModel model = userLoginService.getLogin(); if (model == null) { log.info(" loadMenuList userInfo is null"); return; } HandlerMethod handlerMethod = (HandlerMethod) handler; Method method = handlerMethod.getMethod(); String requestUrl = request.getRequestURI(); // 判断接口是否需要登录 IsMenu methodAnnotation = method.getAnnotation(IsMenu.class); Optional.ofNullable(methodAnnotation).ifPresent(mm -> { List<Menu> menuBarList = getMenus(model); if (CollectionUtils.isNotEmpty(menuBarList)) { menuBarList = searchChildMenu(menuBarList, requestUrl); } request.setAttribute("menuList", menuBarList); }); } } public List<Menu> getMenus(LoginModel model) { Long userId = model.getUserId(); List<Menu> menuBarList; if (CollectionUtils.isNotEmpty(MENU_LIST_MAP) && MENU_LIST_MAP.containsKey(userId)) { menuBarList = MENU_LIST_MAP.get(userId); } else { if (model.isSuperAdmin()){ menuBarList = menuDao.queryAll(null); }else{ menuBarList