如何在Hibernate的Criteria API中可靠高效实现多列子查询？

针对你需要用Criteria API实现「获取每个会话最后一条消息」的需求，同时不能使用原生查询、HQL/JPQL，这里有一个可靠且高效的替代方案——将多列IN子查询转换为EXISTS关联子查询，完美规避Criteria不支持多列IN的限制，同时解决拼接列不可靠、聚合找ID性能差的问题。

方案思路 #

你的目标SQL是通过多列(conversation_id, created_at)匹配子查询的分组结果，我们可以把这个逻辑转换成存在性判断：对于每条消息，检查它是否是对应会话中创建时间最晚的那条。对应的等价SQL如下：

SELECT m.* FROM message m 
WHERE EXISTS (
    SELECT 1 FROM message m_sub
    WHERE m_sub.conversation_id = m.conversation_id
    GROUP BY m_sub.conversation_id
    HAVING MAX(m_sub.created_at) = m.created_at
)

这个SQL和你原来的需求完全等价，而且可以直接用Criteria API实现。

Criteria API 代码实现 #

下面是完整的代码示例，假设你使用JPA的标准Criteria API：

// 获取CriteriaBuilder和CriteriaQuery实例
CriteriaBuilder cb = entityManager.getCriteriaBuilder();
CriteriaQuery<Message> mainQuery = cb.createQuery(Message.class);
Root<Message> mainRoot = mainQuery.from(Message.class);

// 构建EXISTS子查询
Subquery<Boolean> existsSubquery = mainQuery.subquery(Boolean.class);
Root<Message> subRoot = existsSubquery.from(Message.class);

// 子查询逻辑：关联主查询的会话ID，分组后验证当前消息的创建时间是该组最大值
existsSubquery.select(cb.literal(true))
        // 关联主查询的conversation_id，确保子查询只处理当前消息所属的会话
        .where(cb.equal(subRoot.get("conversation_id"), mainRoot.get("conversation_id")))
        // 按会话ID分组
        .groupBy(subRoot.get("conversation_id"))
        // 筛选出该会话中创建时间最晚的消息
        .having(cb.equal(cb.max(subRoot.get("created_at")), mainRoot.get("created_at")));

// 主查询添加EXISTS条件
mainQuery.where(cb.exists(existsSubquery));

// 执行查询，得到每个会话的最后一条消息（如果同一时间有多条，会全部返回，和原SQL行为一致）
List<Message> lastMessages = entityManager.createQuery(mainQuery).getResultList();

方案优势 #

• 可靠性高：不需要拼接conversation_id和created_at，完全避免了拼接字符串可能出现的歧义（比如ID为123和时间2024-01-01，与ID为12和时间32024-01-01拼接后结果相同的问题）。
• 性能优异：如果你的message表有(conversation_id, created_at)的联合索引，子查询的分组和MAX计算会非常高效，主查询的EXISTS判断也能利用索引快速匹配，性能比聚合找ID的方法好很多。
• 纯Criteria API实现：完全符合要求，不需要依赖原生查询、HQL或JPQL，兼容所有支持JPA标准的ORM框架。

内容的提问来源于stack exchange，提问作者Alexander Makeev