Spring 数据 JPA。如何仅从 findAll() 方法获取 ID 列表
Spring Data JPA. How to get only a list of IDs from findAll() method
我有一个非常复杂的模型。实体有很多关系等等。
我尝试使用 Spring Data JPA 并准备了一个存储库。
但是当我调用带有对象规范的方法 findAll() 时,会出现性能问题,因为对象非常大。我知道,因为当我调用这样的方法时:
@Query(value = "select id, name from Customer ")
List<Object[]> myFindCustomerIds();
我没有遇到任何性能问题。
但是当我调用
List<Customer> findAll();
我在性能方面遇到了很大的问题。
问题是我需要使用客户规范调用 findAll 方法,这就是为什么我不能使用 returns 对象数组列表的方法。
如何编写一种方法来查找具有客户实体规格但 returns 只有一个 ID 的所有客户。
像这样:
List<Long> findAll(Specification<Customer> spec);
- 我不能在这种情况下使用分页。
请帮忙。
我解决了问题。
(结果我们将有一个只有 ID 和名称的稀疏 Customer 对象)
定义自己的存储库:
public interface SparseCustomerRepository {
List<Customer> findAllWithNameOnly(Specification<Customer> spec);
}
和一个实现(记住后缀 - Impl 作为默认值)
@Service
public class SparseCustomerRepositoryImpl implements SparseCustomerRepository {
private final EntityManager entityManager;
@Autowired
public SparseCustomerRepositoryImpl(EntityManager entityManager) {
this.entityManager = entityManager;
}
@Override
public List<Customer> findAllWithNameOnly(Specification<Customer> spec) {
CriteriaBuilder criteriaBuilder = entityManager.getCriteriaBuilder();
CriteriaQuery<Tuple> tupleQuery = criteriaBuilder.createTupleQuery();
Root<Customer> root = tupleQuery.from(Customer.class);
tupleQuery.multiselect(getSelection(root, Customer_.id),
getSelection(root, Customer_.name));
if (spec != null) {
tupleQuery.where(spec.toPredicate(root, tupleQuery, criteriaBuilder));
}
List<Tuple> CustomerNames = entityManager.createQuery(tupleQuery).getResultList();
return createEntitiesFromTuples(CustomerNames);
}
private Selection<?> getSelection(Root<Customer> root,
SingularAttribute<Customer, ?> attribute) {
return root.get(attribute).alias(attribute.getName());
}
private List<Customer> createEntitiesFromTuples(List<Tuple> CustomerNames) {
List<Customer> customers = new ArrayList<>();
for (Tuple customer : CustomerNames) {
Customer c = new Customer();
c.setId(customer.get(Customer_.id.getName(), Long.class));
c.setName(customer.get(Customer_.name.getName(), String.class));
c.add(customer);
}
return customers;
}
}
Spring 数据现在支持 Projections:
interface SparseCustomer {
String getId();
String getName();
}
比你的 Customer
存储库
List<SparseCustomer> findAll(Specification<Customer> spec);
编辑:
正如 Radouane 所指出的,由于 bug.
,具有规范的 ROUFID 投影目前不起作用
但是您可以使用 specification-with-projection 库来解决这个 Spring Data Jpa 缺陷。
为什么不使用 @Query
注释?
@Query("select p.id from #{#entityName} p")
List<Long> getAllIds();
我看到的唯一缺点是当属性 id
更改时,但由于这是一个非常常见的名称并且不太可能更改(id = 主键),所以这应该没问题。
不幸的是Projections does not work with specifications。 JpaSpecificationExecutor
return 仅使用存储库管理的聚合根键入的列表 ( List<T> findAll(Specification<T> var1);
)
实际的解决方法是使用 Tuple。示例:
@Override
public <D> D findOne(Projections<DOMAIN> projections, Specification<DOMAIN> specification, SingleTupleMapper<D> tupleMapper) {
Tuple tuple = this.getTupleQuery(projections, specification).getSingleResult();
return tupleMapper.map(tuple);
}
@Override
public <D extends Dto<ID>> List<D> findAll(Projections<DOMAIN> projections, Specification<DOMAIN> specification, TupleMapper<D> tupleMapper) {
List<Tuple> tupleList = this.getTupleQuery(projections, specification).getResultList();
return tupleMapper.map(tupleList);
}
private TypedQuery<Tuple> getTupleQuery(Projections<DOMAIN> projections, Specification<DOMAIN> specification) {
CriteriaBuilder cb = entityManager.getCriteriaBuilder();
CriteriaQuery<Tuple> query = cb.createTupleQuery();
Root<DOMAIN> root = query.from((Class<DOMAIN>) domainClass);
query.multiselect(projections.project(root));
query.where(specification.toPredicate(root, query, cb));
return entityManager.createQuery(query);
}
其中 Projections
是根投影的功能接口。
@FunctionalInterface
public interface Projections<D> {
List<Selection<?>> project(Root<D> root);
}
SingleTupleMapper
和 TupleMapper
用于将 TupleQuery
结果映射到您想要 return.
的对象
@FunctionalInterface
public interface SingleTupleMapper<D> {
D map(Tuple tuple);
}
@FunctionalInterface
public interface TupleMapper<D> {
List<D> map(List<Tuple> tuples);
}
使用示例:
Projections<User> userProjections = (root) -> Arrays.asList(
root.get(User_.uid).alias(User_.uid.getName()),
root.get(User_.active).alias(User_.active.getName()),
root.get(User_.userProvider).alias(User_.userProvider.getName()),
root.join(User_.profile).get(Profile_.firstName).alias(Profile_.firstName.getName()),
root.join(User_.profile).get(Profile_.lastName).alias(Profile_.lastName.getName()),
root.join(User_.profile).get(Profile_.picture).alias(Profile_.picture.getName()),
root.join(User_.profile).get(Profile_.gender).alias(Profile_.gender.getName())
);
Specification<User> userSpecification = UserSpecifications.withUid(userUid);
SingleTupleMapper<BasicUserDto> singleMapper = tuple -> {
BasicUserDto basicUserDto = new BasicUserDto();
basicUserDto.setUid(tuple.get(User_.uid.getName(), String.class));
basicUserDto.setActive(tuple.get(User_.active.getName(), Boolean.class));
basicUserDto.setUserProvider(tuple.get(User_.userProvider.getName(), UserProvider.class));
basicUserDto.setFirstName(tuple.get(Profile_.firstName.getName(), String.class));
basicUserDto.setLastName(tuple.get(Profile_.lastName.getName(), String.class));
basicUserDto.setPicture(tuple.get(Profile_.picture.getName(), String.class));
basicUserDto.setGender(tuple.get(Profile_.gender.getName(), Gender.class));
return basicUserDto;
};
BasicUserDto basicUser = findOne(userProjections, userSpecification, singleMapper);
希望对您有所帮助。
我有一个非常复杂的模型。实体有很多关系等等。
我尝试使用 Spring Data JPA 并准备了一个存储库。
但是当我调用带有对象规范的方法 findAll() 时,会出现性能问题,因为对象非常大。我知道,因为当我调用这样的方法时:
@Query(value = "select id, name from Customer ")
List<Object[]> myFindCustomerIds();
我没有遇到任何性能问题。
但是当我调用
List<Customer> findAll();
我在性能方面遇到了很大的问题。
问题是我需要使用客户规范调用 findAll 方法,这就是为什么我不能使用 returns 对象数组列表的方法。
如何编写一种方法来查找具有客户实体规格但 returns 只有一个 ID 的所有客户。
像这样:
List<Long> findAll(Specification<Customer> spec);
- 我不能在这种情况下使用分页。
请帮忙。
我解决了问题。
(结果我们将有一个只有 ID 和名称的稀疏 Customer 对象)
定义自己的存储库:
public interface SparseCustomerRepository {
List<Customer> findAllWithNameOnly(Specification<Customer> spec);
}
和一个实现(记住后缀 - Impl 作为默认值)
@Service
public class SparseCustomerRepositoryImpl implements SparseCustomerRepository {
private final EntityManager entityManager;
@Autowired
public SparseCustomerRepositoryImpl(EntityManager entityManager) {
this.entityManager = entityManager;
}
@Override
public List<Customer> findAllWithNameOnly(Specification<Customer> spec) {
CriteriaBuilder criteriaBuilder = entityManager.getCriteriaBuilder();
CriteriaQuery<Tuple> tupleQuery = criteriaBuilder.createTupleQuery();
Root<Customer> root = tupleQuery.from(Customer.class);
tupleQuery.multiselect(getSelection(root, Customer_.id),
getSelection(root, Customer_.name));
if (spec != null) {
tupleQuery.where(spec.toPredicate(root, tupleQuery, criteriaBuilder));
}
List<Tuple> CustomerNames = entityManager.createQuery(tupleQuery).getResultList();
return createEntitiesFromTuples(CustomerNames);
}
private Selection<?> getSelection(Root<Customer> root,
SingularAttribute<Customer, ?> attribute) {
return root.get(attribute).alias(attribute.getName());
}
private List<Customer> createEntitiesFromTuples(List<Tuple> CustomerNames) {
List<Customer> customers = new ArrayList<>();
for (Tuple customer : CustomerNames) {
Customer c = new Customer();
c.setId(customer.get(Customer_.id.getName(), Long.class));
c.setName(customer.get(Customer_.name.getName(), String.class));
c.add(customer);
}
return customers;
}
}
Spring 数据现在支持 Projections:
interface SparseCustomer {
String getId();
String getName();
}
比你的 Customer
存储库
List<SparseCustomer> findAll(Specification<Customer> spec);
编辑:
正如 Radouane 所指出的,由于 bug.
但是您可以使用 specification-with-projection 库来解决这个 Spring Data Jpa 缺陷。
为什么不使用 @Query
注释?
@Query("select p.id from #{#entityName} p")
List<Long> getAllIds();
我看到的唯一缺点是当属性 id
更改时,但由于这是一个非常常见的名称并且不太可能更改(id = 主键),所以这应该没问题。
不幸的是Projections does not work with specifications。 JpaSpecificationExecutor
return 仅使用存储库管理的聚合根键入的列表 ( List<T> findAll(Specification<T> var1);
)
实际的解决方法是使用 Tuple。示例:
@Override
public <D> D findOne(Projections<DOMAIN> projections, Specification<DOMAIN> specification, SingleTupleMapper<D> tupleMapper) {
Tuple tuple = this.getTupleQuery(projections, specification).getSingleResult();
return tupleMapper.map(tuple);
}
@Override
public <D extends Dto<ID>> List<D> findAll(Projections<DOMAIN> projections, Specification<DOMAIN> specification, TupleMapper<D> tupleMapper) {
List<Tuple> tupleList = this.getTupleQuery(projections, specification).getResultList();
return tupleMapper.map(tupleList);
}
private TypedQuery<Tuple> getTupleQuery(Projections<DOMAIN> projections, Specification<DOMAIN> specification) {
CriteriaBuilder cb = entityManager.getCriteriaBuilder();
CriteriaQuery<Tuple> query = cb.createTupleQuery();
Root<DOMAIN> root = query.from((Class<DOMAIN>) domainClass);
query.multiselect(projections.project(root));
query.where(specification.toPredicate(root, query, cb));
return entityManager.createQuery(query);
}
其中 Projections
是根投影的功能接口。
@FunctionalInterface
public interface Projections<D> {
List<Selection<?>> project(Root<D> root);
}
SingleTupleMapper
和 TupleMapper
用于将 TupleQuery
结果映射到您想要 return.
@FunctionalInterface
public interface SingleTupleMapper<D> {
D map(Tuple tuple);
}
@FunctionalInterface
public interface TupleMapper<D> {
List<D> map(List<Tuple> tuples);
}
使用示例:
Projections<User> userProjections = (root) -> Arrays.asList(
root.get(User_.uid).alias(User_.uid.getName()),
root.get(User_.active).alias(User_.active.getName()),
root.get(User_.userProvider).alias(User_.userProvider.getName()),
root.join(User_.profile).get(Profile_.firstName).alias(Profile_.firstName.getName()),
root.join(User_.profile).get(Profile_.lastName).alias(Profile_.lastName.getName()),
root.join(User_.profile).get(Profile_.picture).alias(Profile_.picture.getName()),
root.join(User_.profile).get(Profile_.gender).alias(Profile_.gender.getName())
);
Specification<User> userSpecification = UserSpecifications.withUid(userUid);
SingleTupleMapper<BasicUserDto> singleMapper = tuple -> {
BasicUserDto basicUserDto = new BasicUserDto();
basicUserDto.setUid(tuple.get(User_.uid.getName(), String.class));
basicUserDto.setActive(tuple.get(User_.active.getName(), Boolean.class));
basicUserDto.setUserProvider(tuple.get(User_.userProvider.getName(), UserProvider.class));
basicUserDto.setFirstName(tuple.get(Profile_.firstName.getName(), String.class));
basicUserDto.setLastName(tuple.get(Profile_.lastName.getName(), String.class));
basicUserDto.setPicture(tuple.get(Profile_.picture.getName(), String.class));
basicUserDto.setGender(tuple.get(Profile_.gender.getName(), Gender.class));
return basicUserDto;
};
BasicUserDto basicUser = findOne(userProjections, userSpecification, singleMapper);
希望对您有所帮助。