使用单表继承在实体层次结构上进行JPA标准查询

Question

说我有以下实体：

@Entity
@Inheritance(strategy = SINGLE_TABLE)
@DiscriminatorColumn(name = "type")
public abstract class BaseEntity { 
    private Date someDate;
    private Date otherDate;
    private boolean flag;
}

@Entity
@DiscriminatorValue("entity1")
public class Entity1 extends BaseEntity { 
    private String someProperty;
}

@Entity
@DiscriminatorValue("entity2")
public class Entity2 extends BaseEntity { 
    private String otherProperty;
}

我正在尝试建立一个标准查询，该查询根据BaseEntity和两个子类中的属性返回BaseEntity的实例。因此，本质上，我正在寻找与该伪SQL相对应的条件查询：

SELECT * FROM <BaseEntity table name>
WHERE someDate < ? AND otherDate > ? AND flag = ?
AND someProperty = ? AND otherProperty = ?;

我不想构建两个单独的查询，因为它们有很多重叠（即，大多数属性都在基类中）。但是，如果我将BaseEntity声明为根，我还没有想出一种在查询中引用子类属性的方法。是否可以建立这样的条件查询？

更新：

也许一些代码可以澄清这个问题。我本质上想做这样的事情：

CriteriaBuilder builder = ...;
CriteriaQuery<BaseEntity> query = ...;
Root<BaseEntity> root = ...;

query.select(root).where(builder.and(
        builder.lessThan(root.get(BaseEntity_.someDate), new Date()),
        builder.greaterThan(root.get(BaseEntity_.otherDate), new Date()),
        builder.isTrue(root.get(BaseEntity_.flag)),
        builder.equal(root.get(Entity1_.someProperty), "foo"),   <-- This won't work
        builder.equal(root.get(Entity2_.otherProperty), "bar")   <-- Neither will this
));

现在，我明白了为什么上面的代码示例不起作用，但是我想知道是否有解决方法。

Answer 1

我设法通过使用CriteriaBuilder.treat（）将BaseEntity根向下转换为与子类类型相对应的新根来解决此问题，如下所示：

CriteriaBuilder builder = ...;
CriteriaQuery<BaseEntity> query = ...;
Root<BaseEntity> root = ...;
Root<Entity1> entity1 = builder.treat(root, Entity1.class);
Root<Entity2> entity2 = builder.treat(root, Entity2.class);

query.select(root).where(builder.and(
        builder.lessThan(root.get(BaseEntity_.someDate), new Date()),
        builder.greaterThan(root.get(BaseEntity_.otherDate), new Date()),
        builder.isTrue(root.get(BaseEntity_.flag)),
        builder.equal(entity1.get(Entity1_.someProperty), "foo"),
        builder.equal(entity2.get(Entity2_.otherProperty), "bar")
));