Java内存分页工具类
前言
工作过程中，经常会遇到基于内存数据进行分页处理的情况，例如批量更新数据库时，集合过大需要分批更新的情况，还有例如对缓存中的集合数据进行分页获取这种情况。
本文提供了通用的内存分页工具，参考了网络上的一些代码，主要基于 subList() 方法实现，希望对你有所帮助！工具类源码在本文底部。
优化前
首先来看一下正常如果要实现这些需求的话，代码的实现是怎么样的
// 分页进行批量更新
private void batchUpdateStudent(List<Student> students) {
int limit = 100;
int size = students.size();
int m = size / limit;
int n = size % limit;
for (int i = 1; i <= m; i++) {
List<Student> list = students.subList((i - 1) * limit, i * limit);
studentDao.batchUpdate(list);
}
if (n != 0) {
List<Student> list = students.subList(m * limit, students.size());
studentDao.batchUpdate(list);
}
}
// 分页获取数据
public List<Student> pageStudents(Integer page, Integer pageSize) {
if (page < 1) {
page = 1;
}
int start = (page - 1) * pageSize;
int limit = page * pageSize;
// 从缓存中获取全量数据
List<Student> students = studentCache.getStudents();
if (CollectionUtils.isEmpty(students)) {
return new ArrayList<>();
}
if (limit > students.size()) {
limit = students.size();
}
return students.subList(start, limit), students.size();
}
可以看出方法的代码比较冗余，如果多处需要内存分页，重复代码不可避免会有很多！
优化后
// 分页进行批量更新
private void batchUpdateStudent(List<Student> students) {
RAMPager<Student> pager = new RAMPager<>(students, 100);
// 方式一：使用迭代器
Iterator<List<Student>> iterator = pager.iterator();
while (iterator.hasNext()) {
studentDao.batchUpdate(iterator.next());
}
// 方式二：使用索引
//for (int i = 1; i <= pager.getPageCount(); i++) {
// studentDao.batchUpdate(pager.page(i));
//}
}
// 分页获取数据
public List<Student> pageStudents(Integer page, Integer pageSize) {
// 从缓存中获取全量数据
List<Student> students = studentCache.getStudents();
RAMPager<Student> pager = new RAMPager<>(students, pageSize);
return pager.page(page);
}
注：如果只是分页，而不需要关注页码，使用迭代器即可；
工具类源码
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Iterator;
import java.util.List;
/**
* 类名称：RAMPager
* 类描述：内存分页工具
* 创建人：WeJan
* 创建时间：2019年07月22日 13:43
* Version: 1.1
*/
public class RAMPager<T> {
private List<T> data;
private int pageSize;
/**
* @param data 原始数据
* @param pageSize 每页条数
*/
public RAMPager(List<T> data, int pageSize) {
this.data = data;
this.pageSize = pageSize;
}
/**
* 获取某页数据，从第1页开始
*
* @param pageNum 第几页
* @return 分页数据
*/
public List<T> page(int pageNum) {
if (pageNum < 1) {
pageNum = 1;
}
int from = (pageNum - 1) * pageSize;
int to = Math.min(pageNum * pageSize, data.size());
if (from > to) {
from = to;
}
return data.subList(from, to);
}
/**
* 获取总页数
*/
public int getPageCount() {
if (pageSize == 0) {
return 0;
}
return data.size() % pageSize == 0 ? (data.size() / pageSize) : (data.size() / pageSize + 1);
}
/**
* 元素迭代器
*/
public Iterator<List<T>> iterator() {
return new Itr();
}
private class Itr implements Iterator<List<T>> {
int page = 1;
Itr() {
}
public boolean hasNext() {
return page <= getPageCount();
}
public List<T> next() {
int i = page;
if (i > getPageCount())
return new ArrayList<>();
page = i + 1;
return RAMPager.this.page(i);
}
}
public static void main(String[] args) {
List<Integer> list = Arrays.asList(1, 2, 3, 4, 5, 6, 7, 8);
System.out.println("原始数据是：" + list);
int pageSize = 2;
System.out.println("每页大小是：" + pageSize);
RAMPager<Integer> pager = new RAMPager<>(list, pageSize);
System.out.println("总页数是: " + pager.getPageCount());
System.out.println("<- - - - - - - - - - - - - ->");
// 无需感知页码情况下使用
Iterator<List<Integer>> iterator = pager.iterator();
while (iterator.hasNext()) {
List<Integer> next = iterator.next();
System.out.println("next: " + next);
}
System.out.println("<- - - - - - - - - - - - - ->");
// 需要指定页码情况使用，页码从第一页开始，且小于等于总页数！
for (int i = 1; i <= pager.getPageCount(); i++) {
List<Integer> page = pager.page(i);
System.out.println("第 " + i + " 页数据是:" + page);
}
}
}
结语
希望对你有用~
