如何以页面的形式获取JPA自定义查询的结果

Question

我有一个存储库，它返回一个页面

public interface MindRepository extends PagingAndSortingRepository<Mind, Integer> {
    Page<Mind> findByCountry(String country, Pageable pageable);
}

以及使用它的控制器：

private MindRepository mindRepository;

@GetMapping(path = "/minds", produces = "application/json")
public Page<Mind> getMinds(String country, Integer page, Integer size) {
    Pageable pageable = PageRequest.of(page,size);
    return mindRepository.findByCountry(country,pageable);        
}

一切都很好。控制器返回页面

但是现在我必须使查询更加复杂，有几个过滤器，动态变化。我想像这样使用createQuery：

public interface CustomizedMindRepository<T> {
    Page<T> findByCountry(String country, Pageable pageable);
}

public interface MindRepository extends PagingAndSortingRepository<Mind, Integer>,CustomizedMindRepository {
    Page<Mind> findByCountry(String country, Pageable pageable);
}

public class CustomizedMindRepositoryImpl implements CustomizedMindRepository {
    @PersistenceContext
    private EntityManager em;

    @Override
    public Page<Mind> findByCountry(String country, Pageable pageable) {
        return em.createQuery("from minds where <dynamical filter> AND <another dynamical filter> AND <...etc>", Mind.class)
                .getResultList();
    }
}

但是getResultList（）返回列表，而不是页面：(

解决这个问题的最好方法是什么？




Answer 1

如果要使用EntityManager。createQuery，您可以使用setFirstResult和setMaxResults方法来获得相同的结果。

@Override
public List<Mind> findByCountry(String country, Pageable pageable) {
  return em.createQuery("from minds where <dynamical filter> AND <another dynamical filter> AND <...etc>", Mind.class)
           .setFirstResult(startPosition)
           .setMaxResults(size)
           .getResultList();
}

在本例中，大小的含义与您的情况相同，但起始位置不是一个页面，而是计算为：

startPosition = page * size

但是，如果您需要建立DyMic查询-考虑使用规范或JPA标准API。

Answer 2

• 页面存储库调用执行两个调用，一个用于获取结果，另一个用于获取总大小。因此，您必须通过执行计数查询来复制这种行为

    @Override
    public Page<Mind> findByCountry(String country, Pageable pageable) {
        long offset = pageable.getPageNumber() * pageable.getPageSize();
        long limit = pageable.getPageSize();
        
        List<Item> itemsInNextPage = em.createQuery(query)
                .setFirstResult(offset)
                .setMaxResults(limit)
                .getResultList();
        
        long total = // Another query to get the total count
                
        List<Mind>  results = em.createQuery("from minds ...", Mind.class)
                             .getResultList();
        return new PageImpl(results, pageable, total); 
                            
    }