springboot - SpringDataJpa中 Specification怎样使用in查询?

Question

SpringDataJpa中 Specification怎样使用in查询

实体类Menu.java如下

@Entity@Getter@Table(name = "t_sys_menu")@[e-mail]@org.hibernate.annotations.Where(clause = SysConstants.WHERE_DELETE)public class Menu extends TreeEntity<Menu, Long> {    @Column(nullable = false, length = 50)    private String name;    @Column(length = 100)    private String path;    @Column(length = 100)    private String componentPath;    @Column(columnDefinition = "int unsigned")    private Integer level;    /**     * 0表示目录，1表示菜单，2表示按钮     */    @Column(nullable = false, columnDefinition = "tinyint unsigned")    private Integer type;    @Column(length = 20)    private String icon;    @JoinColumn    @ManyToOne(fetch = FetchType.LAZY)    private Application application;    @Column(nullable = true, length = 50)    private String permission;    @Convert(converter = Status.Converter.class)    @Column(nullable = false, columnDefinition = "tinyint unsigned")    private Status status;    private String visible;    /**     * 打开方式 页签 或是 新窗口     */    private String target;    @Column(columnDefinition = "int unsigned")    private Integer sort;    @Column(length = 200)    public String description;    @ManyToMany(fetch = FetchType.LAZY)    @JoinTable(name = "t_sys_role_menu", joinColumns = @JoinColumn(name = "menu_id"), inverseJoinColumns = @JoinColumn(name = "role_id"))    private Collection<Role> roles;

想通过roles属性做in查询，代码如下

    private Specification<Menu> buildSearchSpecification(MenuDTO menuDTO) {        return (root, query, criteriaBuilder) -> {            List<Predicate> predicates = new ArrayList<>();            if (StringUtils.hasText(menuDTO.getRoleIds())) {                List<Role> roles = new ArrayList<>();                for (Long roleId : Convert.toList(Long.class, menuDTO.getRoleIds())) {                    roles.add(new Role(roleId));                }                predicates.add(root.get(Menu_.roles).in(roles));            }            return criteriaBuilder.and(predicates.toArray(new Predicate[0]));        };    }

错误如下：

Parameter value [com.appmtce.pojo.entity.role.Role@20f81e2a] did not match expected type [java.util.Collection (n/a)]

我的Specification方式的in查询写的哪里不对，如何修改正确？

Answer 1

我知道怎么弄了

predicates.add(root.get(Menu_.roles).in(roles));

改成了

predicates.add(root.join(Menu_.roles, JoinType.LEFT).in(roles));

Answer 2

你的代码中，in查询的参数类型不正确。in查询期望的是一个集合（Collection），而你传递的是一个Role对象。你需要将Role对象转换为对应的ID集合。

修改后的代码如下：

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());            predicates.add(root.get(Menu_.roles).in(roleIds));        }        return criteriaBuilder.and(predicates.toArray(new Predicate[0]));    };}

在这个修改后的代码中，我们首先将menuDTO.getRoleIds()转换为Long类型的列表，然后将其传递给in方法。这样，in方法就能正确地处理一个集合类型的参数了。