如何用JPA Specification限制联合查询仅包含指定子实体类型？

现有如下Java JPA实体模型：

@Entity
@Table(name = "PARENT")
@Inheritance(strategy = InheritanceType.TABLE_PER_CLASS)
public abstract class ParentEntity { }

@Entity
@Table(name = "PARENT_Child1")
public abstract class Child1Entity extends ParentEntity  { }

@Entity
@Table(name = "PARENT_Child2")
public abstract class Child2Entity extends ParentEntity { }

@Entity
@Table(name = "PARENT_Child3")
public abstract class Child3Entity extends ParentEntity { }

调用Spring Data JPA仓库方法parentRepository.findAll(specification);时，生成的SQL查询如下：

select parent.id from (select id, 0 as clazz_ from PARENT union all select id, 1 as clazz_ from PARENT_CHILD1 union all select id, 2 as clazz_ from PARENT_CHILD2 union all select id, 3 as clazz_ from PARENT_CHILD3) parent

（注：修正了原SQL中的语法错误）

目前该查询符合预期，但希望仅包含Child1Entity和Child2Entity对应的表，排除Child3Entity。由于JPA的HQL不支持union关键字，请问能否通过JPA Specification实现该需求？

可以通过JPA Specification结合JPA的多态查询机制实现该需求，以下是两种可行方案：

方案一：直接通过type()匹配子类 #

利用JPA提供的type()函数，在Specification中构建条件，仅筛选Child1Entity和Child2Entity类型的实例：

public class ParentSpecifications {
    public static Specification<ParentEntity> includeOnlyChild1AndChild2() {
        return (root, query, criteriaBuilder) -> {
            return criteriaBuilder.or(
                criteriaBuilder.equal(root.type(), Child1Entity.class),
                criteriaBuilder.equal(root.type(), Child2Entity.class)
            );
        };
    }
}

调用时传入该Specification即可：

parentRepository.findAll(ParentSpecifications.includeOnlyChild1AndChild2());

使用该方案后，Hibernate会自动调整底层SQL，仅对PARENT_CHILD1和PARENT_CHILD2表执行union all操作，自动排除PARENT_CHILD3表。

方案二：显式配置鉴别器列（可选优化） #

如果允许调整实体结构，可以给父类添加鉴别器列配置，让类型标识更直观：

@Entity
@Table(name = "PARENT")
@Inheritance(strategy = InheritanceType.TABLE_PER_CLASS)
@DiscriminatorColumn(name = "CLAZZ_", discriminatorType = DiscriminatorType.INTEGER)
public abstract class ParentEntity { }

给每个子类指定对应的鉴别器值：

@Entity
@Table(name = "PARENT_Child1")
@DiscriminatorValue("1")
public abstract class Child1Entity extends ParentEntity { }

@Entity
@Table(name = "PARENT_Child2")
@DiscriminatorValue("2")
public abstract class Child2Entity extends ParentEntity { }

@Entity
@Table(name = "PARENT_Child3")
@DiscriminatorValue("3")
public abstract class Child3Entity extends ParentEntity { }

之后在Specification中直接筛选鉴别器值：

public static Specification<ParentEntity> includeOnlyChild1AndChild2() {
    return (root, query, criteriaBuilder) -> {
        return root.get("clazz_").in(1, 2);
    };
}

这种方式生成的SQL会直接过滤掉clazz_=3的记录，从而排除PARENT_CHILD3表的参与，逻辑更清晰。

注意事项 #

• 对于TABLE_PER_CLASS继承策略，即使不手动配置鉴别器列，JPA实现（如Hibernate）也会自动生成类型标识列，因此方案一无需修改实体结构即可生效。
• 不要尝试手动在HQL中拼接union，JPA规范不支持该关键字，必须依赖多态查询机制实现表的筛选。

内容的提问来源于stack exchange，提问作者Thomas