选择特定列的Spring Data JPA规范

Question

我们可以通过在存储库接口中编写自定义@Query方法来选择特定列。但是，我不想为不同的属性编写这么多方法。

我试过这个，但它一直返回整个对象。

public class MySpecifications {

    public static Specification<MyInfo> propertiesWithId(final String[] properties, final Object id, final String idProperty)
    {

        return new Specification<MyInfo>() {

            @Override
            public Predicate toPredicate(Root<MyInfo> root,
                    CriteriaQuery<?> query, CriteriaBuilder cb) {

                query = cb.createTupleQuery(); //tried cb.createQuery(MyInfo.class); as well

                List<Selection<? extends Object>> selectionList = new ArrayList<Selection<? extends Object>>();

                for (String property : properties) {

                    Selection<? extends Object> selection = root.get(property);

                    selectionList.add(selection);
                }

                return query.multiselect(selectionList).where(cb.equal(root.get(idProperty), id)).getRestriction();
            }

        };
    }
}

用作：

MyInfo findOne(Specification(properties,idValue, idProperty));

这是正确的方法吗？错在哪里？

Answer 1

我试过这个，但它一直返回整个对象。

此方法返回与给定规范匹配的单个实体。请检查这里

根据我的理解，这是正确的方法。U可以正常访问实体的属性（例如MyInfo.getId属性（））

Answer 2

如果这是错误的，请纠正我。

通常Specification位于XxxSpecification类中，并具有静态方法。它们通常以以下两种方式使用：

public class SwVecSpecifications {

    // Method 1. 
    public static Specification<WhiteVecEntity> conditions(String groupId, String appId, String typeId) {
        return (Specification<WhiteVecEntity>) (root, query, cb) -> {
            Predicate p3 = cb.equal(root.get("groupId"), groupId);
            Predicate p2 = cb.equal(root.get("appId"), appId);
            Predicate condition = cb.and(p2,p3);
            if (typeId != null && !typeId.isEmpty()) {
                Predicate p1 = cb.equal(root.get("typeId"), typeId);
                condition = cb.and(condition, p1);
            }
            return query.select(root.get("id")).where(condition).getRestriction();
        };
    }

    // Method 2.
    static void findByConditions(EntityManager em) {
        String groupId = "";
        String typeId = "";

        CriteriaBuilder cb = em.getCriteriaBuilder();
        CriteriaQuery<SuspectVecEntity> query = cb.createQuery(SuspectVecEntity.class);
        Root<SuspectVecEntity> root = query.from(SuspectVecEntity.class);

        Predicate p1 = cb.equal(root.get("groupId"), groupId);
        Predicate p2 = cb.equal(root.get("appId"), appId);
        Predicate condition = cb.and(p2,p3);
        if (typeId != null && !typeId.isEmpty()) {
            Predicate p1 = cb.equal(root.get("typeId"), typeId);
            condition = cb.and(condition, p1);
        }
        CriteriaQuery<SuspectVecEntity> cq = query.select(root.get("id")).where(condition);
        List<SuspectVecEntity> resultList = em.createQuery(cq).getResultList(); // resultList has Entity that only contains id
    }
}

方法1的存储库：

public interface SuspectVecRepository extends JpaRepository<SuspectVecEntity, Long>, JpaSpecificationExecutor<SuspectVecEntity>  {
     List<SuspectVecEntity> findAll(Specification specs);
}

服务：

@Autowired
EntityManager em;
void foo() {
     // Method 1:
     List<SuspectVecEntity> r1 = findAll(SwVecSpecifications.conditions());
     // Method 2:
     List<SuspectVecEntity> r2 = SwVecSpecifications.findByConditions(em); 
}

这两种方法的区别在于:

方法1，DB查询返回整列。

原因是存储库findAll采用规范。Specification作为限制返回，因此，它查询 中的所有列。它有一个＜code＞select＜/code＞来指定列，它位于＜code＞选择＜/code＞中，而不是＜code＞限制＜/code>中。在类＜code＞QueryStructure＜/code＞中查看更多信息。

方法二，DB查询只返回选择列。

它看起来像返回一个Entity，这是因为它是一个链接的泛型类型调用，并且该泛型类型是Entity。

Answer 3

当前的Spring data jpa规范执行器仅限于在其中子句中的条件，因此您不能更改选定的列，它隐含地仅限于完整的实体（请查看Jpa专业代码执行器接口留档）。您必须使用自定义存储库实现，或者移动到命名查询-

Spring Data JPA和Querydsl使用bean/构造函数投影获取列子集