通过Spring Data

时间:2018-09-26 13:26:58

标签: java spring-boot cassandra spring-data spring-data-cassandra

我正在寻找一种解决方案,以使用Cassandra(版本3.11.3)数据库为基于Spring Boot的REST服务实现分页。我们将Spring Boot 2.0.5.RELEASEspring-boot-starter-data-cassandra用作依赖项。

由于Spring Data的CassandraRepository<T, ID>接口没有扩展PagingAndSortingRepository,因此我们没有像JPA那样获得完整的分页功能。

我阅读了Spring Data Cassandra文档,并找到了一种可能的方法来实现Cassandra和Spring Data的分页,因为CassandraRepository接口具有可用的Slice<T> findAll(Pageable pageable);以下方法。我知道Cassandra无法获得特定的页面即席显示,并且始终需要页面0来遍历所有页面,如CassandraPageRequest中所述:

  

特定于Cassandra的{@link PageRequest}实现,提供对{@link PagingState}的访问。该类允许创建第一个页面请求,并通过Cassandra分页表示,它基于获取的页面的进度,并允许仅向前导航。 访问特定页面需要提取所有页面,直到到达所需页面为止。

在我的用例中,我们有> 1.000.000数据库条目,并希望在我们的单页应用程序中分页显示它们。

我当前的方法如下:

@RestController
@RequestMapping("/users")
public class UsersResource {

  @Autowired
  UserRepository    userRepository;

  @GetMapping
  public ResponseEntity<List<User>> getAllTests(
            @RequestParam(defaultValue = "0", name = "page") @Positive int requiredPage, 
            @RequestParam(defaultValue = "500", name = "size") int size) {

    Slice<User> resultList = userRepository.findAll(CassandraPageRequest.first(size));

    int currentPage = 0;

    while (resultList.hasNext() && currentPage <= requiredPage) {
      System.out.println("Current Page Number: " + currentPage);
      resultList = userRepository.findAll(resultList.nextPageable());
      currentPage++;
    }

    return ResponseEntity.ok(resultList.getContent());
  }
}

使用这种方法,我必须在将所有数据库条目提取到内存并进行迭代直到找到正确的页面的同时找到请求的页面。是否有其他方法可以找到正确的页面,还是必须使用当前的解决方案?

我的Cassandra表定义如下:

CREATE TABLE user (
  id int, firstname varchar, 
  lastname varchar, 
  code varchar, 
  PRIMARY KEY(id)
);

3 个答案:

答案 0 :(得分:1)

我要做的是创建一个包含内容和pageingState哈希的页面对象。

在初始页面中,我们进行了简单的分页

Pageable pageRequest = CassandraPageRequest.of(0,5);

执行查找后,我们将获得切片

Slice<Group> slice = groupRepository.findAll(pageRequest);

通过切片可以获取分页状态

page.setPageHash(getPageHash((CassandraPageRequest) slice.getPageable()));

其中

private String getPageHash(CassandraPageRequest pageRequest) { return Base64.toBase64String(pageRequest.getPagingState().toBytes()); }

最终返回具有List内容和pageingState为pageHash的Page对象

答案 1 :(得分:1)

请参见下面的代码。可能会有帮助。

    @GetMapping("/loadData")
    public Mono<DataTable> loadData(@RequestParam boolean reset, @RequestParam(required = false) String tag, WebSession session) {
        final String sessionId = session.getId();
        IMap<String, String> map = Context.get(HazelcastInstance.class).getMap("companygrouping-pageable-map");
        int pageSize = Context.get(EnvProperties.class).getPageSize();
        Pageable pageRequest;

        if (reset)
            map.remove(sessionId);

        String serializedPagingState = map.compute(sessionId, (k, v) -> (v == null) ? null : map.get(session.getId()));

        pageRequest = StringUtils.isBlank(serializedPagingState) ? CassandraPageRequest.of(0, pageSize)
                : CassandraPageRequest.of(PageRequest.of(0, pageSize), PagingState.fromString(serializedPagingState)).next();

        Mono<Slice<TagMerge>> sliceMono = StringUtils.isNotBlank(tag)
                ? Context.get(TagMergeRepository.class).findByKeyStatusAndKeyTag(Status.NEW, tag, pageRequest)
                : Context.get(TagMergeRepository.class).findByKeyStatus(Status.NEW, pageRequest);

        Flux<TagMerge> flux = sliceMono.map(t -> convert(t, map, sessionId)).flatMapMany(Flux::fromIterable);
        Mono<DataTable> dataTabelMono = createTableFrom(flux).doOnError(e -> log.error("{}", e));
        if (reset) {
            Mono<Long> countMono = Mono.empty();
            if (StringUtils.isNotBlank(tag))
                countMono = Context.get(TagMergeRepository.class).countByKeyStatusAndKeyTag(Status.NEW, tag);
            else
                countMono = Context.get(TagMergeRepository.class).countByKeyStatus(Status.NEW);
            dataTabelMono = dataTabelMono.zipWith(countMono, (t, k) -> {
                t.setTotalRows(k);
                return t;
            });
        }
        return dataTabelMono;
    }

private List<TagMerge> convert(Slice<TagMerge> slice, IMap<String, String> map, String id) {
        PagingState pagingState = ((CassandraPageRequest) slice.getPageable()).getPagingState();
        if (pagingState != null)
            map.put(id, pagingState.toString());
        return slice.getContent();
    }


答案 2 :(得分:0)

Cassandra支持前向分页,这意味着您可以获取前n行,然后可以获取n + 1和2n之间的行,依此类推,直到数据结束,但是您不能直接获取n + 1和2n之间的行。 >