Spring JPA 衍生查询: 解决 getTotalElements 为 0 的问题

2024-03-18 08:50:20

使用 Spring JPA 衍生查询：解决返回 getTotalElements 为 0 的问题

介绍

在使用 Spring JPA 时，我们可以使用衍生查询来简化数据库访问。其中，findByCategoryOrCategory 衍生查询方法允许我们根据类属性查找实体。本文将探讨如何使用该方法，并解决使用该方法时遇到的 getTotalElements 返回 0 的问题。

问题分析

当 getTotalElements 为 0 时，表示查询未返回任何结果。这可能是因为在调用 findByCategoryOrCategory 方法时，传递了不正确的参数值。

解决方法

要解决此问题，我们需要确保 categoryOne 和 categoryTwo 参数的值与数据库中存在的类别相匹配。此外，我们可以使用 PageRequest.of(0, Integer.MAX_VALUE) 创建一个包含所有结果的分页对象，以检索所有匹配的实体。

解决后的方法

public ResponseEntity<APIResponse> getQuizByTwoCategoryOnly(String categoryOne, String categoryTwo, int page, int size) {
    Pageable pageable = PageRequest.of(0, Integer.MAX_VALUE);
    Page<TriviaQuestion> response = questionRepository.findByCategoryOrCategory(categoryOne, categoryTwo, pageable);
    log.info("I retrieved {} entities", response.getTotalElements());
    Set<QuizResponse> quizResponses = new HashSet<>();
    for (TriviaQuestion question : response) {
        // ... (rest of the code remains the same)
    }
    log.info("I am suppose to pass {} ", quizResponses.size());
    return ResponseEntity.ok(new APIResponse(HttpStatus.FOUND, quizResponses, "Quiz questions based on to categories"));
}

衍生查询用法

除了 findByCategoryOrCategory，我们还可以使用其他衍生查询方法，包括：

findByCategory：根据单个类别查找实体
findByCategoryIn：根据一组类别查找实体
findByCategoryNameContaining：根据类别名称的一部分查找实体

使用原始 SQL 查询或命名查询

如果衍生查询无法满足我们的需求，我们还可以使用原始 SQL 查询或命名查询。

原始 SQL 查询

List<TriviaQuestion> questions = entityManager.createNativeQuery("SELECT * FROM TRIVIA_QUESTION WHERE CATEGORY = ?1 OR CATEGORY = ?2", TriviaQuestion.class)
                .setParameter(1, categoryOne)
                .setParameter(2, categoryTwo)
                .getResultList();

命名查询

List<TriviaQuestion> questions = entityManager.createNamedQuery("TriviaQuestion.findByCategoryOrCategory")
                .setParameter("categoryOne", categoryOne)
                .setParameter("categoryTwo", categoryTwo)
                .getResultList();