Spring Predicate JpaSpecificationExecutor IN (Выбрать) Выражение

Question

У меня есть следующий Запрос , который возвращает меня одна строка :

SELECT * FROM activity_entity 
WHERE activity_id IN (SELECT activity_id FROM category_entity WHERE activity_type = 9999) 
AND activity_id = 'AA924EDC-7D55-2555-6569-7D54BDF4F71F' 
AND category_id = '45CF9A18-1718-482A-A1C6-CA77E23D29B1' 
AND internal_module_id = '311';

И следующий Spring Predicate :

public List<ActivityEntity> getRepositoryDataPrincipalActivity(AccessRepVisitLogInDto inDto,AccessRepEntity accessRep) {
    List<ActivityEntity> activityL = activityRepository.findAll(new Specification<ActivityEntity>() {

        @Override
        public Predicate toPredicate(Root<ActivityEntity> root, CriteriaQuery<?> query, CriteriaBuilder cb) {
            List<Predicate> predicates = new ArrayList<>();

            predicates.add(cb.equal(root.get("activityId"), accessRep.getActivityId()));
            predicates.add(cb.equal(root.get("categoryId"), inDto.getCategory_id()));
            predicates.add(cb.equal(root.get("internalModuleId"), inDto.getInternalModuleId()));

            return cb.and(predicates.toArray(new Predicate[0]));
        }
    });
    return activityL;
}

В моем предикате я хочу включить выражение IN (SELECT) :

activity_id IN (SELECT activity_id FROM category_entity WHERE activity_type = 9999)

Это проверка, которую мне нужно добавить в мой предикат, чтобы она правильно возвращает строку, которая мне нужна.

Answer 1

Subquery<Long> subquery = query.subquery(Long.class);
Root<EntityCategory> category = subquery.from(EntityCategory.class);

Predicate subPredicate = cb.equal(category.get("activityType"), 9999); 
subquery.select(category.get("activityId")).where(subPredicate);

predicates.add(root.get("id").in(subquery));