com.github.aidensuen.mongo.pagehelper.PageMethod Maven / Gradle / Ivy

Go to download
package com.github.aidensuen.mongo.pagehelper;


import org.springframework.data.domain.Sort;

/**
 * thanks https://github.com/pagehelper/
 */
public abstract class PageMethod {

    protected static final ThreadLocal LOCAL_PAGE = new ThreadLocal<>();
    protected static boolean DEFAULT_COUNT = true;


    protected static void setLocalPage(Page page) {
        LOCAL_PAGE.set(page);
    }

    public static  Page getLocalPage() {
        return LOCAL_PAGE.get();
    }

    public static void clearPage() {
        LOCAL_PAGE.remove();
    }

    public static  Page startPage(PageObject pageObject, Object params) {
        Page page = pageObject.getPageFromObject(params);
        setLocalPage(page);
        return page;
    }


    public static  Page startPage(int pageNum, int pageSize) {
        return startPage(pageNum, pageSize, DEFAULT_COUNT);
    }


    public static  Page startPage(int pageNum, int pageSize, boolean count) {
        return startPage(pageNum, pageSize, count, Sort.unsorted());
    }

    public static  Page startPage(int pageNum, int pageSize, Sort sort) {
        return startPage(pageNum, pageSize, DEFAULT_COUNT, sort);
    }

    public static  Page startPage(int pageNum, int pageSize, Object lastId) {
        return startPage(pageNum, pageSize, true, lastId);
    }

    public static  Page startPage(int pageNum, int pageSize, boolean count, Object lastId) {
        Page page = startPage(pageNum, pageSize, count, Sort.unsorted());
        page.setLastId(lastId);
        return page;
    }

    public static  Page startPage(int pageNum, int pageSize, boolean count, Sort sort) {
        Page page = Page.of(pageNum, pageSize, count, sort);
        setLocalPage(page);
        return page;
    }

    public static  Page offsetPage(int skip, int limit) {
        return offsetPage(skip, limit, DEFAULT_COUNT);
    }

    public static  Page offsetPage(int skip, int limit, boolean count) {
        Page page = Page.of(new int[]{skip, limit}, count);
        setLocalPage(page);
        return page;
    }

}