怎么在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中實現一個分頁遍歷功能的方法了嗎？如果還想學到更多技能或想了解更多相關內容，歡迎關注億速云行業資訊頻道，感謝各位的閱讀！