怎么在Java中实现一个分页遍历功能

怎么在Java中实现一个分页遍历功能？相信很多没有经验的人对此束手无策，为此本文总结了问题出现的原因和解决方法，通过这篇文章希望你能解决这个问题。

1. 数据查询模拟

首先mock一个分页获取数据的逻辑，直接随机生成数据，并且控制最多返回三页

public static int cnt = 0;

private static List<String> randStr(int start, int size) {
  ++cnt;
  if (cnt > 3) {
    return Collections.emptyList();
  } else if (cnt == 3) {
    cnt = 0;
    size -= 2;
  }

  System.out.println("======================= start to gen randList ====================");
  List<String> ans = new ArrayList<>(size);
  for (int i = 0; i < size; i++) {
    ans.add((start + i) + "_" + UUID.randomUUID().toString());
  }
  return ans;
}

2. 基本实现方式

针对这种场景，最常见也是最简单直观的实现方式

• while死循环

• 内部遍历

private static void scanByNormal() {
  int start = 0;
  int size = 5;
  while (true) {
    List<String> list = randStr(start, size);
    for (String str : list) {
      System.out.println(str);
    }

    if (list.size() < size) {
      break;
    }
    start += list.size();
  }
}

3. 迭代器实现方式

接下来介绍一种更有意思的方式，借助迭代器的遍历特性来实现，首先自定义一个通用分页迭代器

public static abstract class MyIterator<T> implements Iterator<T> {
  private int start = 0;
  private int size = 5;

  private int currentIndex;
  private boolean hasMore = true;
  private List<T> list;

  public MyIterator() {
  }

  @Override
  public boolean hasNext() {
    if (list != null && list.size() > currentIndex) {
      return true;
    }

    // 当前的数据已经加载完毕，尝试加载下一批
    if (!hasMore) {
      return false;
    }

    list = load(start, size);
    if (list == null || list.isEmpty()) {
      // 没有加载到数据，结束
      return false;
    }

    if (list.size() < size) {
      // 返回条数小于限制条数，表示还有更多的数据可以加载
      hasMore = false;
    }

    currentIndex = 0;
    start += list.size();
    return true;
  }

  @Override
  public T next() {
    return list.get(currentIndex++);
  }

  public abstract List<T> load(int start, int size);
}

接下来借助上面的迭代器可以比较简单的实现我们的需求了

private static void scanByIterator() {
  MyIterator<String> iterator = new MyIterator<String>() {
    @Override
    public List<String> load(int start, int size) {
      return randStr(start, size);
    }
  };

  while (iterator.hasNext()) {
    String str = iterator.next();
    System.out.println(str);
  }
}

那么问题来了，上面这种使用方式比前面的优势体现再哪儿呢？

双层循环改为单层循环

接下来接入重点了，在jdk1.8引入了函数方法 + lambda之后，又提供了一个更简洁的使用姿势

public class IteratorTestForJdk18 {

  @FunctionalInterface
  public interface LoadFunc<T> {
    List<T> load(int start, int size);
  }

  public static class MyIterator<T> implements Iterator<T> {
    private int start = 0;
    private int size = 5;

    private int currentIndex;
    private boolean hasMore = true;
    private List<T> list;
    private LoadFunc<T> loadFunc;

    public MyIterator(LoadFunc<T> loadFunc) {
      this.loadFunc = loadFunc;
    }

    @Override
    public boolean hasNext() {
      if (list != null && list.size() > currentIndex) {
        return true;
      }

      // 当前的数据已经加载完毕，尝试加载下一批
      if (!hasMore) {
        return false;
      }

      list = loadFunc.load(start, size);
      if (list == null || list.isEmpty()) {
        // 没有加载到数据，结束
        return false;
      }

      if (list.size() < size) {
        // 返回条数小于限制条数，表示还有更多的数据可以加载
        hasMore = false;
      }

      currentIndex = 0;
      start += list.size();
      return true;
    }

    @Override
    public T next() {
      return list.get(currentIndex++);
    }
  }
}

在jdk1.8及之后的使用姿势，一行代码即可

private static void scanByIteratorInJdk8() {
  new MyIterator<>(IteratorTestForJdk18::randStr)
    .forEachRemaining(System.out::println);
}

看完上述内容，你们掌握怎么在Java中实现一个分页遍历功能的方法了吗？如果还想学到更多技能或想了解更多相关内容，欢迎关注亿速云行业资讯频道，感谢各位的阅读！