com.cybermkd.kit.MongoPaginate Maven / Gradle / Ivy
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 <= 0 || count > totalRow || page <= 0 || page > totalPage || (long) page * (long) 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();
}
}