Mybatis-Plus手动分页

TableDataInfo

java 复制代码
package com.ict.lux.core.page;

import cn.hutool.http.HttpStatus;
import com.baomidou.mybatisplus.core.metadata.IPage;
import lombok.AllArgsConstructor;
import lombok.Builder;
import lombok.Data;
import lombok.NoArgsConstructor;

import java.io.Serializable;
import java.util.List;

/**
 * 表格分页数据对象
 *
 * @author Lion Li
 */

@Data
@Builder
@AllArgsConstructor
@NoArgsConstructor
public class TableDataInfo<T> implements Serializable {
    private static final long serialVersionUID = 1L;

    /**
     * 总记录数
     */
    private long total;

    /**
     * 列表数据
     */
    private List<T> rows;

    /**
     * 消息状态码
     */
    private int code;

    /**
     * 消息内容
     */
    private String msg;

    /**
     * 分页
     *
     * @param list  列表数据
     * @param total 总记录数
     */
    public TableDataInfo(List<T> list, long total) {
        this.rows = list;
        this.total = total;
    }

    public static <T> TableDataInfo<T> build(IPage<T> page) {
        TableDataInfo<T> rspData = new TableDataInfo<>();
        rspData.setCode(HttpStatus.HTTP_OK);
        rspData.setMsg("查询成功");
        rspData.setRows(page.getRecords());
        rspData.setTotal(page.getTotal());
        return rspData;
    }

    public static <T> TableDataInfo<T> build(List<T> list) {
        TableDataInfo<T> rspData = new TableDataInfo<>();
        rspData.setCode(HttpStatus.HTTP_OK);
        rspData.setMsg("查询成功");
        rspData.setRows(list);
        rspData.setTotal(list.size());
        return rspData;
    }

    public static <T> TableDataInfo<T> build(List<T> list, int total) {
        TableDataInfo<T> rspData = new TableDataInfo<>();
        rspData.setCode(HttpStatus.HTTP_OK);
        rspData.setMsg("查询成功");
        rspData.setRows(list);
        rspData.setTotal(total);
        return rspData;
    }

    public static <T> TableDataInfo<T> build() {
        TableDataInfo<T> rspData = new TableDataInfo<>();
        rspData.setCode(HttpStatus.HTTP_OK);
        rspData.setMsg("查询成功");
        return rspData;
    }

}

serviceImpl.selectPageQueryUserList

java 复制代码
 @Override
    public TableDataInfo<SysUser> selectPageQueryUserList(SysUserBo bo, PageQuery pageQuery) {

        if (StrUtil.isNotEmpty(bo.getKey())) {
            //如果key值为数字,则按照工号查
            if (bo.getKey().matches("^[0-9]+$")) {
                bo.setUserName(bo.getKey());
            }

            //如果key值为中文,则按照姓名查
            if (bo.getKey().matches("^[\\u4e00-\\u9fa5]+$")) {
                bo.setNickName(bo.getKey());
            }
        }


        //根据公司编码查询该公司所有用户
        List<SysUser> firmList = baseMapper.selectUsersByDeptCode(bo);

        //如果key值为字母,则按照'姓'的拼音查,大小写均可
        if (StrUtil.isNotEmpty(bo.getKey()) && bo.getKey().matches("^[a-zA-Z]+$")){
            //字母一律转换为小写
            String name = bo.getKey().toLowerCase(Locale.ROOT);
            //存放符合搜索条件的数据
            List<SysUser> charNameList = new ArrayList<>();
            for (SysUser sysUser:firmList){
                //定义surname代表姓氏
                char surname = 0;
                //挑出nickName第一个字符
                surname = sysUser.getNickName().charAt(0);

                //挑出第一个字符的拼音
                List<Pinyin> pinyins = HanLP.convertToPinyinList(String.valueOf(surname));
                for (Pinyin pinyin : pinyins) {
//                    String s = String.valueOf(pinyin.getShengmu());
//                    String s1 = String.valueOf(pinyin.getFirstChar());
                    //获取纯字母的拼音
                    String s = String.valueOf(pinyin.getPinyinWithoutTone());
                    //进行匹配
                    if (s.contains(name)) {
                        charNameList.add(sysUser);
                    }
//                    else {
//                        continue;
//                    }
                }
            }
            firmList = charNameList;
        }


        //如果bo.deptId不为空,则在firmPage(公司内)向下查询所有子部门用户
        if (bo.getDeptId() != null) {
            // 根据deptId查询所有子部门id
            List<Long> firmDeptIdList = baseMapper.selectFirmDeptId(bo.getDeptId());
            //根据子部门id筛选firmPage中符合条件的用户
            firmList = firmList.stream()
                    .filter(user -> firmDeptIdList.contains(user.getDeptId()))
                    .collect(Collectors.toList());

        }

        //如果bo.roleId不为空,则在firmPage(公司内)查询所有与之关联用户
        if (bo.getRoleId() != null) {
            //根据roleId查询关联的用户id
            List<Long> userIdList = baseMapper.selectUserIdByRoleId(bo.getRoleId());
            //根据用户id筛选firmPage中符合条件的用户
            firmList = firmList.stream()
                    .filter(user -> userIdList.contains(user.getUserId()))
                    .collect(Collectors.toList());

        }

        // 创建Page对象
        Page<SysUser> page = new Page<>();
        if (pageQuery.getPageNum() != null && pageQuery.getPageSize() != null) {
            page.setCurrent(pageQuery.getPageNum());
            page.setSize(pageQuery.getPageSize());

            // 手动分页,获取当前页的数据列表
            int startIndex = (int) ((page.getCurrent() - 1) * page.getSize());
            int endIndex = (int) Math.min(startIndex + page.getSize(), firmList.size());
            List<SysUser> currentList = firmList.subList(startIndex, endIndex);

            // 设置总记录数和当前页数据列表
            page.setTotal(firmList.size());
            page.setRecords(currentList);
        } else {
            page.setRecords(firmList);
            page.setTotal(firmList.size());
        }

        return TableDataInfo.build(page);
    }
相关推荐
llwszx3 小时前
深入理解Java锁原理(一):偏向锁的设计原理与性能优化
java·spring··偏向锁
云泽野3 小时前
【Java|集合类】list遍历的6种方式
java·python·list
二进制person4 小时前
Java SE--方法的使用
java·开发语言·算法
OneQ6664 小时前
C++讲解---创建日期类
开发语言·c++·算法
JoJo_Way4 小时前
LeetCode三数之和-js题解
javascript·算法·leetcode
小阳拱白菜5 小时前
java异常学习
java
.30-06Springfield5 小时前
人工智能概念之七:集成学习思想(Bagging、Boosting、Stacking)
人工智能·算法·机器学习·集成学习
FrankYoou6 小时前
Jenkins 与 GitLab CI/CD 的核心对比
java·docker
麦兜*6 小时前
Spring Boot启动优化7板斧(延迟初始化、组件扫描精准打击、JVM参数调优):砍掉70%启动时间的魔鬼实践
java·jvm·spring boot·后端·spring·spring cloud·系统架构
KK溜了溜了7 小时前
JAVA-springboot 整合Redis
java·spring boot·redis