java - 选择给定元素的所有唯一对 - Criteria API

Question

我有一个“ session ”实体，其参数为:meetingId、catId、dogId。可能有很多 session 的 catId 和dogId 是相同的。我需要计算一只给定的猫见过多少只狗。换句话说，我需要计算给定猫的 session 次数，其中每次 session 都与不同的狗进行。

如何使用 springframework.data.mongodb.core.query.Criteria 实现此目的？

public long countUniqueByCatId(String catId) {
            Query query = new Query()
                    .addCriteria(Criteria.where(CAT_ID).is(catId))
                    ....
            return template.count(query, Meeting.class);

同样的问题也适用于流。

meetingsList.entrySet()
                    .stream()
                    .map { it -> it.value }
                    .filter { m -> (m.getCatId() == catId) }
                    ...
                    .count();

Answer 1

您可以尝试此代码

CriteriaBuilder cb = entityManager.getCriteriaBuilder();
CriteriaQuery<Long> query = cb.createQuery(Long.class);
Root<meeting> meetings = query.from(meeting.class);

//here you can change meetings to your new catdogIds class 
cq.select(cb.countDistinct(meetings)).where(Criteria.where(CAT_ID).is(catId));
return entityManager.createQuery(cq).getSingleResult();

对于流，您可以使用.distinct()，但您应该实现 equals 方法就像这个例子一样 https://www.journaldev.com/31860/java-stream-distinct-function

也可以创建一个包含catId和dogId的类，然后重写equals方法，当比较对象具有相同的猫和狗id时，将结果列表转换为集合，以返回true。