权限管理-列表查询

fiveep-controller/src/main/java/com/bizmatics/controller/web/UserSiteController.java

@@ -1,12 +1,16 @@
 package com.bizmatics.controller.web;
 
 
+import com.bizmatics.common.core.bean.ApiResult;
+import com.bizmatics.model.system.SysMenu;
 import com.bizmatics.service.UserSiteService;
+import com.bizmatics.service.system.ISysMenuService;
 import org.springframework.beans.factory.annotation.Autowired;
-import org.springframework.web.bind.annotation.RequestMapping;
+import org.springframework.web.bind.annotation.*;
 
 import org.springframework.stereotype.Controller;
-import org.springframework.web.bind.annotation.RestController;
+
+import java.util.List;
 
 /**
  * 用户站点信息
@@ -20,5 +24,24 @@ public class UserSiteController {
 
     @Autowired
     private UserSiteService userSiteService;
+    @Autowired
+    private ISysMenuService menuService;
+
+
+    /**
+     * 根据菜单查询按钮
+     * @param menuName 菜单名称
+     * @return
+     */
+    @GetMapping(value = "/getButtonlistone")
+    public ApiResult<List<SysMenu>> getButtonlistone(@RequestParam(value = "menuName", required = false) String menuName){
+        return  ApiResult.success(menuService.getButtonlist(menuName));
+    }
+
+    @GetMapping(value = "/getAuthority")
+    public ApiResult<List<Object>> getAuthority(){
+        return  ApiResult.success(menuService.getAuthority());
+    }
+
 }
 

fiveep-model/src/main/java/com/bizmatics/model/vo/SysMenuOneVo.java

@@ -0,0 +1,39 @@
+package com.bizmatics.model.vo;
+
+import com.bizmatics.model.base.BaseEntity;
+import lombok.Data;
+import lombok.EqualsAndHashCode;
+import lombok.experimental.Accessors;
+import org.apache.commons.lang3.builder.ToStringBuilder;
+import org.apache.commons.lang3.builder.ToStringStyle;
+
+import javax.validation.constraints.NotBlank;
+import javax.validation.constraints.Size;
+import java.io.Serializable;
+import java.util.ArrayList;
+import java.util.List;
+
+
+/**
+ *
+ * 
+ * @author yq
+ */
+@Data
+@EqualsAndHashCode(callSuper = false)
+@Accessors(chain = true)
+public class SysMenuOneVo  implements Serializable
+{
+    private static final long serialVersionUID = 1L;
+
+    /** 菜单ID */
+    private Long id;
+
+    /** 菜单名称 */
+    private String menu;
+
+    /** 子菜单名称 */
+    private Object authority;
+
+    private Long fid;
+}

fiveep-model/src/main/java/com/bizmatics/model/vo/SysMenuTwoVo.java

@@ -0,0 +1,34 @@
+package com.bizmatics.model.vo;
+
+import lombok.Data;
+import lombok.EqualsAndHashCode;
+import lombok.experimental.Accessors;
+
+import java.io.Serializable;
+
+
+/**
+ * @author yq
+ */
+@Data
+@EqualsAndHashCode(callSuper = false)
+@Accessors(chain = true)
+public class SysMenuTwoVo implements Serializable {
+
+    /**
+     * 菜单ID
+     */
+    private Long id;
+
+    /**
+     * 菜单名称
+     */
+    private String name;
+
+    /**
+     * 父菜单名称
+     */
+    private Boolean checked;
+
+    private Long fid;
+}

fiveep-model/src/main/java/com/bizmatics/model/vo/SysMenuVo.java

@@ -0,0 +1,41 @@
+package com.bizmatics.model.vo;
+
+import com.bizmatics.model.base.BaseEntity;
+import lombok.Data;
+import lombok.EqualsAndHashCode;
+import lombok.experimental.Accessors;
+import org.apache.commons.lang3.builder.ToStringBuilder;
+import org.apache.commons.lang3.builder.ToStringStyle;
+
+import javax.validation.constraints.NotBlank;
+import javax.validation.constraints.Size;
+import java.io.Serializable;
+import java.util.ArrayList;
+import java.util.List;
+
+
+/**
+ *
+ * 
+ * @author yq
+ */
+@Data
+@EqualsAndHashCode(callSuper = false)
+@Accessors(chain = true)
+public class SysMenuVo implements Serializable
+{
+    private static final long serialVersionUID = 1L;
+
+    /** 菜单ID */
+    private Long id;
+
+    /** 菜单名称 */
+    private String menu;
+
+    /** 父菜单名称 */
+    private Object children;
+
+
+    private Long fid;
+
+}

fiveep-persistence/src/main/java/com/bizmatics/persistence/mapper/system/SysMenuMapper.java

@@ -61,6 +61,19 @@ public interface SysMenuMapper
      */
     public List<SysMenu> selectMenuTreeByUserId(Long userId);
 
+
+    /**
+     * 根据菜单名称获取按钮列表
+     *
+     * @param userId
+     * @return
+     */
+    public List<SysMenu> getButtonlist(Long userId, String menuName);
+
+    public List<SysMenu> getButtonlistOne(String menuName);
+
+    public List<SysMenu> getSysMenuList(String menuType);
+
     /**
      * 根据角色ID查询菜单树信息
      * 

fiveep-persistence/src/main/resources/mapper/mysql/system/SysMenuMapper.xml

@@ -27,10 +27,10 @@
 	</resultMap>
 
 	<sql id="selectMenuVo">
-        select menu_id, menu_name, parent_id, order_num, path, component, is_frame, is_cache, menu_type, visible, status, ifnull(perms,'') as perms, icon, create_time 
+        select menu_id, menu_name, parent_id, order_num, path, component, is_frame, is_cache, menu_type, visible, status, ifnull(perms,'') as perms, icon, create_time
 		from sys_menu
     </sql>
-    
+
     <select id="selectMenuList" parameterType="com.bizmatics.model.system.SysMenu" resultMap="SysMenuResult">
 		<include refid="selectMenuVo"/>
 		<where>
@@ -46,13 +46,13 @@
 		</where>
 		order by parent_id, order_num
 	</select>
-	
+
 	<select id="selectMenuTreeAll" resultMap="SysMenuResult">
 		select distinct m.menu_id, m.parent_id, m.menu_name, m.path, m.component, m.visible, m.status, ifnull(m.perms,'') as perms, m.is_frame, m.is_cache, m.menu_type, m.icon, m.order_num, m.create_time
 		from sys_menu m where m.menu_type in ('M', 'C') and m.status = 0
 		order by m.parent_id, m.order_num
 	</select>
-	
+
 	<select id="selectMenuListByUserId" parameterType="com.bizmatics.model.system.SysMenu" resultMap="SysMenuResult">
 		select distinct m.menu_id, m.parent_id, m.menu_name, m.path, m.component, m.visible, m.status, ifnull(m.perms,'') as perms, m.is_frame, m.is_cache, m.menu_type, m.icon, m.order_num, m.create_time
 		from sys_menu m
@@ -71,7 +71,7 @@
 		</if>
 		order by m.parent_id, m.order_num
 	</select>
-    
+
     <select id="selectMenuTreeByUserId" parameterType="Long" resultMap="SysMenuResult">
 		select distinct m.menu_id, m.parent_id, m.menu_name, m.path, m.component, m.visible, m.status, ifnull(m.perms,'') as perms, m.is_frame, m.is_cache, m.menu_type, m.icon, m.order_num, m.create_time
 		from sys_menu m
@@ -82,7 +82,81 @@
 		where u.user_id = #{userId} and m.menu_type in ('M', 'C') and m.status = 0  AND ro.status = 0
 		order by m.parent_id, m.order_num
 	</select>
-	
+
+	<select id="getButtonlist"  resultType="com.bizmatics.model.system.SysMenu" >
+		SELECT
+		a.*
+		FROM
+		sys_menu a
+		JOIN sys_role_menu b ON a.menu_id = b.menu_id
+		JOIN sys_user_role c ON b.role_id = c.role_id
+		JOIN sys_role d ON c.role_id = d.role_id
+		JOIN sys_user e ON c.user_id = e.user_id
+		WHERE
+		a. STATUS = 0
+		AND d. STATUS = 0
+		AND a.menu_type = 'F'
+		AND c.user_id = #{userId}
+		AND a.parent_id in (
+		SELECT
+		menu_id
+		FROM
+		sys_menu
+		WHERE
+		STATUS = 0
+		AND menu_type = 'C'
+		<if test="menuName != null and menuName != ''">
+			AND menu_name = #{menuName}
+		</if>
+		)
+		GROUP BY
+		a.menu_id
+		order by a.parent_id, a.order_num
+	</select>
+
+	<select id="getButtonlistOne"  resultType="com.bizmatics.model.system.SysMenu" >
+		SELECT
+			*
+		FROM
+			sys_menu
+		WHERE
+			`status` = 0
+		  AND menu_type = 'F'
+		  AND parent_id = (
+			SELECT
+				menu_id
+			FROM
+				sys_menu
+			WHERE
+				STATUS = 0
+			  AND menu_type = 'C'
+			AND menu_name in (#{menuName})
+		)
+		GROUP BY
+			menu_id
+		ORDER BY
+			parent_id,
+			order_num
+	</select>
+
+	<select id="getSysMenuList"  resultType="com.bizmatics.model.system.SysMenu" >
+		SELECT
+			*
+		FROM
+			sys_menu
+		WHERE
+			STATUS = 0
+		<if test="menuType != null and menuType != ''">
+		  AND menu_type = #{menuType}
+		</if>
+		GROUP BY
+		menu_id
+		ORDER BY
+		parent_id,
+		order_num
+	</select>
+
+
 	<select id="selectMenuListByRoleId" resultType="Integer">
 		select m.menu_id
 		from sys_menu m
@@ -93,7 +167,7 @@
             </if>
 		order by m.parent_id, m.order_num
 	</select>
-	
+
 	<select id="selectMenuPerms" resultType="String">
 		select distinct m.perms
 		from sys_menu m
@@ -109,21 +183,21 @@
 			 left join sys_role r on r.role_id = ur.role_id
 		where m.status = '0' and r.status = '0' and ur.user_id = #{userId}
 	</select>
-	
+
 	<select id="selectMenuById" parameterType="Long" resultMap="SysMenuResult">
 		<include refid="selectMenuVo"/>
 		where menu_id = #{menuId}
 	</select>
-	
+
 	<select id="hasChildByMenuId" resultType="Integer">
-	    select count(1) from sys_menu where parent_id = #{menuId}  
+	    select count(1) from sys_menu where parent_id = #{menuId}
 	</select>
-	
+
 	<select id="checkMenuNameUnique" parameterType="com.bizmatics.model.system.SysMenu" resultMap="SysMenuResult">
 		<include refid="selectMenuVo"/>
 		where menu_name=#{menuName} and parent_id = #{parentId} limit 1
 	</select>
-	
+
 	<update id="updateMenu" parameterType="com.bizmatics.model.system.SysMenu">
 		update sys_menu
 		<set>
@@ -183,7 +257,7 @@
 		sysdate()
 		)
 	</insert>
-	
+
 	<delete id="deleteMenuById" parameterType="Long">
 	    delete from sys_menu where menu_id = #{menuId}
 	</delete>

fiveep-service/src/main/java/com/bizmatics/service/system/ISysMenuService.java

@@ -82,6 +82,8 @@ public interface ISysMenuService
      */
     public List<TreeSelect> buildMenuTreeSelect(List<SysMenu> menus);
 
+    public List<Object> getAuthority();
+
     /**
      * 根据菜单ID查询信息
      * 
@@ -137,4 +139,7 @@ public interface ISysMenuService
      * @return 结果
      */
     public String checkMenuNameUnique(SysMenu menu);
+
+
+    public List<SysMenu> getButtonlist(String menuName);
 }

fiveep-service/src/main/java/com/bizmatics/service/system/impl/SysMenuServiceImpl.java

@@ -1,17 +1,20 @@
 package com.bizmatics.service.system.impl;
 
+import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
+import com.baomidou.mybatisplus.core.toolkit.Wrappers;
 import com.bizmatics.common.core.util.StringUtils;
+import com.bizmatics.model.PlatformArea;
 import com.bizmatics.model.base.TreeSelect;
 import com.bizmatics.model.constants.Constants;
 import com.bizmatics.model.constants.UserConstants;
 import com.bizmatics.model.system.SysMenu;
 import com.bizmatics.model.system.SysRole;
 import com.bizmatics.model.system.SysUser;
-import com.bizmatics.model.vo.MetaVo;
-import com.bizmatics.model.vo.RouterVo;
+import com.bizmatics.model.vo.*;
 import com.bizmatics.persistence.mapper.system.SysMenuMapper;
 import com.bizmatics.persistence.mapper.system.SysRoleMapper;
 import com.bizmatics.persistence.mapper.system.SysRoleMenuMapper;
+import com.bizmatics.service.impl.DeviceServiceImpl;
 import com.bizmatics.service.system.ISysMenuService;
 import com.bizmatics.service.util.SecurityUtils;
 import org.springframework.beans.factory.annotation.Autowired;
@@ -23,12 +26,11 @@ import java.util.stream.Collectors;
 
 /**
  * 菜单 业务层处理
- * 
+ *
  * @author yq
  */
 @Service
-public class SysMenuServiceImpl implements ISysMenuService
-{
+public class SysMenuServiceImpl implements ISysMenuService {
     public static final String PREMISSION_STRING = "perms[\"{0}\"]";
 
     @Autowired
@@ -40,35 +42,33 @@ public class SysMenuServiceImpl implements ISysMenuService
     @Autowired
     private SysRoleMenuMapper roleMenuMapper;
 
+    @Autowired
+    private ISysMenuService menuService;
+
     /**
      * 根据用户查询系统菜单列表
-     * 
+     *
      * @param userId 用户ID
      * @return 菜单列表
      */
     @Override
-    public List<SysMenu> selectMenuList(Long userId)
-    {
+    public List<SysMenu> selectMenuList(Long userId) {
         return selectMenuList(new SysMenu(), userId);
     }
 
     /**
      * 查询系统菜单列表
-     * 
+     *
      * @param menu 菜单信息
      * @return 菜单列表
      */
     @Override
-    public List<SysMenu> selectMenuList(SysMenu menu, Long userId)
-    {
+    public List<SysMenu> selectMenuList(SysMenu menu, Long userId) {
         List<SysMenu> menuList = null;
         // 管理员显示所有菜单信息
-        if (SysUser.isAdmin(userId))
-        {
+        if (SysUser.isAdmin(userId)) {
             menuList = menuMapper.selectMenuList(menu);
-        }
-        else
-        {
+        } else {
             menu.getParams().put("userId", userId);
             menuList = menuMapper.selectMenuListByUserId(menu);
         }
@@ -77,19 +77,16 @@ public class SysMenuServiceImpl implements ISysMenuService
 
     /**
      * 根据用户ID查询权限
-     * 
+     *
      * @param userId 用户ID
      * @return 权限列表
      */
     @Override
-    public Set<String> selectMenuPermsByUserId(Long userId)
-    {
+    public Set<String> selectMenuPermsByUserId(Long userId) {
         List<String> perms = menuMapper.selectMenuPermsByUserId(userId);
         Set<String> permsSet = new HashSet<>();
-        for (String perm : perms)
-        {
-            if (StringUtils.isNotBlank(perm))
-            {
+        for (String perm : perms) {
+            if (StringUtils.isNotBlank(perm)) {
                 permsSet.addAll(Arrays.asList(perm.trim().split(",")));
             }
         }
@@ -98,20 +95,16 @@ public class SysMenuServiceImpl implements ISysMenuService
 
     /**
      * 根据用户ID查询菜单
-     * 
+     *
      * @param userId 用户名称
      * @return 菜单列表
      */
     @Override
-    public List<SysMenu> selectMenuTreeByUserId(Long userId)
-    {
+    public List<SysMenu> selectMenuTreeByUserId(Long userId) {
         List<SysMenu> menus = null;
-        if (SecurityUtils.isAdmin(userId))
-        {
+        if (SecurityUtils.isAdmin(userId)) {
             menus = menuMapper.selectMenuTreeAll();
-        }
-        else
-        {
+        } else {
             menus = menuMapper.selectMenuTreeByUserId(userId);
         }
         return getChildPerms(menus, 0);
@@ -119,29 +112,26 @@ public class SysMenuServiceImpl implements ISysMenuService
 
     /**
      * 根据角色ID查询菜单树信息
-     * 
+     *
      * @param roleId 角色ID
      * @return 选中菜单列表
      */
     @Override
-    public List<Integer> selectMenuListByRoleId(Long roleId)
-    {
+    public List<Integer> selectMenuListByRoleId(Long roleId) {
         SysRole role = roleMapper.selectRoleById(roleId);
         return menuMapper.selectMenuListByRoleId(roleId, role.isMenuCheckStrictly());
     }
 
     /**
      * 构建前端路由所需要的菜单
-     * 
+     *
      * @param menus 菜单列表
      * @return 路由列表
      */
     @Override
-    public List<RouterVo> buildMenus(List<SysMenu> menus)
-    {
+    public List<RouterVo> buildMenus(List<SysMenu> menus) {
         List<RouterVo> routers = new LinkedList<RouterVo>();
-        for (SysMenu menu : menus)
-        {
+        for (SysMenu menu : menus) {
             RouterVo router = new RouterVo();
             router.setHidden("1".equals(menu.getVisible()));
             router.setName(getRouteName(menu));
@@ -149,19 +139,16 @@ public class SysMenuServiceImpl implements ISysMenuService
             router.setComponent(getComponent(menu));
             router.setMeta(new MetaVo(menu.getMenuName(), menu.getIcon(), StringUtils.equals("1", menu.getIsCache()), menu.getPath()));
             List<SysMenu> cMenus = menu.getChildren();
-            if (!cMenus.isEmpty() && cMenus.size() > 0 && UserConstants.TYPE_DIR.equals(menu.getMenuType()))
-            {
+            if (!cMenus.isEmpty() && cMenus.size() > 0 && UserConstants.TYPE_DIR.equals(menu.getMenuType())) {
                 router.setAlwaysShow(true);
                 //TODO
                 List<RouterVo> routerVos = buildMenus(cMenus);
                 router.setChildren(routerVos);
                 router.setRedirect(routerVos.get(0).getPath());
-                if (router.getMeta().getTitle().equals(routerVos.get(0).getMeta().getTitle())){
+                if (router.getMeta().getTitle().equals(routerVos.get(0).getMeta().getTitle())) {
                     routerVos.get(0).setHidden(true);
                 }
-            }
-            else if (isMenuFrame(menu))
-            {
+            } else if (isMenuFrame(menu)) {
                 router.setMeta(null);
                 List<RouterVo> childrenList = new ArrayList<RouterVo>();
                 RouterVo children = new RouterVo();
@@ -171,14 +158,12 @@ public class SysMenuServiceImpl implements ISysMenuService
                 children.setMeta(new MetaVo(menu.getMenuName(), menu.getIcon(), StringUtils.equals("1", menu.getIsCache()), menu.getPath()));
                 childrenList.add(children);
                 router.setChildren(childrenList);
-            }
-            else if (menu.getParentId().intValue() == 0 && isInnerLink(menu))
-            {
+            } else if (menu.getParentId().intValue() == 0 && isInnerLink(menu)) {
                 router.setMeta(new MetaVo(menu.getMenuName(), menu.getIcon()));
                 router.setPath("/inner");
                 List<RouterVo> childrenList = new ArrayList<RouterVo>();
                 RouterVo children = new RouterVo();
-                String routerPath = StringUtils.replaceEach(menu.getPath(), new String[] { Constants.HTTP, Constants.HTTPS }, new String[] { "", "" });
+                String routerPath = StringUtils.replaceEach(menu.getPath(), new String[]{Constants.HTTP, Constants.HTTPS}, new String[]{"", ""});
                 children.setPath(routerPath);
                 children.setComponent(UserConstants.INNER_LINK);
                 children.setName(StringUtils.capitalize(routerPath));
@@ -195,13 +180,12 @@ public class SysMenuServiceImpl implements ISysMenuService
     public List<RouterVo> buildMenusNew(List<SysMenu> menus) {
         List<RouterVo> routers = new LinkedList<RouterVo>();
         RouterVo routerOne;
-        for (SysMenu menu : menus)
-        {
+        for (SysMenu menu : menus) {
             List<RouterVo> routesList = new LinkedList<RouterVo>();
             routerOne = checkParent(menu, menus.indexOf(menu), true);
             RouterVo routerVo;
-            for (SysMenu sysMenu:menu.getChildren()) {
-                routerVo = checkParent(sysMenu,1,false);
+            for (SysMenu sysMenu : menu.getChildren()) {
+                routerVo = checkParent(sysMenu, 1, false);
                 routesList.add(routerVo);
             }
             routerOne.setChildren(routesList);
@@ -210,20 +194,20 @@ public class SysMenuServiceImpl implements ISysMenuService
         return routers;
     }
 
-    public RouterVo checkParent(SysMenu menu,Integer affix,Boolean isParent){
+    public RouterVo checkParent(SysMenu menu, Integer affix, Boolean isParent) {
         RouterVo router = new RouterVo();
         MetaVo metaVo = new MetaVo();
         router.setPath(menu.getPath());
         List<SysMenu> cMenus = menu.getChildren();
-        if (isParent){
+        if (isParent) {
             String redirect = cMenus.get(0).getComponent();
-            Integer index = redirect.indexOf("/",2);
-            redirect =redirect.substring(index,redirect.length());
-            if (!menu.getMenuName().equals("告警管理")){
+            Integer index = redirect.indexOf("/", 2);
+            redirect = redirect.substring(index, redirect.length());
+            if (!menu.getMenuName().equals("告警管理")) {
                 redirect = redirect.replaceAll("/index", "");
             }
             router.setRedirect(redirect);
-            if (0 == affix){
+            if (0 == affix) {
                 metaVo.setAffix(true);
             }
         }
@@ -236,38 +220,33 @@ public class SysMenuServiceImpl implements ISysMenuService
 
     public static void main(String[] args) {
         String url = "@/a/b/c/d/e";
-        Integer index = url.indexOf("/",2);
-        url =url.substring(index,url.length());
+        Integer index = url.indexOf("/", 2);
+        url = url.substring(index, url.length());
         System.out.println(url);
     }
 
     /**
      * 构建前端所需要树结构
-     * 
+     *
      * @param menus 菜单列表
      * @return 树结构列表
      */
     @Override
-    public List<SysMenu> buildMenuTree(List<SysMenu> menus)
-    {
+    public List<SysMenu> buildMenuTree(List<SysMenu> menus) {
         List<SysMenu> returnList = new ArrayList<SysMenu>();
         List<Long> tempList = new ArrayList<Long>();
-        for (SysMenu dept : menus)
-        {
+        for (SysMenu dept : menus) {
             tempList.add(dept.getMenuId());
         }
-        for (Iterator<SysMenu> iterator = menus.iterator(); iterator.hasNext();)
-        {
+        for (Iterator<SysMenu> iterator = menus.iterator(); iterator.hasNext(); ) {
             SysMenu menu = (SysMenu) iterator.next();
             // 如果是顶级节点, 遍历该父节点的所有子节点
-            if (!tempList.contains(menu.getParentId()))
-            {
+            if (!tempList.contains(menu.getParentId())) {
                 recursionFn(menus, menu);
                 returnList.add(menu);
             }
         }
-        if (returnList.isEmpty())
-        {
+        if (returnList.isEmpty()) {
             returnList = menus;
         }
         return returnList;
@@ -275,121 +254,216 @@ public class SysMenuServiceImpl implements ISysMenuService
 
     /**
      * 构建前端所需要下拉树结构
-     * 
+     *
      * @param menus 菜单列表
      * @return 下拉树结构列表
      */
     @Override
-    public List<TreeSelect> buildMenuTreeSelect(List<SysMenu> menus)
-    {
+    public List<TreeSelect> buildMenuTreeSelect(List<SysMenu> menus) {
         List<SysMenu> menuTrees = buildMenuTree(menus);
         return menuTrees.stream().map(TreeSelect::new).collect(Collectors.toList());
     }
 
+    @Override
+    public List<Object> getAuthority(){
+
+        SysUser user = SecurityUtils.getLoginUser().getUser();
+        //全部目录查询
+        List<SysMenu> sysMenuListTwo = menuMapper.getSysMenuList("M");
+        //全部菜单查询
+        List<SysMenu> sysMenuListOne = menuMapper.getSysMenuList("C");
+        //全部按钮列表
+        List<SysMenu> sysMenuListThree = menuMapper.getSysMenuList("F");
+        //已选中按钮列表
+        List<SysMenu> sysMenuListFour = menuMapper.getButtonlist(user.getUserId(), null);
+        List<SysMenuTwoVo> sysMenuVoListOne = new ArrayList<>();
+        List<SysMenuOneVo> SysMenuVoList = new ArrayList<>();
+        List<SysMenuVo> SysMenuVoListOne = new ArrayList<>();
+        List<Object> list = new ArrayList<>();
+        if (sysMenuListFour.size()>0){
+            for (int i=0;i<sysMenuListThree.size();i++){
+                SysMenuTwoVo sysMenuTwoVo = new SysMenuTwoVo();
+                sysMenuTwoVo.setId(sysMenuListThree.get(i).getMenuId());
+                sysMenuTwoVo.setName(sysMenuListThree.get(i).getMenuName());
+                sysMenuTwoVo.setFid(sysMenuListThree.get(i).getParentId());
+                long Id=sysMenuListThree.get(i).getMenuId();
+                for (int j=0;j<sysMenuListFour.size();j++){
+                    long MenuId=sysMenuListFour.get(j).getMenuId();
+                    if (Id==MenuId){
+                        sysMenuTwoVo.setChecked(true);
+                    }else {
+                        sysMenuTwoVo.setChecked(false);
+                    }
+                }
+                sysMenuVoListOne.add(sysMenuTwoVo);
+            }
+
+            for (int i=0;i<sysMenuListOne.size();i++){
+                SysMenuOneVo sysMenuOneVo = new SysMenuOneVo();
+                sysMenuOneVo.setId(sysMenuListOne.get(i).getMenuId());
+                sysMenuOneVo.setMenu(sysMenuListOne.get(i).getMenuName());
+                sysMenuOneVo.setFid(sysMenuListOne.get(i).getParentId());
+                long id = sysMenuListOne.get(i).getMenuId();
+                List<SysMenuTwoVo> sysMenuVoListTwo = new ArrayList<>();
+                for (int j=0;j<sysMenuVoListOne.size();j++){
+                    long FId = sysMenuVoListOne.get(j).getFid();
+                    if (id==FId){
+                        sysMenuVoListTwo.add(sysMenuVoListOne.get(j));
+                    }
+                }
+                sysMenuOneVo.setAuthority(sysMenuVoListTwo);
+                SysMenuVoList.add(sysMenuOneVo);
+            }
+
+            for (int i=0;i<sysMenuListTwo.size();i++){
+                SysMenuVo sysMenuVo = new SysMenuVo();
+                sysMenuVo.setId(sysMenuListTwo.get(i).getMenuId());
+                sysMenuVo.setMenu(sysMenuListTwo.get(i).getMenuName());
+                sysMenuVo.setFid(sysMenuListTwo.get(i).getParentId());
+                long id = sysMenuListTwo.get(i).getMenuId();
+                List<SysMenuOneVo> SysMenuVoListThree = new ArrayList<>();
+                for (int j=0;j<SysMenuVoList.size();j++){
+                    long FId = SysMenuVoList.get(j).getFid();
+                    if (id==FId){
+                        SysMenuVoListThree.add(SysMenuVoList.get(j));
+                    }
+                }
+                sysMenuVo.setChildren(SysMenuVoListThree);
+                SysMenuVoListOne.add(sysMenuVo);
+            }
+
+            for (int i=0;i<SysMenuVoListOne.size();i++){
+                String name = SysMenuVoListOne.get(i).getMenu();
+                if (name.equals("数据管理")||name.equals("电能质量")||name.equals("系统管理")){
+                    list.add(SysMenuVoListOne.get(i));
+                }else {
+                    for (int j=0;j<SysMenuVoList.size();j++){
+                        String name2 = SysMenuVoList.get(j).getMenu();
+                        if (name.equals(name2)){
+                            list.add(SysMenuVoList.get(j));
+                        }
+                    }
+                }
+            }
+        }
+        return list;
+    }
+
+
     /**
      * 根据菜单ID查询信息
-     * 
+     *
      * @param menuId 菜单ID
      * @return 菜单信息
      */
     @Override
-    public SysMenu selectMenuById(Long menuId)
-    {
+
+    public SysMenu selectMenuById(Long menuId) {
         return menuMapper.selectMenuById(menuId);
     }
 
     /**
      * 是否存在菜单子节点
-     * 
+     *
      * @param menuId 菜单ID
      * @return 结果
      */
     @Override
-    public boolean hasChildByMenuId(Long menuId)
-    {
+    public boolean hasChildByMenuId(Long menuId) {
         int result = menuMapper.hasChildByMenuId(menuId);
         return result > 0 ? true : false;
     }
 
     /**
      * 查询菜单使用数量
-     * 
+     *
      * @param menuId 菜单ID
      * @return 结果
      */
     @Override
-    public boolean checkMenuExistRole(Long menuId)
-    {
+    public boolean checkMenuExistRole(Long menuId) {
         int result = roleMenuMapper.checkMenuExistRole(menuId);
         return result > 0 ? true : false;
     }
 
     /**
      * 新增保存菜单信息
-     * 
+     *
      * @param menu 菜单信息
      * @return 结果
      */
     @Override
-    public int insertMenu(SysMenu menu)
-    {
+    public int insertMenu(SysMenu menu) {
         return menuMapper.insertMenu(menu);
     }
 
     /**
      * 修改保存菜单信息
-     * 
+     *
      * @param menu 菜单信息
      * @return 结果
      */
     @Override
-    public int updateMenu(SysMenu menu)
-    {
+    public int updateMenu(SysMenu menu) {
         return menuMapper.updateMenu(menu);
     }
 
     /**
      * 删除菜单管理信息
-     * 
+     *
      * @param menuId 菜单ID
      * @return 结果
      */
     @Override
-    public int deleteMenuById(Long menuId)
-    {
+    public int deleteMenuById(Long menuId) {
         return menuMapper.deleteMenuById(menuId);
     }
 
     /**
      * 校验菜单名称是否唯一
-     * 
+     *
      * @param menu 菜单信息
      * @return 结果
      */
     @Override
-    public String checkMenuNameUnique(SysMenu menu)
-    {
+    public String checkMenuNameUnique(SysMenu menu) {
         Long menuId = Objects.isNull(menu.getMenuId()) ? -1L : menu.getMenuId();
         SysMenu info = menuMapper.checkMenuNameUnique(menu.getMenuName(), menu.getParentId());
-        if (Objects.nonNull(info) && info.getMenuId().longValue() != menuId.longValue())
-        {
+        if (Objects.nonNull(info) && info.getMenuId().longValue() != menuId.longValue()) {
             return UserConstants.NOT_UNIQUE;
         }
         return UserConstants.UNIQUE;
     }
 
+    /**
+     * 根据菜单查询按钮
+     *
+     * @param menuName 菜单名称
+     * @return
+     */
+    @Override
+    public List<SysMenu> getButtonlist(String menuName) {
+        SysUser user = SecurityUtils.getLoginUser().getUser();
+        List<SysMenu> sysMenuList = new ArrayList<>();
+        if (SecurityUtils.isAdmin(user.getUserId())) {
+            sysMenuList = menuMapper.getButtonlistOne(menuName);
+        } else {
+            sysMenuList = menuMapper.getButtonlist(user.getUserId(), menuName);
+        }
+        return sysMenuList;
+    }
+
+
     /**
      * 获取路由名称
-     * 
+     *
      * @param menu 菜单信息
      * @return 路由名称
      */
-    public String getRouteName(SysMenu menu)
-    {
+    public String getRouteName(SysMenu menu) {
         String routerName = StringUtils.capitalize(menu.getPath());
         // 非外链并且是一级目录（类型为目录）
-        if (isMenuFrame(menu))
-        {
+        if (isMenuFrame(menu)) {
             routerName = StringUtils.EMPTY;
         }
         return routerName;
@@ -397,27 +471,23 @@ public class SysMenuServiceImpl implements ISysMenuService
 
     /**
      * 获取路由地址
-     * 
+     *
      * @param menu 菜单信息
      * @return 路由地址
      */
-    public String getRouterPath(SysMenu menu)
-    {
+    public String getRouterPath(SysMenu menu) {
         String routerPath = menu.getPath();
         // 内链打开外网方式
-        if (menu.getParentId().intValue() != 0 && isInnerLink(menu))
-        {
-            routerPath = StringUtils.replaceEach(routerPath, new String[] { Constants.HTTP, Constants.HTTPS }, new String[] { "", "" });
+        if (menu.getParentId().intValue() != 0 && isInnerLink(menu)) {
+            routerPath = StringUtils.replaceEach(routerPath, new String[]{Constants.HTTP, Constants.HTTPS}, new String[]{"", ""});
         }
         // 非外链并且是一级目录（类型为目录）
         if (0 == menu.getParentId().intValue() && UserConstants.TYPE_DIR.equals(menu.getMenuType())
-                && UserConstants.NO_FRAME.equals(menu.getIsFrame()))
-        {
+                && UserConstants.NO_FRAME.equals(menu.getIsFrame())) {
             routerPath = "/" + menu.getPath();
         }
         // 非外链并且是一级目录（类型为菜单）
-        else if (isMenuFrame(menu))
-        {
+        else if (isMenuFrame(menu)) {
             routerPath = "/";
         }
         return routerPath;
@@ -425,23 +495,17 @@ public class SysMenuServiceImpl implements ISysMenuService
 
     /**
      * 获取组件信息
-     * 
+     *
      * @param menu 菜单信息
      * @return 组件信息
      */
-    public String getComponent(SysMenu menu)
-    {
+    public String getComponent(SysMenu menu) {
         String component = UserConstants.LAYOUT;
-        if (StringUtils.isNotEmpty(menu.getComponent()) && !isMenuFrame(menu))
-        {
+        if (StringUtils.isNotEmpty(menu.getComponent()) && !isMenuFrame(menu)) {
             component = menu.getComponent();
-        }
-        else if (StringUtils.isEmpty(menu.getComponent()) && menu.getParentId().intValue() != 0 && isInnerLink(menu))
-        {
+        } else if (StringUtils.isEmpty(menu.getComponent()) && menu.getParentId().intValue() != 0 && isInnerLink(menu)) {
             component = UserConstants.INNER_LINK;
-        }
-        else if (StringUtils.isEmpty(menu.getComponent()) && isParentView(menu))
-        {
+        } else if (StringUtils.isEmpty(menu.getComponent()) && isParentView(menu)) {
             component = UserConstants.PARENT_VIEW;
         }
         return component;
@@ -449,54 +513,48 @@ public class SysMenuServiceImpl implements ISysMenuService
 
     /**
      * 是否为菜单内部跳转
-     * 
+     *
      * @param menu 菜单信息
      * @return 结果
      */
-    public boolean isMenuFrame(SysMenu menu)
-    {
+    public boolean isMenuFrame(SysMenu menu) {
         return menu.getParentId().intValue() == 0 && UserConstants.TYPE_MENU.equals(menu.getMenuType())
                 && menu.getIsFrame().equals(UserConstants.NO_FRAME);
     }
 
     /**
      * 是否为内链组件
-     * 
+     *
      * @param menu 菜单信息
      * @return 结果
      */
-    public boolean isInnerLink(SysMenu menu)
-    {
+    public boolean isInnerLink(SysMenu menu) {
         return menu.getIsFrame().equals(UserConstants.NO_FRAME) && StringUtils.startsWithAny(menu.getPath(), Constants.HTTP, Constants.HTTPS);
     }
 
     /**
      * 是否为parent_view组件
-     * 
+     *
      * @param menu 菜单信息
      * @return 结果
      */
-    public boolean isParentView(SysMenu menu)
-    {
+    public boolean isParentView(SysMenu menu) {
         return menu.getParentId().intValue() != 0 && UserConstants.TYPE_DIR.equals(menu.getMenuType());
     }
 
     /**
      * 根据父节点的ID获取所有子节点
-     * 
-     * @param list 分类表
+     *
+     * @param list     分类表
      * @param parentId 传入的父节点ID
      * @return String
      */
-    public List<SysMenu> getChildPerms(List<SysMenu> list, int parentId)
-    {
+    public List<SysMenu> getChildPerms(List<SysMenu> list, int parentId) {
         List<SysMenu> returnList = new ArrayList<SysMenu>();
-        for (Iterator<SysMenu> iterator = list.iterator(); iterator.hasNext();)
-        {
+        for (Iterator<SysMenu> iterator = list.iterator(); iterator.hasNext(); ) {
             SysMenu t = (SysMenu) iterator.next();
             // 一、根据传入的某个父节点ID,遍历该父节点的所有子节点
-            if (t.getParentId() == parentId)
-            {
+            if (t.getParentId() == parentId) {
                 recursionFn(list, t);
                 returnList.add(t);
             }
@@ -506,19 +564,16 @@ public class SysMenuServiceImpl implements ISysMenuService
 
     /**
      * 递归列表
-     * 
+     *
      * @param list
      * @param t
      */
-    private void recursionFn(List<SysMenu> list, SysMenu t)
-    {
+    private void recursionFn(List<SysMenu> list, SysMenu t) {
         // 得到子节点列表
         List<SysMenu> childList = getChildList(list, t);
         t.setChildren(childList);
-        for (SysMenu tChild : childList)
-        {
-            if (hasChild(list, tChild))
-            {
+        for (SysMenu tChild : childList) {
+            if (hasChild(list, tChild)) {
                 recursionFn(list, tChild);
             }
         }
@@ -527,15 +582,12 @@ public class SysMenuServiceImpl implements ISysMenuService
     /**
      * 得到子节点列表
      */
-    private List<SysMenu> getChildList(List<SysMenu> list, SysMenu t)
-    {
+    private List<SysMenu> getChildList(List<SysMenu> list, SysMenu t) {
         List<SysMenu> tlist = new ArrayList<SysMenu>();
         Iterator<SysMenu> it = list.iterator();
-        while (it.hasNext())
-        {
+        while (it.hasNext()) {
             SysMenu n = (SysMenu) it.next();
-            if (n.getParentId().longValue() == t.getMenuId().longValue())
-            {
+            if (n.getParentId().longValue() == t.getMenuId().longValue()) {
                 tlist.add(n);
             }
         }
@@ -545,8 +597,8 @@ public class SysMenuServiceImpl implements ISysMenuService
     /**
      * 判断是否有子节点
      */
-    private boolean hasChild(List<SysMenu> list, SysMenu t)
-    {
+    private boolean hasChild(List<SysMenu> list, SysMenu t) {
         return getChildList(list, t).size() > 0 ? true : false;
     }
+
 }