处理 JPA 规范和 spring-data-jpa 时如何使用声明 Stream 作为返回类型

Posted 2023-02-26

技术标签:

【中文标题】处理 JPA 规范和 spring-data-jpa 时如何使用声明 Stream 作为返回类型【英文标题】：How to use declare Stream as return type when dealing with JPA Specification and spring-data-jpa 【发布时间】：2017-05-06 18:36:08 【问题描述】：

我想知道是否可以在自定义查询中使用 JPA 规范谓词？

我试过了，但没有成功。

假设我有一个实体Customer 和一个存储库：

@Repository
public interface CustomerRepository 
    extends JpaRepository<Customer, Long>,
   JpaSpecificationExecutor<Customer>

这样查询是可以的

@Query("select c from Customer c")
Stream<Customer> streamAllCustomers();

这样不行

Stream<Customer> streamAllCustomersWithFilter(Specification<Customer> filter);

有没有办法做到这一点？

注意，我知道我可以将参数放在 @Query 中，但我想留在当前应用程序的设计中并一直使用规范。

【问题讨论】：

没有“JPA 规范”之类的东西；那只是 Spring Data JPA API (!= JPA API)。标签固定您的标题和问题正文中有不同的问题。您希望我们回答哪一个？ 【参考方案1】：

有一种方法可以从我使用的 Spring Data JPA 流式传输数据。这种方法有助于处理大量数据，同时避免高内存消耗，因为整个查询结果不会加载到内存中。

使用以下实现创建custom individual repository

public class YourCustomRepositoryImpl implements YourCustomRepository 

    @PersistenceContext(unitName = "yorEntityManagerFactory")
    private EntityManager em;

    @Override
    public Stream<SomeEntity> streamAll(Specification<SomeEntity> spec) 

            CriteriaBuilder cb = em.getCriteriaBuilder();
            CriteriaQuery<SomeEntity> query = cb.createQuery(SomeEntity.class);
            Root<SomeEntity> root = query.from(SomeEntity.class);
            query.where(spec.toPredicate(root, query, cb));

            return em.createQuery(query).getResultStream();

当然，这需要 JPA 2.2 支持 ORM 框架（我使用的是 Hibernate 5.3）。

此外，您应该注意在处理流时提供连接以保持活动状态。

【讨论】：

【参考方案2】：

TL;DR;

不，不，但手动是

我认为issue DATAJPA-906 回答了你的两个问题

在处理 JPA 规范和 spring-data-jpa 时如何使用 declare Stream 作为返回类型？

你没有，至少没有直接支持的方式：