Gangmax Blog

解决Hibernate Criteria查询中distinct root entity的问题

Migarated from here at ‘2012-06-04 17:16:25’.

举个例子说明这个问题：有一种Entity叫article，另外一种Entity叫tag，一个article可以关联多个tag。在查询article的时候，我们可以使用tag作为查询条件。

比如有一个article “ABC” 关联”tagA”和”tagB”。我们在查询的时候，同时把”tagA”和”tagB”作为查询条件，想要查询出所有标示为”tagA”或者”tagB”的article。这种情况下，使用Hibernate criteria查询时会将article “ABC”作为查询结果返回两次。这是由SQL的join决定的。如果用户要求这种情况下只返回一条记录该怎么做呢？

看了Hibernate的javadoc之后我本来以为“criteria.setResultTransformer(Criteria.DISTINCT_ROOT_ENTITY)”方法可以解决这个问题，但是经过尝试不行。而且即使可以，据我所知（如果不对请指正），ResultTransformer也只是对查询结果集做手脚，这样会造成分页的问题（本来一页应该有10条记录，但如果仅仅是在查询当前页做distinct，会导致本页只有9条）。所以不得不找其他方法。

给我帮助的是来自以下链接的文章。

该文章解决的方法基于以下思路：

  1. 使用criteria查询的时候不查询整个记录，只查询”distinct(root_entity_id_field)”，这样就可以保证查询的结果是唯一的；

  2. 再使用一个条件为in (id_lists got from the above query)的查询、并包含该entity的所有字段的语句进行查询，得到的结果就是想要的结果。

代码如下：

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
  if (queryDate != null) {
  // project result to distinct ids,
  // including columns used for sorting
  criteria.setProjection(Projections.distinct(
    Projections.projectionList()
      .add(Projections.id())
      .add(Projections.property("birthDate"))
  ));
  List list = criteria.list();

  // unfortunately we have to copy the ids out of the
  // resulting Object[] List
  List idlist = new ArrayList<long>();
  for (Iterator iditer = list.iterator(); iditer.hasNext();) {
    Object[] record = (Object[]) iditer.next();
    idlist.add((Long)record[0]);
  }

  // another Hibernate stupidity: empty Lists cause
  // Expression.in to throw an error
  if (idlist.size() > 0) {
    criteria = getSession().createCriteria(Cat.class);
    criteria.add(Expression.in("id", idlist));
  }
  else {
    return new ArrayList();
  }
}

Comments