Spring Данные JPA. Как получить только список идентификаторов метода findAll()

У меня очень сложная модель. У сущности много отношений и так далее.

Я пытаюсь использовать Spring Data JPA, и я подготовил хранилище.

но когда я вызываю метод findAll() со спецификацией для объекта, возникает проблема производительности, потому что объекты очень большие. Я знаю это, потому что когда я вызываю такой метод:

@Query(value = "select id, name from Customer ")
List<Object[]> myFindCustomerIds();

У меня не было проблем с производительностью.

Но когда я призываю

List<Customer> findAll();

У меня была большая проблема с производительностью.

Проблема заключается в том, что мне нужно вызвать метод findAll со спецификациями для клиента, поэтому я не могу использовать метод, который возвращает список массивов объектов.

Как написать метод для поиска всех клиентов со спецификациями для объекта "Клиент", который возвращает только идентификаторы.

как это:

List<Long> findAll(Specification<Customer> spec);

Я не могу использовать в этом случае нумерацию страниц.

Пожалуйста помоги.

Ответ 1

Я решил проблему.

(В результате у нас будет разреженный объект Customer только с id и именем)

Определите свой собственный репозиторий:

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;
    }
}

Ответ 2

Почему бы не использовать аннотацию @Query?

@Query("select p.id from #{#entityName} p")
List<Long> getAllIds();

Единственный недостаток, который я вижу, - это когда атрибут id изменяется, но поскольку это очень распространенное имя и вряд ли изменится (id = первичный ключ), это должно быть в порядке.

Ответ 3

Теперь это поддерживается Spring Data с использованием проекций:

interface SparseCustomer {  

  String getId(); 

  String getName();  
}

Чем в вашем Customer хранилище

List<SparseCustomer> findAll(Specification<Customer> spec);

EDIT:
Как отмечает Radouane, проекции ROUFID со спецификациями в настоящее время не работают из-за ошибки.

Но вы можете использовать библиотеку спецификации с проекцией, чтобы обойти этот недостаток Spring Data Jpa.

Ответ 4

К сожалению Projections не работает с specifications. JpaSpecificationExecutor возвращает только список, набранный агрегированным корнем, управляемым репозиторием (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 с объектом, который вы хотите вернуть.

@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);

Надеюсь, это поможет.