From 220d98241f48d1891cb8d58f8a8081b517171f0e Mon Sep 17 00:00:00 2001
From: Pu Zhibing <393733352@qq.com>
Date: 星期三, 18 十二月 2024 18:00:14 +0800
Subject: [PATCH] 修改菜单功能

---
 ruoyi-modules/ruoyi-system/src/main/java/com/ruoyi/system/mapper/SysMenuMapper.java            |    2 +-
 ruoyi-modules/ruoyi-system/src/main/java/com/ruoyi/system/domain/SysMenu.java                  |   10 ++++++++++
 ruoyi-modules/ruoyi-system/src/main/resources/mapping/system/SysMenuMapper.xml                 |    4 ++--
 ruoyi-modules/ruoyi-system/src/main/java/com/ruoyi/system/service/impl/SysMenuServiceImpl.java |    4 +++-
 4 files changed, 16 insertions(+), 4 deletions(-)

diff --git a/ruoyi-modules/ruoyi-system/src/main/java/com/ruoyi/system/domain/SysMenu.java b/ruoyi-modules/ruoyi-system/src/main/java/com/ruoyi/system/domain/SysMenu.java
index 340d7ef..bf919e8 100644
--- a/ruoyi-modules/ruoyi-system/src/main/java/com/ruoyi/system/domain/SysMenu.java
+++ b/ruoyi-modules/ruoyi-system/src/main/java/com/ruoyi/system/domain/SysMenu.java
@@ -72,6 +72,8 @@
 
     /** 菜单图标 */
     private String icon;
+    @TableField("role_type")
+    private String roleType;
 
     /** 子菜单 */
     @TableField(exist = false)
@@ -244,6 +246,14 @@
         this.children = children;
     }
     
+    public String getRoleType() {
+        return roleType;
+    }
+    
+    public void setRoleType(String roleType) {
+        this.roleType = roleType;
+    }
+    
     @Override
     public String toString() {
         return new ToStringBuilder(this,ToStringStyle.MULTI_LINE_STYLE)
diff --git a/ruoyi-modules/ruoyi-system/src/main/java/com/ruoyi/system/mapper/SysMenuMapper.java b/ruoyi-modules/ruoyi-system/src/main/java/com/ruoyi/system/mapper/SysMenuMapper.java
index 4c9dd4a..db1bfe7 100644
--- a/ruoyi-modules/ruoyi-system/src/main/java/com/ruoyi/system/mapper/SysMenuMapper.java
+++ b/ruoyi-modules/ruoyi-system/src/main/java/com/ruoyi/system/mapper/SysMenuMapper.java
@@ -130,7 +130,7 @@
 
     List<SysMenus> getAllInIds(@Param("menusId") List<Long> menusId);
 
-    List<SysMenus> getAllOne();
+    List<SysMenus> getAllOne(@Param("roleType") Integer roleType);
 
 
 
diff --git a/ruoyi-modules/ruoyi-system/src/main/java/com/ruoyi/system/service/impl/SysMenuServiceImpl.java b/ruoyi-modules/ruoyi-system/src/main/java/com/ruoyi/system/service/impl/SysMenuServiceImpl.java
index 607db4f..b9d09a4 100644
--- a/ruoyi-modules/ruoyi-system/src/main/java/com/ruoyi/system/service/impl/SysMenuServiceImpl.java
+++ b/ruoyi-modules/ruoyi-system/src/main/java/com/ruoyi/system/service/impl/SysMenuServiceImpl.java
@@ -7,6 +7,7 @@
 import com.ruoyi.common.security.service.TokenService;
 import com.ruoyi.common.security.utils.SecurityUtils;
 import com.ruoyi.system.api.domain.SysRole;
+import com.ruoyi.system.api.domain.SysUser;
 import com.ruoyi.system.domain.SysMenu;
 import com.ruoyi.system.domain.SysMenus;
 import com.ruoyi.system.domain.SysUserRole;
@@ -353,11 +354,12 @@
     @Override
     public List<SysMenus> getAllMenu() {
         Long userid = tokenService.getLoginUser().getUserid();
+        SysUser sysUser = sysUserMapper.selectById(userid);
         SysUserRole sysUserRole = sysUserRoleMapper.selectSysUserRoleByUserId(userid);
         Long roleId = sysUserRole.getRoleId();
         List<SysMenus> list=null;
         if(roleId!=1){
-            list = menuMapper.getAllOne();
+            list = menuMapper.getAllOne(sysUser.getRoleType());
         }else {
              list = menuMapper.getAll();
         }
diff --git a/ruoyi-modules/ruoyi-system/src/main/resources/mapping/system/SysMenuMapper.xml b/ruoyi-modules/ruoyi-system/src/main/resources/mapping/system/SysMenuMapper.xml
index 75db7c8..eb18cf3 100644
--- a/ruoyi-modules/ruoyi-system/src/main/resources/mapping/system/SysMenuMapper.xml
+++ b/ruoyi-modules/ruoyi-system/src/main/resources/mapping/system/SysMenuMapper.xml
@@ -152,7 +152,7 @@
 			update_by  updateBy,
 			update_time updateTime,
 			remark
- from sys_menu
+ from sys_menu where role_type = 1
 		order by create_time desc
 	</select>
     <select id="getAllInIds" resultType="com.ruoyi.system.domain.SysMenus">
@@ -201,7 +201,7 @@
 			update_by  updateBy,
 			update_time updateTime,
 			remark
- from sys_menu where menu_id !=1074 and  menu_id !=1075 and  menu_id !=1193 and  menu_id !=1194
+ from sys_menu where role_type = #{roleType}
                order by create_time desc
 	</select>
 

--
Gitblit v1.7.1