菜单权限控制

coolstore-partner-dao/src/main/java/com/cool/store/dao/SysRoleMenuDAO.java

@@ -1,7 +1,15 @@
 package com.cool.store.dao;
 
+import com.aliyun.openservices.shade.org.apache.commons.lang3.StringUtils;
+import com.cool.store.entity.SysRoleMenuDO;
+import com.cool.store.mapper.SysRoleMenuMapper;
+import com.google.common.collect.Lists;
 import org.springframework.stereotype.Repository;
 
+import javax.annotation.Resource;
+import java.util.List;
+import java.util.Objects;
+
 /**
  * @author zhangchenbiao
  * @date 2023-05-19 03:01
@@ -9,4 +17,20 @@ import org.springframework.stereotype.Repository;
 @Repository
 public class SysRoleMenuDAO {
 
+    @Resource
+    private SysRoleMenuMapper sysRoleMenuMapper;
+
+    /**
+     * 获取菜单
+     * @param roleId
+     * @param platform
+     * @return
+     */
+    public List<SysRoleMenuDO> getRoleMenuByRoleId(String roleId, String platform){
+        if(Objects.isNull(roleId) || StringUtils.isBlank(platform)){
+            return Lists.newArrayList();
+        }
+        return sysRoleMenuMapper.getRoleMenuByRoleId(roleId, platform);
+    }
+
 }

coolstore-partner-dao/src/main/java/com/cool/store/mapper/SysRoleMenuMapper.java

@@ -3,6 +3,8 @@ package com.cool.store.mapper;
 import com.cool.store.entity.SysRoleMenuDO;
 import org.apache.ibatis.annotations.Param;
 
+import java.util.List;
+
 /**
  * @author zhangchenbiao
  * @date 2023-05-19 03:01
@@ -22,4 +24,12 @@ public interface SysRoleMenuMapper {
      * dateTime:2023-05-19 03:01
      */
     int updateByPrimaryKeySelective(@Param("record") SysRoleMenuDO record);
+
+    /**
+     * 根据角色获取菜单
+     * @param roleId
+     * @param platform
+     * @return
+     */
+    List<SysRoleMenuDO> getRoleMenuByRoleId(@Param("roleId")String roleId, @Param("platform")String platform);
 }

coolstore-partner-dao/src/main/resources/mapper/SysRoleMenuMapper.xml

@@ -4,7 +4,7 @@
     <resultMap id="BaseResultMap" type="com.cool.store.entity.SysRoleMenuDO">
         <id column="id" jdbcType="BIGINT" property="id"/>
         <result column="menu_id" jdbcType="BIGINT" property="menuId"/>
-        <result column="role_id" jdbcType="BIGINT" property="roleId"/>
+        <result column="role_id" jdbcType="VARCHAR" property="roleId"/>
         <result column="platform" jdbcType="VARCHAR" property="platform"/>
     </resultMap>
     <sql id="Base_Column_List">
@@ -50,4 +50,13 @@
         </set>
         where id = #{record.id}
     </update>
+
+    <select id="getRoleMenuByRoleId" resultMap="BaseResultMap">
+        select
+            <include refid="Base_Column_List"/>
+        from
+            sys_role_menu
+        where
+            role_id=#{roleId} and platform=#{platform}
+    </select>
 </mapper>