JPA分组查询,求和,自定义查询字段,自定义VO承接
一.JPA使用过程中的问题
JPA常用的查询方法,用实体对应的repository的执行find方法,查询都是实体的全部字段或者其中的单个字段。
如果对一个字段进行分组查询时,会出现问题,这里分享一个自定义查询方法。
二.解决问题
一.分组查询
表数据
JPA对应实体
结果集VO
进行分组查询
-
-
-
-
public class JPAController {
-
-
-
private EntityManager entityManager;
-
-
-
-
-
public List<StudentVO> jpa(){
-
CriteriaBuilder cb = entityManager.getCriteriaBuilder();
-
CriteriaQuery<Tuple> cq = cb.createTupleQuery();
-
Root<Student> root = cq.from(Student.class);//具体实体的Root
-
root.alias("o");//别名
-
cq.multiselect(root.get("name").alias("name"),root.get("age").alias("age"));
-
-
Predicate where = cb.conjunction();
-
//增加查询条件
-
cq.where(where).groupBy(root.get("name"),root.get("age"));
-
List<Tuple> tuples = entityManager.createQuery(cq).getResultList();
-
//结果集
-
List<StudentVO> result = new ArrayList<>();
-
for(Tuple tuple : tuples){
-
result.add(new StudentVO(tuple.get(0).toString(),tuple.get(1).toString()));
-
}
-
return result;
-
}
-
}
结果集截图(swagger文档接口调试)
根据自己的情况,写对应的JPA过滤代码。
业务场景:
-
/**
-
* 查询月份集合(JPA分组)
-
* @param type
-
* @return
-
*/
-
public Integer countNum(MonthlyParamVO param,String type,String isF ,Collection<Integer> functionSonList,Collection<Integer> orgIds){
-
//初始化结果集
-
List<String> monthList = new ArrayList<String>();
-
//初始化JPA查询
-
CriteriaBuilder cb = entityManager.getCriteriaBuilder();
-
CriteriaQuery<Tuple> cq = cb.createTupleQuery();
-
Root<AttendanceMonthlyReportEntity> root = cq.from(AttendanceMonthlyReportEntity.class);
-
//加条件
-
if("missDays".equals(type)){
-
cq.multiselect(cb.sum(root.get("missDays")));
-
}else if("absentDays".equals(type)){
-
cq.multiselect(cb.sum(root.get("absentDays")));
-
}else if("lateEarly".equals(type)){
-
cq.multiselect(cb.sum(root.get("lateEarly")));
-
}else if("notPunched".equals(type)){
-
cq.multiselect(cb.sum(root.get("notPunched")));
-
}
-
//查询条件where
-
List<Predicate> predicates = new ArrayList<>();
-
if(StringUtils.isNotBlank(param.getZhName())){
-
predicates.add(cb.equal(root.get("zhName"),param.getZhName()));
-
}
-
if(null!=param.getOrgIds()&¶m.getOrgIds().size()!=0){
-
predicates.add(root.get("orgId").in(orgIds));
-
}
-
if(StringUtils.isNotBlank(param.getYear())&&StringUtils.isBlank(param.getMonth())){
-
predicates.add(cb.equal(root.get("attendanceMonth"),param.getYear()+"%"));
-
}
-
if(StringUtils.isNotBlank(param.getZhName())&&StringUtils.isNotBlank(param.getMonth())){
-
predicates.add(cb.equal(root.get("attendanceMonth"),param.getYear()+"-"+param.getMonth()));
-
}
-
if("Y".equals(isF)){
-
//加入需求限制
-
predicates.add(cb.or(root.get("orgId").in(functionSonList),cb.like(root.get("positionLevel"),"F")));
-
}
-
cq.where(predicates.toArray(new Predicate[predicates.size()]));
-
-
List<Tuple> tuples = entityManager.createQuery(cq).getResultList();
-
//求和
-
Integer count = 0;
-
for(Tuple tuple : tuples){
-
if(tuple.get(0) instanceof Float){
-
count = ((Float) tuple.get(0)).intValue();
-
}else if(tuple.get(0) instanceof Integer){
-
count = (Integer) tuple.get(0);
-
}else if(tuple.get(0) instanceof Double){
-
count = ((Double) tuple.get(0)).intValue();
-
}
-
}
-
return count;
-
}