怎么用 Spring Data 在 RESTful API 中实现更好的分页

493 阅读4分钟

介绍

本文将重点介绍如何使用 Spring MVC 和 Spring Data 在 RESTful API 中实现分页。

REST 分页的可发现性

在分页范围内,满足 REST 的 HATEOAS 约束,意味着使 API 的客户端能够基于导航中的当前页面发现下一页和上一页。 为此,我们将使用Link HTTP 响应头,以及 “next”,“prev”,“first” 和 “last” 链接关系类型。
添加一个侦听器,监听器将检查导航是否允许下一页,上一页,第一页和最后一页。它将相关的 URI 作为 “链接” 添加到 HTTP 响应头中

void addLinkHeaderOnPagedResourceRetrieval(
  UriComponentsBuilder uriBuilder, HttpServletResponse response,
  Class clazz, int page, int totalPages, int size ){
 
    String resourceName = clazz.getSimpleName().toString().toLowerCase();
    uriBuilder.path( "/admin/" + resourceName );
 
    // ...
    
}

接下来,我们将使用 StringJoiner 连接每个链接。我们将使用 uriBuilder 生成 uri。让我们看看我们如何继续链接到下一页:

StringJoiner linkHeader = new StringJoiner(", ");
if (hasNextPage(page, totalPages)){
    String uriForNextPage = constructNextPageUri(uriBuilder, page, size);
    linkHeader.add(createLinkHeader(uriForNextPage, "next"));
}

让我们来看看 constructNextPageUri 方法的逻辑:

String constructNextPageUri(UriComponentsBuilder uriBuilder, int page, int size) {
    return uriBuilder.replaceQueryParam(PAGE, page + 1)
      .replaceQueryParam("size", size)
      .build()
      .encode()
      .toUriString();
}

我们将对希望包含的其余 uri 进行类似的处理。
最后,我们将输出添加为响应头:

response.addHeader("Link", linkHeader.toString());

测试分页

代码如下:

@Test
public void whenResourcesAreRetrievedPaged_then200IsReceived(){
    Response response = RestAssured.get(paths.getFooURL() + "?page=0&size=2");
 
    assertThat(response.getStatusCode(), is(200));
}
@Test
public void whenPageOfResourcesAreRetrievedOutOfBounds_then404IsReceived(){
    String url = getFooURL() + "?page=" + randomNumeric(5) + "&size=2";
    Response response = RestAssured.get.get(url);
 
    assertThat(response.getStatusCode(), is(404));
}
@Test
public void givenResourcesExist_whenFirstPageIsRetrieved_thenPageContainsResources(){
   createResource();
   Response response = RestAssured.get(paths.getFooURL() + "?page=0&size=2");
 
   assertFalse(response.body().as(List.class).isEmpty());
}

测试分页的可发现性

测试的重点是当前页面在导航中的位置,以及应该从每个位置发现的不同 uri:

@Test
public void whenFirstPageOfResourcesAreRetrieved_thenSecondPageIsNext(){
   Response response = RestAssured.get(getFooURL()+"?page=0&size=2");
 
   String uriToNextPage = extractURIByRel(response.getHeader("Link"), "next");
   assertEquals(getFooURL()+"?page=1&size=2", uriToNextPage);
}
@Test
public void whenFirstPageOfResourcesAreRetrieved_thenNoPreviousPage(){
   Response response = RestAssured.get(getFooURL()+"?page=0&size=2");
 
   String uriToPrevPage = extractURIByRel(response.getHeader("Link"), "prev");
   assertNull(uriToPrevPage );
}
@Test
public void whenSecondPageOfResourcesAreRetrieved_thenFirstPageIsPrevious(){
   Response response = RestAssured.get(getFooURL()+"?page=1&size=2");
 
   String uriToPrevPage = extractURIByRel(response.getHeader("Link"), "prev");
   assertEquals(getFooURL()+"?page=0&size=2", uriToPrevPage);
}
@Test
public void whenLastPageOfResourcesIsRetrieved_thenNoNextPageIsDiscoverable(){
   Response first = RestAssured.get(getFooURL()+"?page=0&size=2");
   String uriToLastPage = extractURIByRel(first.getHeader("Link"), "last");
 
   Response response = RestAssured.get(uriToLastPage);
 
   String uriToNextPage = extractURIByRel(response.getHeader("Link"), "next");
   assertNull(uriToNextPage);
}

使用 Spring Data 实现 REST 分页

在 Spring Data 中,如果我们需要从完整的结果集中返回一些结果,则可以使用任何 Pageable 存储库方法,因为它将始终返回 Page。 将根据页码,页面大小和排序方向返回结果。
Spring Data REST 自动识别 URL 参数,例如页码,页面大小,排序等。
要使用任何存储库的分页方法,我们需要扩展 PagingAndSortingRepository:

public interface SubjectRepository extends PagingAndSortingRepository<Subject, Long>{}

如果我们调用 localhost:8080/subjects Spring 会自动添加页码,页面大小,排序参数:

"_links" : {
  "self" : {
    "href" : "http://localhost:8080/subjects{?page,size,sort}",
    "templated" : true
  }
}

默认情况下,页面大小是20,但是我们可以通过调用类似于 localhost:8080/subject?page=10 这样的东西来更改它。
如果我们想实现分页到我们自己的自定义库 API,我们需要传递一个额外的可分页参数,并确保 API 返回一个 Page:

@RestResource(path = "nameContains")
public Page<Subject> findByNameContaining(@Param("name") String name, Pageable p);

每当我们添加自定义API时,就会将 /search 端点添加到生成的链接中。因此,如果我们调用 localhost:8080/subjects/search,我们将看到一个分页功能的端点:

"findByNameContaining" : {
  "href" : "http://localhost:8080/subjects/search/nameContains{?name,page,size,sort}",
  "templated" : true
}

所有实现 PagingAndSortingRepository 的 api 都将返回一个页面。如果我们需要返回来自页面的结果列表,页面的 getContent() API 提供了作为 Spring Data REST API 的结果而获取的记录列表。

将 List 转换为 Page

假设我们有一个可分页的对象作为输入,但是我们需要检索的信息包含在一个 List 中,而不是一个 PagingAndSortingRepository。在这些情况下,我们可能需要将 List 转换为 Page。
例如,假设我们有一个 SOAP 服务的结果列表:

List<Foo> list = getListOfFooFromSoapService();

我们需要在发送给我们的 Pageable 对象指定的特定位置访问列表。那么,让我们定义开始索引:

int start = (int) pageable.getOffset();

结束索引:

int end = (int) ((start + pageable.getPageSize()) > fooList.size() ? fooList.size()
  : (start + pageable.getPageSize()));

有了这两个地方,我们可以创建一个 Page 来获取它们之间的元素列表:

Page<Foo> page = new PageImpl<Foo>(fooList.subList(start, end), pageable, fooList.size());

这样我们就可以返回 Page 作为一个有效的结果。
注意,如果我们还希望支持排序,我们需要在将 List 的子列表之前对其进行排序。

总结

本文演示了如何使用 Spring 在 REST API 中实现分页,并讨论了如何设置和测试可发现性。

欢迎关注我的公众号:曲翎风,获得独家整理的学习资源和日常干货推送。
如果您对我的专题内容感兴趣,也可以关注我的博客:sagowiec.com