JPA+Hibernate单向@ManyToOne关联如何避免N+1查询问题

核心问题原因 #

你当前使用的是非主键关联的单向@ManyToOne，Hibernate对这类关联的懒加载默认需要字节码增强支持，在你禁用字节码增强的前提下，就算配置fetch = FetchType.LAZY也会在加载DummyA时自动触发关联DummyB的查询，这是N+1问题的根本来源。而分页计数异常是因为关联查询时Hibernate自动生成的count语句会把关联表的行数算入总计数，导致结果错误。

可行解决方案 #

方案1：自定义Criteria查询手动关联fetch，同时解决N+1和分页计数问题 #

如果你需要在查询DummyA时同时加载所有关联DummyB，通过自定义查询手动指定fetch关联、单独编写count查询即可，完全兼容现有Specification+Pageable的调用方式：

@Repository
public class CustomDummyARepositoryImpl implements CustomDummyARepository {

    @PersistenceContext
    private EntityManager entityManager;

    @Override
    public Page<DummyA> findAllWithDummyB(Specification<DummyA> spec, Pageable pageable) {
        // 构造内容查询，手动fetch关联避免N+1
        CriteriaBuilder cb = entityManager.getCriteriaBuilder();
        CriteriaQuery<DummyA> contentQuery = cb.createQuery(DummyA.class);
        Root<DummyA> aRoot = contentQuery.from(DummyA.class);
        aRoot.fetch("dummyB", JoinType.LEFT);
        contentQuery.where(spec.toPredicate(aRoot, contentQuery, cb));

        List<DummyA> content = entityManager.createQuery(contentQuery)
                .setFirstResult((int) pageable.getOffset())
                .setMaxResults(pageable.getPageSize())
                .getResultList();

        // 构造独立的count查询，保证分页计数正确
        CriteriaQuery<Long> countQuery = cb.createQuery(Long.class);
        Root<DummyA> countRoot = countQuery.from(DummyA.class);
        countQuery.select(cb.count(countRoot.get("id")))
                .where(spec.toPredicate(countRoot, countQuery, cb));
        Long total = entityManager.createQuery(countQuery).getSingleResult();

        return new PageImpl<>(content, pageable, total);
    }
}

你只需要把原有调用findAll(specification, pageable)的逻辑替换为调用这个自定义方法即可。

方案2：开启Hibernate批量抓取配置，无侵入解决N+1 #

如果不想修改现有业务代码，直接在项目配置文件中添加批量抓取配置即可，会把N次单独查询DummyB的请求合并为批量查询，性能提升明显：

# Spring Boot配置示例，一次最多批量查询50个DummyB，把N+1转化为1 + 向上取整(N/50)次查询
spring.jpa.properties.hibernate.default_batch_fetch_size=50

该方案无需修改任何业务代码，原有分页逻辑完全不受影响，计数不会出现异常。

方案3：拆分映射字段，从根源避免自动关联加载 #

如果大多数业务场景不需要用到DummyB关联，可以拆分实体字段，只有需要时才主动触发关联查询：

@Entity
@Table(name = "dummy_a")
@Data
public class DummyA implements Serializable {
    private static final long serialVersionUID = 1L;

    @Id
    @GeneratedValue(strategy = GenerationType.IDENTITY)
    private Integer id;

    // 直接映射外键字段，正常读写不会触发关联查询
    @Column(name = "dummy_b_name")
    private String dummyBName;

    // 关联字段仅在主动调用getDummyB()时才会触发查询
    @ManyToOne(fetch = FetchType.LAZY)
    @JoinColumn(name = "dummy_b_name", referencedColumnName = "name", updatable = false, insertable = false)
    private DummyB dummyB;
}

如果需要批量加载DummyB，可以先收集所有查询到的DummyA的dummyBName值，再调用DummyBRepository.findByNameIn(names)一次性查询所有关联对象，自己做结果映射，完全规避N+1问题。

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