com.cybermkd.kit.MongoPaginate Maven / Gradle / Ivy

Go to download
package com.cybermkd.kit;

import com.alibaba.fastjson.JSONObject;

import java.util.ArrayList;
import java.util.List;

/**
 * 创建人:T-baby
 * 创建日期: 16/5/17
 * 文件描述:用于分页
 */
public class MongoPaginate {

    private int count;
    private int page;
    private int totalPage;
    private long totalRow;
    private MongoQuery query;
    private boolean firstPage = false;
    private boolean lastPage = false;
    private List list = new ArrayList();

    public MongoPaginate(MongoQuery query, int count, int page) {
        this.query = query;
        this.count = count;
        this.page = page;
        this.totalRow = count();
        this.totalPage = (int) (totalRow / (long) count);
        if (totalRow % (long) count != 0L) {
            ++totalPage;
        }

        if (count > totalRow) {
            this.count = (int) totalRow;
        }

        if (this.count <= 0 || this.page <= 0 || this.page > totalPage || (long) this.page * (long) this.count > totalRow) {
            throw new RuntimeException("MongPage tips: (づ￣ 3￣)づ count or page is error !");
        }



        this.firstPage = this.page == 1;
        this.lastPage = this.page == this.totalPage;

    }


    public MongoPage find() {
        this.list = query.limit(count).skip(skip()).find();
        return new MongoPage(count, page, totalPage, totalRow, firstPage, lastPage, list);
    }

    public  MongoPage find(Class clazz) {
        this.list = query.limit(count).skip(skip()).find(clazz);
        return new MongoPage(count, page, totalPage, totalRow, firstPage, lastPage, list);
    }

    public MongoPage findAll() {
        this.list = query.limit(count).skip(skip()).findAll();
        return new MongoPage(count, page, totalPage, totalRow, firstPage, lastPage, list);
    }

    public  MongoPage findAll(Class clazz) {
        this.list = query.limit(count).skip(skip()).findAll(clazz);
        return new MongoPage(count, page, totalPage, totalRow, firstPage, lastPage, list);
    }

    private int skip() {
        if (firstPage) {
            return 0;
        } else {
            return (page - 1) * count;
        }
    }

    public int count() {
        return (int) this.query.count();
    }


}