如何在Spring Boot 3.X+Spring Data JPA应用中检测Hibernate/JPA的N+1问题

1. 基于Hibernate统计功能的自定义测试断言 #

Hibernate自带的统计模块可以直接用于单元/集成测试中校验查询次数，实现半自动化检测：

• 第一步：在application-test.yml中开启统计功能

spring:
  jpa:
    properties:
      hibernate:
        generate_statistics: true
        format_sql: true

• 第二步：在测试类中注入Statistics对象，执行业务逻辑后断言查询次数

@SpringBootTest
class UserRepositoryTest {

    @Autowired
    private UserRepository userRepository;

    @Autowired
    private Statistics statistics;

    @BeforeEach
    void resetStats() {
        statistics.clear();
    }

    @Test
    void shouldNotHaveNPlusOneWhenFetchingUsersWithOrders() {
        // 执行查询逻辑
        List<User> users = userRepository.findAll();
        users.forEach(user -> user.getOrders().size());

        // 断言总查询次数（预期1次主查询，若触发N+1则次数会远大于1）
        assertThat(statistics.getQueryExecutionCount()).isEqualTo(1);
        // 也可以针对性统计特定SQL的执行次数
        long userQueryCount = statistics.getQueries().entrySet().stream()
                .filter(entry -> entry.getKey().contains("from User"))
                .mapToLong(Map.Entry::getValue)
                .sum();
        assertThat(userQueryCount).isEqualTo(1);
    }
}

这种方案能在测试失败时直接定位问题，适合针对性校验核心业务逻辑。

2. 基于datasource-proxy的SQL重复检测 #

利用你提到的datasource-proxy代理数据源，拦截所有SQL执行并统计重复次数：

• 第一步：引入适配Spring Boot 3的依赖

<dependency>
    <groupId>net.ttddyy</groupId>
    <artifactId>datasource-proxy</artifactId>
    <version>1.10</version>
    <scope>test</scope>
</dependency>

• 第二步：配置代理数据源与自定义拦截器

@TestConfiguration
public class ProxyDataSourceConfig {

    @Bean
    @Primary
    public DataSource dataSource(DataSource originalDataSource) {
        ProxyDataSource proxyDataSource = new ProxyDataSource();
        proxyDataSource.setDataSource(originalDataSource);
        proxyDataSource.setListener(new NPlusOneDetectionListener());
        return proxyDataSource;
    }

    static class NPlusOneDetectionListener implements QueryExecutionListener {
        private final Map<String, Integer> sqlExecutionCount = new HashMap<>();

        @Override
        public void beforeQuery(ExecutionInfo execInfo, List<?> paramValues) {}

        @Override
        public void afterQuery(ExecutionInfo execInfo, List<?> paramValues, QueryResult queryResult) {
            String sql = execInfo.getSql().replaceAll("\\s+", " ").trim();
            sqlExecutionCount.put(sql, sqlExecutionCount.getOrDefault(sql, 0) + 1);
        }

        @PreDestroy
        void reportNPlusOne() {
            System.out.println("=== 疑似N+1问题的SQL ===");
            sqlExecutionCount.entrySet().stream()
                    .filter(entry -> entry.getValue() > 1)
                    .forEach(entry -> System.out.printf("SQL: %s，执行次数: %d%n", entry.getKey(), entry.getValue()));
        }
    }
}

运行测试套件时，拦截器会自动统计所有SQL的执行次数，测试结束后输出重复执行的SQL，快速定位潜在N+1问题。

3. 用Hypersistence Optimizer自动检测 #

Hypersistence Optimizer是专门针对Hibernate/JPA的性能优化工具，内置N+1问题检测：

• 第一步：引入依赖

<dependency>
    <groupId>io.hypersistence</groupId>
    <artifactId>hypersistence-optimizer</artifactId>
    <version>2.2.13</version>
    <scope>test</scope>
</dependency>

• 第二步：初始化工具（在Spring Boot启动类中配置）

@SpringBootApplication
public class Application {
    public static void main(String[] args) {
        SpringApplication.run(Application.class, args);
    }

    @Bean
    public HypersistenceOptimizer hypersistenceOptimizer(EntityManagerFactory entityManagerFactory) {
        return new HypersistenceOptimizer(new HibernateEntityManagerFactoryProvider(entityManagerFactory));
    }
}

运行测试时，工具会自动分析Hibernate的查询行为，在控制台或日志中输出N+1问题告警，包含触发问题的代码位置与SQL语句。

4. 基于SQL日志的批量分析 #

无需额外依赖，通过开启Hibernate SQL日志+脚本分析实现检测：

• 第一步：在application-test.yml中开启SQL日志

logging:
  level:
    org.hibernate.SQL: DEBUG
    org.hibernate.type.descriptor.sql.BasicBinder: TRACE

• 第二步：运行测试并将日志输出到文件

./mvnw test > test-sql.log

• 第三步：用Python/Shell脚本提取日志中的SQL语句，统计重复次数，找出执行次数超过1次的SQL，这些就是疑似N+1的问题点。

内容的提问来源于stack exchange，提问作者Dónal