如何简化两种类似方法的代码？

Question

我想重构这两个方法，除了“maxResult （）”之外，它们实际上是相同的，这两个方法指的是两个不同的 get，一个返回我单个用户，另一个返回列表。我如何简化这两种方法（总是如果有意义的话）这是两种方法：

第一种方法：

public List findFirstByTransactionId(String transactionId) {
        CriteriaBuilder builder = em.getCriteriaBuilder();
        CriteriaQuery<User> criteria = builder.createQuery(User.class);
        Root<User> root = criteria.from(User.class);
        criteria.select(root).where(builder.equal(root.get(User_.transactionId), transactionId));
        criteria.orderBy(builder.asc(root.get(User_.date)));
        TypedQuery<User> query = em.createQuery(criteria).setMaxResults(1);
         return query.getSingleResult();

第二种方法：

public List<User> findAllByTransactionId(String transactionId) {
        CriteriaBuilder builder = em.getCriteriaBuilder();
        CriteriaQuery<User> criteria = builder.createQuery(User.class);
        Root<User> root = criteria.from(User.class);
        criteria.select(root).where(builder.equal(root.get(User_.transactionId), transactionId));
        criteria.orderBy(builder.asc(root.get(User_.date)));
        TypedQuery<User> query = em.createQuery(criteria);
         return query.getResultList();

Answer 1

您在查询中调用的方法就像任何其他方法一样工作，因此您可以将一些调用放在 if 块中，如下所示：

public List<User> findByTransactionId(String transactionId, boolean onlyFirst) {
    CriteriaBuilder builder = em.getCriteriaBuilder();
    CriteriaQuery<User> criteria = builder.createQuery(User.class);
    Root<User> root = criteria.from(User.class);
    criteria.select(root).where(builder.equal(root.get(User_.transactionId), transactionId));
    criteria.orderBy(builder.asc(root.get(User_.date)));
    TypedQuery<User> query = em.createQuery(criteria);
    if (onlyFirst) {
      query=query.setMaxResults(1);
    }
    return query.getResultList();
}

Answer 2

最简单的选择通常是尝试将相同的代码移到它自己的方法中。在这里，您有多种选择，其中一种可能是这样的:

1. 编写一个新的（私有）方法：

private List<User> findFirstByTransactionId(String transactionId, boolean onlyOneResult) {
    CriteriaBuilder builder = em.getCriteriaBuilder();
    CriteriaQuery<User> criteria = builder.createQuery(User.class);
    Root<User> root = criteria.from(User.class);

    criteria.select(root).where(builder.equal(root.get(User_.transactionId), transactionId));
    criteria.orderBy(builder.asc(root.get(User_.date)));
    TypedQuery<User> query = em.createQuery(criteria);

    if (onlyOneResult) {
        query = query.setMaxResults(1);
    }
    
    return query.getResultList();
}

然后像这样重构你现有的方法：

public List<User> findFirstByTransactionId(String transactionId) {
    return findFirstByTransactionId(transactionId, false);
}

public List<User> findFirstByTransactionId(String transactionId) {
    return findFirstByTransactionId(transactionId, true);
}

现在，您消除了 8 行重复代码:)

Answer 3

更多的是附录。

其他建议添加布尔参数的答案是有效的，但是：干净的编码建议总是尽量减少参数数量。尤其是不鼓励使用此类布尔参数。当然，在这里这样做还是有意义的，以避免代码重复。

但是我会做什么：

< li >是的，内部有一个< code >个人分发名单

然后，这两个公共方法可以调用内部方法并传递< code > true /< code > false 。使用一个布尔值来做决定是一个实现细节，你应该避免让它在公共方面可见。