JPA 继承 @EntityGraph 包括可选的子类关联

Question

鉴于以下域模型，我想加载所有 Answers，包括它们的 Values 和它们各自的子子级，并将其放入 AnswerDTO 中，然后转换为 JSON。我有一个可行的解决方案，但它遇到了我想通过使用临时@EntityGraph 来解决的 N+1 问题。所有关联都配置LAZY。

@Query("SELECT a FROM Answer a")
@EntityGraph(attributePaths = {"value"})
public List<Answer> findAll();

在Repository 方法上使用临时@EntityGraph，我可以确保预取值以防止Answer->Value 关联上的N+1。虽然我的结果很好，但还有另一个 N+1 问题，因为延迟加载了 MCValues 的 selected 关联。

使用这个

@EntityGraph(attributePaths = {"value.selected"})

失败，因为selected 字段当然只是Value 实体的一部分：

Unable to locate Attribute  with the the given name [selected] on this ManagedType [x.model.Value];

如果值为MCValue，我如何告诉JPA 仅尝试获取selected 关联？我需要optionalAttributePaths之类的东西。

Answer 1

如果关联属性是超类的一部分并且也是所有子类的一部分，则只能使用EntityGraph。否则，EntityGraph 将始终以您当前获得的 Exception 失败。

避免 N+1 选择问题的最佳方法是将查询拆分为 2 个查询：

第一个查询使用EntityGraph 获取MCValue 实体，以获取selected 属性映射的关联。在该查询之后，这些实体然后存储在 Hibernate 的第一级缓存/持久性上下文中。 Hibernate 将在处理第二个查询的结果时使用它们。

@Query("SELECT m FROM MCValue m") // add WHERE clause as needed ...
@EntityGraph(attributePaths = {"selected"})
public List<MCValue> findAll();

第二个查询然后获取Answer 实体并使用EntityGraph 来获取关联的Value 实体。对于每个Value 实体，Hibernate 将实例化特定的子类并检查第一级缓存是否已经包含该类和主键组合的对象。如果是这种情况，Hibernate 会使用一级缓存中的对象，而不是查询返回的数据。

@Query("SELECT a FROM Answer a")
@EntityGraph(attributePaths = {"value"})
public List<Answer> findAll();

因为我们已经获取了所有 MCValue 实体以及关联的 selected 实体，所以我们现在获取了 Answer 实体以及初始化的 value 关联。如果关联包含MCValue 实体，则其selected 关联也将被初始化。

Answer 2

我不知道 Spring-Data 在那里做什么，但要做到这一点，您通常必须使用 TREAT 操作符才能访问子关联，但该操作符的实现非常错误。 Hibernate 支持隐式子类型属性访问，这是您在此处需要的，但显然 Spring-Data 无法正确处理此问题。我可以建议您查看Blaze-Persistence Entity-Views，这是一个在 JPA 之上工作的库，它允许您将任意结构映射到您的实体模型。您可以以类型安全的方式映射 DTO 模型，也可以映射继承结构。您的用例的实体视图可能如下所示

@EntityView(Answer.class)
interface AnswerDTO {
  @IdMapping
  Long getId();
  ValueDTO getValue();
}
@EntityView(Value.class)
@EntityViewInheritance
interface ValueDTO {
  @IdMapping
  Long getId();
}
@EntityView(TextValue.class)
interface TextValueDTO extends ValueDTO {
  String getText();
}
@EntityView(RatingValue.class)
interface RatingValueDTO extends ValueDTO {
  int getRating();
}
@EntityView(MCValue.class)
interface TextValueDTO extends ValueDTO {
  @Mapping("selected.id")
  Set<Long> getOption();
}

通过 Blaze-Persistence 提供的 spring 数据集成，您可以像这样定义一个存储库并直接使用结果

@Transactional(readOnly = true)
interface AnswerRepository extends Repository<Answer, Long> {
  List<AnswerDTO> findAll();
}

它将生成一个 HQL 查询，该查询仅选择您在 AnswerDTO 中映射的内容，如下所示。

SELECT
  a.id, 
  v.id,
  TYPE(v), 
  CASE WHEN TYPE(v) = TextValue THEN v.text END,
  CASE WHEN TYPE(v) = RatingValue THEN v.rating END,
  CASE WHEN TYPE(v) = MCValue THEN s.id END
FROM Answer a
LEFT JOIN a.value v
LEFT JOIN v.selected s

Answer 3

我的最新项目使用了 GraphQL（对我来说是第一次），我们在 N+1 查询方面遇到了一个大问题，并试图优化查询以仅在需要时加入表。我发现Cosium / spring-data-jpa-entity-graph 无可替代。它扩展了JpaRepository 并添加了将实体图传递给查询的方法。然后，您可以在运行时构建动态实体图，以仅为您需要的数据添加左连接。

我们的数据流如下所示：

1. 接收 GraphQL 请求
2. 解析 GraphQL 请求并转换为查询中的实体图节点列表
3. 从发现的节点创建实体图并传递到存储库中执行

为了解决实体图中不包含无效节点的问题（例如来自 graphql 的__typename），我创建了一个处理实体图生成的实用程序类。调用类传入它为其生成图的类名，然后根据 ORM 维护的元模型验证图中的每个节点。如果节点不在模型中，则将其从图形节点列表中删除。 （这个检查需要递归并且检查每个孩子）

在找到这个之前，我已经尝试过 Spring JPA / Hibernate 文档中推荐的投影和所有其他替代方案，但似乎没有什么可以优雅地解决问题，或者至少需要大量额外的代码

Answer 4

在您发表评论后编辑：

抱歉，我在第一轮中没有理解您的问题，您的问题发生在 spring-data 启动时，不仅在您尝试调用 findAll() 时。

所以，您现在可以浏览完整的示例，可以从我的 github 中提取： https://github.com/bdzzaid/stackoverflow-java/blob/master/jpa-hibernate/

您可以在此项目中轻松重现并修复您的问题。

实际上，Spring data 和 hibernate 默认无法确定“选定”图形，您需要指定收集选定选项的方式。

所以首先，你必须声明类Answer的NamedEntityGraphs

如你所见，Answer

类的value属性有两个NamedEntityGraph

• 所有值中第一个与加载没有特定关系的

• 特定Multichoice值的第二个。如果你删除这个，你会重现异常。

其次，如果要获取 LAZY

类型的数据，则需要在事务上下文中answerRepository.findAll()

@Entity
@Table(name = "answer")
@NamedEntityGraphs({
    @NamedEntityGraph(
            name = "graph.Answer", 
            attributeNodes = @NamedAttributeNode(value = "value")
    ),
    @NamedEntityGraph(
            name = "graph.AnswerMultichoice",
            attributeNodes = @NamedAttributeNode(value = "value"),
            subgraphs = {
                    @NamedSubgraph(
                            name = "graph.AnswerMultichoice.selected",
                            attributeNodes = {
                                    @NamedAttributeNode("selected")
                            }
                    )
            }
    )
}
)
public class Answer
{

    @Id
    @GeneratedValue(strategy = GenerationType.IDENTITY)
    @Column(updatable = false, nullable = false)
    private int id;

    @OneToOne(cascade = CascadeType.ALL)
    @JoinColumn(name = "value_id", referencedColumnName = "id")
    private Value value;
// ..
}