Different Ways to Use Limit and Skip in MongoRepository

1. 引言

在使用 Spring Data MongoDB 时，MongoRepository 接口提供了一种简单的方式来通过内置方法与 MongoDB 集合交互。

在这个快速教程中，我们将学习如何在 MongoRepository 中使用 limit 和 skip 功能。

2. 设置

首先，我们创建一个名为 StudentRepository 的仓库。这里我们将存储有关学生的信息：

public interface StudentRepository extends MongoRepository<Student, String> {}

然后，我们在这个仓库中添加一些学生记录作为示例数据。这些就像练习数据，帮助我们进行操作和学习：

@Before
public void setUp() {
    Student student1 = new Student("A", "Abraham", 15L);
    Student student2 = new Student("B", "Usman", 30L);
    Student student3 = new Student("C", "David", 20L);
    Student student4 = new Student("D", "Tina", 45L);
    Student student5 = new Student("E", "Maria", 33L);

    studentList = Arrays.asList(student1, student2, student3, student4, student5);
    studentRepository.saveAll(studentList);
}

接下来，我们将深入讲解如何使用 skip 和 limit 来获取特定的学生信息。

3. 使用聚合管道

聚合管道是一种强大的数据处理、转换和分析工具。 它通过将多个阶段串联起来工作，每个阶段执行特定的操作，如过滤、分组、排序、分页等。

让我们在一个基本示例中应用 limit 和 skip：

@Test
public void whenRetrievingAllStudents_thenReturnsCorrectNumberOfRecords() {
    // WHEN
    List<Student> result = studentRepository.findAll(0L, 5L);
    // THEN
    assertEquals(5, result.size());
}

上述聚合管道会跳过指定数量的文档，并限制输出到指定数量。

@Test
public void whenLimitingAndSkipping_thenReturnsLimitedStudents() {
    // WHEN
    List<Student> result = studentRepository.findAll(3L, 2L);
    // THEN
    assertEquals(2, result.size());
    assertEquals("Tina", result.get(0).getName());
    assertEquals("Maria", result.get(1).getName());
}

我们甚至可以在复杂的聚合管道中应用 limit 和 skip：

@Aggregation(pipeline = {
  "{ '$match': { 'id' : ?0 } }",
  "{ '$sort' : { 'id' : 1 } }",
  "{ '$skip' : ?1 }",
  "{ '$limit' : ?2 }"
})
List<Student> findByStudentId(final String studentId, Long skip, Long limit);

以下是测试结果：

@Test
public void whenFilteringById_thenReturnsStudentsMatchingCriteria() {
    // WHEN
    List<Student> result = studentRepository.findByStudentId("A", 0L, 5L);
    // THEN
    assertEquals(1, result.size());
    assertEquals("Abraham", result.get(0).getName());
}

4. 使用Pageable

在Spring Data中，Pageable 是一个接口，表示以分页方式检索数据的请求。当从 MongoDB 集合查询数据时，Pageable 对象允许我们指定参数，如页码、页大小和排序标准。这对于在应用中显示大量数据，而一次性显示所有项目速度较慢的情况特别有用。

让我们看看如何定义带有 Pageable 的仓库方法以实现高效的数据检索：

Page<Student> findAll(Pageable pageable);

现在添加一个测试：

@Test
public void whenFindByStudentIdUsingPageable_thenReturnsPageOfStudents() {
    // GIVEN
    Sort sort = Sort.by(Sort.Direction.DESC, "id");
    Pageable pageable = PageRequest.of(0, 5, sort);

    // WHEN
    Page<Student> resultPage = studentRepository.findAll(pageable);

    // THEN
    assertEquals(5, resultPage.getTotalElements());
    assertEquals("Maria", resultPage.getContent().get(0).getName());
}

5. 总结

在这篇短文中，我们探讨了在 MongoRepository 中使用 skip 和 limit 功能的方法。此外，我们强调了 Pageable 对流畅分页的实用性，以及 @Aggregation 对查询逻辑的控制和特定过滤需求的灵活性。

如往常一样，所有代码片段可在 GitHub 上找到。