以列表作为参数的spring-data查询方法

时间:2013-04-23 16:14:33

标签: java spring-data spring-data-jpa

我们必须从数据库中查询数据,我们需要找到与键值对列表匹配的实体。我们认为使用Spring Data JPA是一个不错的主意,因为我们还需要分页。

我们创建的表格如下:

terminal(ID,NUMBER,NAME);
terminal_properties(ID,KEY,VALUE,TERMINAL_FK);

是否可以定义查询方法来获取包含给定键/值对的属性的所有终端?

这样的事情: List< Terminal> findByPropertiesKeyAndValue(列表与LT;性>);

1 个答案:

答案 0 :(得分:1)

我没有执行代码,但是给出了正确的import语句,这至少可以编译。根据您的实体定义,某些属性可能需要进行调整,无论如何,您应该了解如何处理此问题。

我的条件查询基于以下SQL:

SELECT * FROM TERMINAL
  WHERE ID IN (
    SELECT TERMINAL_FK FROM TERMINAL_PROPERTIES
      WHERE (KEY = 'key1' AND VALUE = 'value1')
        OR (KEY = 'key2' AND VALUE = 'value2')
        ...
      GROUP BY TERMINAL_FK
      HAVING COUNT(*) = 42
  )

列出每个名称/值对的位置,42只表示名称/值对的数量。

所以我假设你定义了一个这样的存储库:

public interface TerminalRepository extends CrudRepository<Terminal, Long>, JpaSpecificationExecutor {
}

扩展JpaSpecificationExecutor以便使用标准API非常重要。

然后你可以建立一个这样的标准查询:

public class TerminalService {

  private static Specification<Terminal> hasProperties(final Map<String, String> properties) {
    return new Specification<Terminal>() {
      @Override
      public Predicate toPredicate(Root<Terminal> root, CriteriaQuery<?> query, CriteriaBuilder builder) {
        // SELECT TERMINAL_FK FROM TERMINAL_PROPERTIES
        Subquery<TerminalProperty> subQuery = query.subquery(TerminalProperty.class);
        Root propertyRoot = subQuery.from(TerminalProperty.class);
        subQuery.select(propertyRoot.get("terminal.id"));
        Predicate whereClause = null;
        for (Map.Entry<String, String> entry : properties.entrySet()) {
          // (KEY = 'key1' AND VALUE = 'value1')
          Predicate predicate = builder.and(builder.equal(propertyRoot.get("key"),
              entry.getKey()), builder.equal(propertyRoot.get("value"), entry.getValue()));
          if (whereClause == null) {
            whereClause = predicate;
          } else {
            // (...) OR (...)
            whereClause = builder.or(whereClause, predicate);
          }
        }
        subQuery.where(whereClause);
        // GROUP BY TERMINAL_FK
        subQuery.groupBy(propertyRoot.get("terminal.id"));
        // HAVING COUNT(*) = 42
        subQuery.having(builder.equal(builder.count(propertyRoot), properties.size()));

        // WHERE ID IN (...)
        return query.where(builder.in(root.get("id")).value(subQuery)).getRestriction();
      }
    };
  }

  @Autowired
  private TerminalRepository terminalRepository;

  public Iterable<Terminal> findTerminalsWith(Map<String, String> properties) {
    // this works only because our repository implements JpaSpecificationExecutor
    return terminalRepository.findAll(hasProperties(properties));
  }
}

您显然可以将Map<String, String>替换为Iterable<TerminalProperty>,但这会让人觉得奇怪,因为它们似乎与特定Terminal绑定。