cn.ocoop.framework.common.util.Paging Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of x-common Show documentation
util case.
The newest version!
package cn.ocoop.framework.common.util;

import com.github.pagehelper.Page;
import lombok.Data;

import java.io.Serializable;
import java.util.Collection;
import java.util.Iterator;
import java.util.List;
import java.util.Spliterator;
import java.util.function.Consumer;

@Data
public class Paging implements Serializable, Iterable {
    private long totalRow;
    private long pageSize = 10;
    private long totalPage;
    private long currentPage = 1;
    private Collection data;

    public Paging() {
    }


    public static  Paging build(int currentPage, int pageSize, long total, List data) {
        Paging paging = new Paging<>();
        paging.setCurrentPage(currentPage);
        paging.setPageSize(pageSize);
        paging.setTotalRow(total);
        paging.setData(data);
        return paging;
    }

    public static  Paging build(int currentPage, int pageSize, List data) {
        Paging paging = new Paging<>();
        paging.setCurrentPage(currentPage);
        paging.setPageSize(pageSize);
        if (data instanceof Page) {
            paging.setTotalRow(((Page) data).getTotal());
        }
        paging.setData(data);
        return paging;
    }

    public static  Paging build(int currentPage, int pageSize, long totalRow) {
        Paging paging = new Paging<>();
        paging.setCurrentPage(currentPage);
        paging.setPageSize(pageSize);
        paging.setTotalRow(totalRow);
        return paging;
    }

    public Paging setTotalRow(long totalRow) {
        this.totalRow = totalRow;
        this.setTotalPage(this.totalRow / getPageSize() + (this.totalRow % getPageSize() != 0 ? 1 : 0));
//        if (getCurrentPage() > getTotalPage()) {
//            setCurrentPage(getTotalPage());
//        } else if (getCurrentPage() < 1) {
//            setCurrentPage(1);
//        }
        return this;
    }

    public long getStart() {
        return (getCurrentPage() - 1) * getPageSize();
    }

    public long getEnd() {
        return getStart() + getPageSize();
    }

    public Paging setData(Collection data) {
        this.data = data;
        return this;
    }


    @Override
    public Iterator iterator() {
        return data.iterator();
    }

    @Override
    public void forEach(Consumer action) {
        data.forEach(action);
    }

    @Override
    public Spliterator spliterator() {
        return data.spliterator();
    }

}