From e58dd032c89a74bbb8ff4a8d75c435242b7ae47c Mon Sep 17 00:00:00 2001
From: FatttSnake <fatttsnake@gmail.com>
Date: Thu, 1 Jun 2023 18:42:27 +0800
Subject: [PATCH] Fixed pagination in RoleManagement

---
 .../mapper/permission/RoleMapper.java         |  5 +-
 .../permission/impl/RoleServiceImpl.java      | 10 +--
 .../mapper/permission/RoleMapper.xml          | 64 +++++++++++--------
 3 files changed, 45 insertions(+), 34 deletions(-)

diff --git a/Pinnacle/src/main/java/com/cfive/pinnacle/mapper/permission/RoleMapper.java b/Pinnacle/src/main/java/com/cfive/pinnacle/mapper/permission/RoleMapper.java
index 56aa676..266c004 100644
--- a/Pinnacle/src/main/java/com/cfive/pinnacle/mapper/permission/RoleMapper.java
+++ b/Pinnacle/src/main/java/com/cfive/pinnacle/mapper/permission/RoleMapper.java
@@ -1,11 +1,12 @@
 package com.cfive.pinnacle.mapper.permission;
 
-import com.baomidou.mybatisplus.core.metadata.IPage;
 import com.cfive.pinnacle.entity.permission.Role;
 import com.baomidou.mybatisplus.core.mapper.BaseMapper;
 import org.apache.ibatis.annotations.Mapper;
 import org.apache.ibatis.annotations.Param;
 
+import java.util.List;
+
 /**
  * <p>
  * 角色 Mapper 接口
@@ -16,7 +17,7 @@ import org.apache.ibatis.annotations.Param;
  */
 @Mapper
 public interface RoleMapper extends BaseMapper<Role> {
-    IPage<Role> getAll(IPage<Role> roleIPage);
+    List<Role> getAll(@Param("roleList") List<Role> roleList);
 
     Role getOneById(@Param("id") long id);
 }
diff --git a/Pinnacle/src/main/java/com/cfive/pinnacle/service/permission/impl/RoleServiceImpl.java b/Pinnacle/src/main/java/com/cfive/pinnacle/service/permission/impl/RoleServiceImpl.java
index f9aea3f..52cf401 100644
--- a/Pinnacle/src/main/java/com/cfive/pinnacle/service/permission/impl/RoleServiceImpl.java
+++ b/Pinnacle/src/main/java/com/cfive/pinnacle/service/permission/impl/RoleServiceImpl.java
@@ -2,10 +2,9 @@ package com.cfive.pinnacle.service.permission.impl;
 
 import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
 import com.baomidou.mybatisplus.core.metadata.IPage;
+import com.baomidou.mybatisplus.core.toolkit.Wrappers;
 import com.baomidou.mybatisplus.extension.plugins.pagination.PageDTO;
-import com.cfive.pinnacle.entity.permission.Power;
-import com.cfive.pinnacle.entity.permission.Role;
-import com.cfive.pinnacle.entity.permission.PowerRole;
+import com.cfive.pinnacle.entity.permission.*;
 import com.cfive.pinnacle.exception.DataValidationFailedException;
 import com.cfive.pinnacle.mapper.permission.RoleMapper;
 import com.cfive.pinnacle.mapper.permission.PowerRoleMapper;
@@ -46,7 +45,10 @@ public class RoleServiceImpl extends ServiceImpl<RoleMapper, Role> implements IR
     @Override
     public IPage<Role> getAllRole(Long currentPage, Long pageSize) {
         IPage<Role> roleIPage = PageDTO.of(currentPage, pageSize);
-        return roleMapper.getAll(roleIPage);
+        roleIPage = roleMapper.selectPage(roleIPage, Wrappers.emptyWrapper());
+        roleIPage.setRecords(roleMapper.getAll(roleIPage.getRecords()));
+
+        return roleIPage;
     }
 
     @Override
diff --git a/Pinnacle/src/main/resources/mapper/permission/RoleMapper.xml b/Pinnacle/src/main/resources/mapper/permission/RoleMapper.xml
index 37b3044..3c56f09 100644
--- a/Pinnacle/src/main/resources/mapper/permission/RoleMapper.xml
+++ b/Pinnacle/src/main/resources/mapper/permission/RoleMapper.xml
@@ -3,33 +3,38 @@
 <mapper namespace="com.cfive.pinnacle.mapper.permission.RoleMapper">
 
     <select id="getAll" resultMap="roleMap">
-        select t_role.id      as role_id,
-               t_role.name    as role_name,
-               t_role.enable  as role_enable,
-               t_role.deleted as role_deleted,
-               t_role.version as role_version,
-               tm.id          as menu_id,
-               tm.name        as menu_name,
-               tm.url         as menu_url,
-               tm.power_id    as menu_power_id,
-               tm.parent_id   as menu_parent_id,
-               te.id          as element_id,
-               te.name        as element_name,
-               te.power_id    as element_power_id,
-               te.menu_id     as element_menu_id,
-               t.id           as operation_id,
-               t.name         as operation_name,
-               t.code         as operation_code,
-               t.power_id     as operation_power_id,
-               t.element_id   as operation_element_id,
-               t.parent_id    as operation_parent_id
-        from t_role
-                 left join (select * from t_power_role where deleted = 0) as tpr on t_role.id = tpr.role_id
-                 left join t_power tp on tp.id = tpr.power_id
-                 left join t_menu tm on tp.id = tm.power_id
-                 left join t_element te on tp.id = te.power_id
-                 left join t_operation t on tp.id = t.power_id
-        where t_role.deleted = 0
+        select distinct t_role.id as role_id,
+        t_role.name as role_name,
+        t_role.enable as role_enable,
+        t_role.deleted as role_deleted,
+        t_role.version as role_version,
+        tm.id as menu_id,
+        tm.name as menu_name,
+        tm.url as menu_url,
+        tm.power_id as menu_power_id,
+        tm.parent_id as menu_parent_id,
+        te.id as element_id,
+        te.name as element_name,
+        te.power_id as element_power_id,
+        te.menu_id as element_menu_id,
+        t.id as operation_id,
+        t.name as operation_name,
+        t.code as operation_code,
+        t.power_id as operation_power_id,
+        t.element_id as operation_element_id,
+        t.parent_id as operation_parent_id
+        from (select * from t_role where deleted = 0) as t_role
+        left join (select * from t_power_role where deleted = 0) as tpr on t_role.id = tpr.role_id
+        left join t_power tp on tp.id = tpr.power_id
+        left join t_menu tm on tp.id = tm.power_id
+        left join t_element te on tp.id = te.power_id
+        left join t_operation t on tp.id = t.power_id
+        <where>
+            <foreach collection="roleList" item="item" index="index" open="t_role.id in (" separator="," close=")"
+                     nullable="true">
+                #{item.id}
+            </foreach>
+        </where>
     </select>
     <select id="getOneById" resultMap="roleMap">
         select t_role.id      as role_id,
@@ -62,12 +67,15 @@
           and t_role.id = #{id}
     </select>
 
-    <resultMap id="roleMap" type="role">
+    <resultMap id="roleBase" type="role">
         <id property="id" column="role_id"/>
         <result property="name" column="role_name"/>
         <result property="enable" column="role_enable"/>
         <result property="deleted" column="role_deleted"/>
         <result property="version" column="role_version"/>
+    </resultMap>
+
+    <resultMap id="roleMap" type="role" extends="roleBase">
         <collection property="menus" ofType="menu">
             <id property="id" column="menu_id"/>
             <result property="name" column="menu_name"/>