Spring Data JPA中基于roles字段使用IN查询报错，求解决方案

问题原因 #

你当前的写法是尝试判断Menu的roles集合是否包含在传入的roles集合中，但root.get(Menu_.roles)返回的是Collection<Role>类型，in()方法接收集合时会触发类型不匹配错误——JPA无法直接将Role对象集合与集合类型的字段做IN比较。实际需求应该是筛选出关联的Role的ID在指定列表中的Menu。

解决方案1：通过JOIN关联Role表查询 #

修改buildSearchSpecification方法，通过JOIN关联Role实体，直接对Role的id字段执行IN查询，同时注意去重避免多对多关联导致的重复结果：

private Specification<Menu> buildSearchSpecification(MenuDTO menuDTO) {
    return (root, query, criteriaBuilder) -> {
        List<Predicate> predicates = new ArrayList<>();
        if (StringUtils.hasText(menuDTO.getRoleIds())) {
            List<Long> roleIds = Convert.toList(Long.class, menuDTO.getRoleIds());
            // 显式关联Role表
            Join<Menu, Role> roleJoin = root.join(Menu_.roles);
            // 对Role的id字段执行IN查询
            predicates.add(roleJoin.get(Role_.id).in(roleIds));
        }
        // 多对多关联查询必须去重，避免同一Menu因关联多个符合条件的Role而重复返回
        query.distinct(true);
        return criteriaBuilder.and(predicates.toArray(new Predicate[0]));
    };
}

解决方案2：使用Exists子查询实现 #

如果不想通过JOIN，也可以用Exists子查询完成需求，这种写法更直观体现"存在关联指定Role的Menu"的逻辑：

private Specification<Menu> buildSearchSpecification(MenuDTO menuDTO) {
    return (root, query, criteriaBuilder) -> {
        List<Predicate> predicates = new ArrayList<>();
        if (StringUtils.hasText(menuDTO.getRoleIds())) {
            List<Long> roleIds = Convert.toList(Long.class, menuDTO.getRoleIds());
            // 构建子查询：查询关联当前Menu且Role id在指定列表中的记录
            Subquery<Long> subquery = query.subquery(Long.class);
            Root<Menu> subRoot = subquery.from(Menu.class);
            Join<Menu, Role> subJoin = subRoot.join(Menu_.roles);
            subquery.select(subRoot.get(Menu_.id))
                    .where(criteriaBuilder.equal(subRoot.get(Menu_.id), root.get(Menu_.id)),
                            subJoin.get(Role_.id).in(roleIds));
            predicates.add(criteriaBuilder.exists(subquery));
        }
        return criteriaBuilder.and(predicates.toArray(new Predicate[0]));
    };
}

内容的提问来源于stack exchange，提问作者炸娃程序猿