亲宝软件园·资讯

展开

使用jpa聚合函数的问题

你怎么不笑了 人气:2

使用jpa聚合函数遇到的问题

spring data jpa 可以通过在接口中按照规定语法创建一个方法进行增删改查,可以快速开发。

但是一些特殊情况,如按条件查询统计,这个时候通过规定语法还是直接@Query写原生sql都不能解决,因为这个统计的条件不好添加。

百度之后,可以使用EntityManager来解决。

EntityManager 对象在一组实体类与底层数据源之间进行 O/R 映射的管理。它可以用来管理和更新 Entity Bean, 根椐主键查找 Entity Bean, 还可以通过JPQL语句增删改查实体。

首先通过@PersistenceContext注入EntityManager

@PersistenceContext
private EntityManager em;

使用如下

BigDecimal totals = new BigDecimal(0);
StringBuffer sql= new StringBuffer("SELECT SUM(total) FROM Order WHERE state = 1");
if (StringUtils.isNotBlank(order.getOrderNo())) {
    sql.append(" AND orderNo = ").append(order.getOrderNo());
}
if (StringUtils.isNotBlank(orderTime)) {
    sql.append(" AND orderPayTime LIKE '").append(orderTime).append("%'");
}
if (order.getCustomer() != null) {
    //where条件后面是对象信息,不能直接传入对象拼成sql,会报错.
    sql.append(" AND customer = :customer");
}
Query query = em.createQuery(sql.toString());
if (order.getCustomer() != null) {
    //通过setParameter,给参数赋值.
    query.setParameter("customer", order.getCustomer());
}
Object result = query.getSingleResult();
if (result != null) {
    totals = new BigDecimal(result.toString());
}
return totals;

jpa聚合函数(适用于联合查询)

最近帮老同事解决一个问题,场景是这样的,查询条件比较多,也就是我们说的联合查询,比如下面的,时间可以选不同的,状态和来源也可以选不同,而且可选可不选

如果这个时候写sql,是不是要各种条件判断,各种纠结,各种难写,各种sql,这个时候大家一般都想到了springdata的jpa貌似很好用,可以直接拼接sql,但是怎么拼接呢,又怎么支持非表字段的展示呢,比如表中一个字段 aaa 好展示,但是查总和sum(aaa) ,怎么把这个作为一个字段展示呢。

不罗嗦了,直接上代码

以下语句对应的sql大概是

select count(***) from *** where *** group by ***
private List<Tuple> getCountByStatusOrSource(Integer status, Integer source, Integer userId, String startTime, String endTime, Integer timeSlot, String type) throws Exception {
        CriteriaBuilder cb = entityManager.getCriteriaBuilder();
        Date sTime = new Date(), eTime = new Date();
        CriteriaQuery<Tuple> query = cb.createTupleQuery();
        Root<AAA> root = query.from(AAA.class);//AAA是对应数据库的类名,替换成自己的
        Path<Integer> statusPath = root.get("status");
        Path<String> statusNamePath = root.get("status").get("name");
        Path<Integer> sourcePath = root.get("source");
        Path<Date> operatorPath = root.get("operator");
 
        List<Predicate> predicateList = new ArrayList<>();
        if (source != null) {
            predicateList.add(
                cb.equal(sourcePath, source)
            );
        }
        if (userId != null) {
            predicateList.add(
                cb.equal(operatorPath, userId)
            );
        }
        Map timeMap = getChangedTime(startTime, endTime, timeSlot);//获取时间的方法,具体代码我就不沾了,自己写个就行了
        sTime = (Date) timeMap.get("sTime");
        eTime = (Date) timeMap.get("eTime");
       
            Expression<Date> startDateExpression = cb.literal(sTime);
            Expression<Date> endDateExpression = cb.literal(eTime);
            predicateList.add(
                cb.between(updateTimePath, startDateExpression, endDateExpression)
            );      
 
        Predicate[] predicates = new Predicate[predicateList.size()];
        predicates = predicateList.toArray(predicates);
        query.where(predicates);//where条件加上
        if ("status".equals(type)) {
            query.select(cb.tuple(statusPath, cb.count(root)));
            query.groupBy(statusPath);
        }
        
        //query.multiselect(statusPath, cb.count(root));//
        TypedQuery<Tuple> q = entityManager.createQuery(query);
        List<Tuple> result = q.getResultList();
        return result;
    }

这个sql,我们分别查到了两个值 status 和数量,那么怎么获取呢

循环一下那个获取的list

Tuple tuple = list.get(i);
(long)tuple.get(0)获取的是数量
(Integer)tuple.get(1)获取的是状态id

到这是否全部清晰了呢,这个例子也适用sum,max等其他聚合函数

以上为个人经验,希望能给大家一个参考,也希望大家多多支持。

加载全部内容

相关教程
猜你喜欢
用户评论