PageUtil.java 2.81 KB
package com.infoloop.tianting.utils;

import com.infoloop.tianting.model.common.OrderPageResult;
import com.infoloop.tianting.model.common.PageResult;
import com.infoloop.tianting.model.dto.OrderDbDTO.OrderCountByStatusDto;

import java.lang.reflect.InvocationTargetException;
import java.util.LinkedList;
import java.util.List;

public class PageUtil {

    public static <T> PageResult<T> buildEmpty(Class<T> clazz) {
        T instance;
        try {
            instance = clazz.getDeclaredConstructor().newInstance();
        } catch (InstantiationException | IllegalAccessException | InvocationTargetException | NoSuchMethodException e) {
            throw new RuntimeException(e);
        }
        return buildEmpty(instance);
    }

    public static <T> PageResult<T> buildEmpty(T t) {
        PageResult<T> pageRet = new PageResult<>();
        pageRet.setTotalCount(0);
        pageRet.setDataList(List.of(t));
        pageRet.setPageNo(0);
        pageRet.setPageSize(0);
        pageRet.setTotalPage(0);
        return pageRet;
    }

    /**
     * 自定义分页
     */
    public static <T> PageResult<T> subListPage(List<T> list, int pageNo, int pageSize) {
        PageResult<T> pageRet = new PageResult<T>();
        //总条数
        int count = list.size();
        pageRet.setTotalCount(count);
        int pages = count % pageSize == 0 ? count / pageSize : (count / pageSize + 1);
        int fromIndex = (pageNo - 1) * pageSize;
        int toIndex = Math.min(pageNo * pageSize, count);
        if (pageNo > pages) {
            pageRet.setDataList(new LinkedList<>());
            pageRet.setPageNo(pageNo);
            pageRet.setPageSize(pageSize);
            pageRet.setTotalPage(pages);
            return pageRet;
        }
        List<T> pageList = list.subList(fromIndex, toIndex);
        pageRet.setDataList(pageList);
        pageRet.setPageNo(pageNo);
        pageRet.setPageSize(pageSize);
        pageRet.setTotalPage(pages);
        return pageRet;
    }

    public static <T> PageResult<T> from(List<T> list, long totalCount, long pages, int pageNo, int pageSize) {
        PageResult<T> pageRet = new PageResult<T>();
        pageRet.setTotalCount(totalCount);
        pageRet.setDataList(list);
        pageRet.setPageNo(pageNo);
        pageRet.setPageSize(pageSize);
        pageRet.setTotalPage(pages);
        return pageRet;
    }

    public static <T> OrderPageResult<T> from(List<T> list, long totalCount, long pages, int pageNo, int pageSize, OrderCountByStatusDto countByStatus) {
        OrderPageResult<T> pageRet = new OrderPageResult<T>();
        pageRet.setTotalCount(totalCount);
        pageRet.setDataList(list);
        pageRet.setPageNo(pageNo);
        pageRet.setPageSize(pageSize);
        pageRet.setTotalPage(pages);
        pageRet.setCountByStatus(countByStatus);
        return pageRet;
    }
}