SpringDataJpa的Specification查询

jopen 8年前

Spring Data JPA支持JPA2.0的Criteria查询,相应的接口是JpaSpecificationExecutor。Criteria 查询:是一种类型安全和更面向对象的查询 。

这个接口基本是围绕着Specification接口来定义的, Specification接口中只定义了如下一个方法:

Predicate toPredicate(Root<T> root, CriteriaQuery<?> query, CriteriaBuilder cb); 

要理解这个方法,以及正确的使用它,就需要对JPA2.0的Criteria查询有一个足够的熟悉和理解,因为这个方法的参数和返回值都是JPA标准里面定义的对象。 

Criteria查询基本概念

Criteria 查询是以元模型的概念为基础的,元模型是为具体持久化单元的受管实体定义的,这些实体可以是实体类,嵌入类或者映射的父类。

CriteriaQuery接口:代表一个specific的顶层查询对象,它包含着查询的各个部分,比如:select 、from、where、group by、order by等注意:CriteriaQuery对象只对实体类型或嵌入式类型的Criteria查询起作用

Root接口:代表Criteria查询的根对象,Criteria查询的查询根定义了实体类型,能为将来导航获得想要的结果,它与SQL查询中的FROM子句类似

1:Root实例是类型化的,且定义了查询的FROM子句中能够出现的类型。

2:查询根实例能通过传入一个实体类型给 AbstractQuery.from方法获得。

3:Criteria查询,可以有多个查询根。 

4:AbstractQuery是CriteriaQuery 接口的父类,它提供得到查询根的方法。CriteriaBuilder接口:用来构建CritiaQuery的构建器对象Predicate:一个简单或复杂的谓词类型,其实就相当于条件或者是条件组合。 

Criteria查询基本对象的构建

1:通过EntityManager的getCriteriaBuilder或EntityManagerFactory的getCriteriaBuilder方法可以得到CriteriaBuilder对象2:通过调用CriteriaBuilder的createQuery或createTupleQuery方法可以获得CriteriaQuery的实例

3:通过调用CriteriaQuery的from方法可以获得Root实例过滤条件

    A:过滤条件会被应用到SQL语句的FROM子句中。在criteria 查询中,查询条件通过Predicate或Expression实例应用到CriteriaQuery对象上。

    B:这些条件使用 CriteriaQuery .where 方法应用到CriteriaQuery 对象上

    C:CriteriaBuilder也作为Predicate实例的工厂,通过调用CriteriaBuilder 的条件方( equal,notEqual, gt, ge,lt, le,between,like等)创建Predicate对象。

    D:复合的Predicate 语句可以使用CriteriaBuilder的and, or andnot 方法构建。 

        构建简单的Predicate示例:

            Predicate p1=cb.like(root.get(“name”).as(String.class), “%”+uqm.getName()+“%”);

            Predicate p2=cb.equal(root.get("uuid").as(Integer.class), uqm.getUuid());

            Predicate p3=cb.gt(root.get("age").as(Integer.class), uqm.getAge());

        构建组合的Predicate示例:

           Predicate p = cb.and(p3,cb.or(p1,p2)); 

        当然也可以形如前面动态拼接查询语句的方式,比如:

Specification<UserModel> spec = new Specification<UserModel>() {        public Predicate toPredicate(Root<UserModel> root,                CriteriaQuery<?> query, CriteriaBuilder cb) {            List<Predicate> list = new ArrayList<Predicate>();                            if(um.getName()!=null && um.getName().trim().length()>0){                list.add(cb.like(root.get("name").as(String.class), "%"+um.getName()+"%"));            }            if(um.getUuid()>0){                list.add(cb.equal(root.get("uuid").as(Integer.class), um.getUuid()));            }            Predicate[] p = new Predicate[list.size()];            return cb.and(list.toArray(p));        }    };

        也可以使用CriteriaQuery来得到最后的Predicate,示例如下:

Specification<UserModel> spec = new Specification<UserModel>() {        public Predicate toPredicate(Root<UserModel> root,                CriteriaQuery<?> query, CriteriaBuilder cb) {            Predicate p1 = cb.like(root.get("name").as(String.class), "%"+um.getName()+"%");            Predicate p2 = cb.equal(root.get("uuid").as(Integer.class), um.getUuid());            Predicate p3 = cb.gt(root.get("age").as(Integer.class), um.getAge());            //把Predicate应用到CriteriaQuery中去,因为还可以给CriteriaQuery添加其他的功能,比如排序、分组啥的            query.where(cb.and(p3,cb.or(p1,p2)));            //添加排序的功能            query.orderBy(cb.desc(root.get("uuid").as(Integer.class)));                        return query.getRestriction();        }    };

综上所述,我们可以用Specification的toPredicate方法构建更多更复杂的查询方法。甚至如果你想多表链接进行查询,首先在对象上用@onetoone、@onetomany、@manytoone等注解,然后利用root.jion()方法也能做到,还 可以设定链表方式等。具体可以查看相关jpa文档。

下面我们用两个示例代码来更深入的了解:

    1.复杂条件多表查询

//需要查询的对象  public class Qfjbxxdz {      @Id      @GeneratedValue(generator = "system-uuid")      @GenericGenerator(name = "system-uuid", strategy = "uuid.hex")      private String id;      @OneToOne      @JoinColumn(name = "qfjbxx")      private Qfjbxx qfjbxx; //关联表      private String fzcc;      private String fzccName;      @ManyToOne      @JoinColumn(name = "criminalInfo")      private CriminalInfo criminalInfo;//关联表      @Column(length=800)      private String bz;      //get/set......  }
//创建构造Specification的方法  //这里我传入两个条件参数因为与前段框架有关,你们写的时候具体自己业务自行决断  private Specification<Qfjbxxdz> getWhereClause(final JSONArray condetion,final JSONArray search) {          return new Specification<Qfjbxxdz>() {              @Override              public Predicate toPredicate(Root<Qfjbxxdz> root, CriteriaQuery<?> query, CriteriaBuilder cb) {                  List<Predicate> predicate = new ArrayList<>();                  Iterator<JSONObject> iterator = condetion.iterator();                  Predicate preP = null;                  while(iterator.hasNext()){                      JSONObject jsonObject = iterator.next();                      //注意:这里用的root.join 因为我们要用qfjbxx对象里的字段作为条件就必须这样做join方法有很多重载,使用的时候可以多根据自己业务决断                      Predicate p1 = cb.equal(root.join("qfjbxx").get("id").as(String.class),jsonObject.get("fzId").toString());                      Predicate p2 = cb.equal(root.get("fzcc").as(String.class),jsonObject.get("ccId").toString());                      if(preP!=null){                          preP = cb.or(preP,cb.and(p1,p2));                      }else{                          preP = cb.and(p1,p2);                      }                  }                  JSONObject jsonSearch=(JSONObject) search.get(0);                  Predicate p3=null;                  if(null!=jsonSearch.get("xm")&&jsonSearch.get("xm").toString().length()>0){                     p3=cb.like(root.join("criminalInfo").get("xm").as(String.class),"%"+jsonSearch.get("xm").toString()+"%");                  }                  Predicate p4=null;                  if(null!=jsonSearch.get("fzmc")&&jsonSearch.get("fzmc").toString().length()>0){                      p4=cb.like(root.join("qfjbxx").get("fzmc").as(String.class),"%"+jsonSearch.get("fzmc").toString()+"%");                  }                  Predicate preA;                  if(null!=p3&&null!=p4){                      Predicate  preS =cb.and(p3,p4);                      preA =cb.and(preP,preS);                  }else if(null==p3&&null!=p4){                      preA=cb.and(preP,p4);                  }else if(null!=p3&&null==p4){                      preA=cb.and(preP,p3);                  }else{                      preA=preP;                  }                  predicate.add(preA);                  Predicate[] pre = new Predicate[predicate.size()];                  query.where(predicate.toArray(pre));                  return query.getRestriction();              }          };      }
public void findByPage(SONArray condetion,JSONArray search) {      Pageable pageable = ....//构建分页对象       Specification<Qfjbxxdz> spec = getWhereClause(condetion,search)      //利用JpaSpecificationExecutor接口的分页查询方法:      //Page<T> findAll(Specification<T> spec, Pageable pageable);      //这就是上一个文章里我为什么要继承JpaSpecificationExecutor接口的原因      Page<Qfjbxxdz> page = qfjbxxdzRepository.findAll(spec, pageable );  }

    2.EntityManager的使用

@PersistenceContext  private EntityManager em;//注入entitymanager
//A.利用entitymanager构建Criteria查询  //其实和sql一样,就是换成了面向对象的方式。这种方式的好处就是可以防止sql拼写错误。当然这种方式也不是特别直观  public List<Qfsqmx> countQfsq(List<Qfsq> list) {      CriteriaBuilder cb = em.getCriteriaBuilder();      CriteriaQuery<Qfsqmx> q = cb.createQuery(Qfsqmx.class);      Root<Qfsqmx> root = q.from(Qfsqmx.class);      List<Predicate> predicate = new ArrayList<>();      //in条件拼接      In<String> in = cb.in(root.get("qfsqid").as(String.class));      for (int i = 0; i < list.size(); i++) {          in.value(list.get(i).getId());      }     //设定select字段      q.multiselect(          root.get("qfmc"),          root.get("qfcc"),          root.get("jldw"),          cb.sum(root.get("sl").as(Integer.class))      );      predicate.add(in);      Predicate[] pre = new Predicate[predicate.size()];      //设定where条件      q.where(predicate.toArray(pre));      //设定groupby条件      q.groupBy(          root.get("qflxid").as(String.class),          root.get("qfccid").as(String.class),          root.get("qfmc").as(String.class),          root.get("qfcc").as(String.class),          root.get("jldw").as(String.class)      );      //设定orderby条件      q.orderBy(cb.asc(root.get("qfmc").as(String.class)));      List<Qfsqmx> rs = em.createQuery(q).getResultList();      return rs;  }
//B.利用em进行本地sql分页查询  public JsonReader selectSrzc(QueryParams queryParams,String qsrq, String jzrq, String zmlx, String zlx, String zhlx, String ssjq, String zfxm) {      StringBuffer hql = new StringBuffer();       hql.append("SELECT to_char(srzc.sj,'yyyy-mm-dd'),ZFZH.ZHBH,ZF.xm,zf.gyjq,decode(srzc.zhlx,'ykt','***','grzh','***'),srzc.zmlx,srzc.zlx,srzc.jsr,srzc.srzc,ye.ye,srzc.sjly FROM ");      hql.append("(SELECT * FROM (");      hql.append("SELECT CRIMINALSACCOUNT AS zhid,RZSJ AS sj,ZHLX,RZLX AS zmlx,RZZLX AS zlx,RZR AS jsr,RZJE AS srzc,'**' AS sjly FROM DZ_PERSON_INCOME WHERE SHZT = '**' ");      hql.append("UNION ALL ");      hql.append("SELECT CRIMINALSACCOUNT AS zhid,CZSJ AS sj,ZHLX,CZLX AS zmlx,CZZLX AS zlx,CZR AS jsr,CZJE AS srzc,'***' AS sjly FROM DZ_PERSON_OUTGOING ");      hql.append(")) srzc ");      hql.append("LEFT JOIN SW_DZ_ZFZH zfzh ON srzc.zhid = zfzh.id ");      hql.append("LEFT JOIN BS_CRIMINALINFO zf ON zfzh.zfid = zf.id ");      hql.append("LEFT JOIN (SELECT * FROM SW_DZ_LSYEJL WHERE TO_CHAR(rq,'yyyy-mm-dd') = ?1) ye ON zf.id = ye.zfid ");      hql.append("WHERE TO_CHAR(sj,'yyyy-mm-dd') >= ?2 AND TO_CHAR(sj,'yyyy-mm-dd') <= ?3 ");      if(StringUtils.isNotEmpty(zmlx)){          hql.append("AND zmlx = '"+zmlx+"'");      }          //....省略      if(StringUtils.isNotEmpty(zfxm)){          String[] list = zfxm.toString().split("[, ]");          if(list.length>1){              hql.append("AND (xm like '%"+list[0]+"%' or xm like '%"+list[1]+"%' or zhbh like '%"+list[1]+"%' or zhbh like '%"+list[1]+"%')");          }else{              hql.append("AND xm like '%"+list[0]+"%' or zhbh like '%"+list[0]+"%'");          }      }      if(StringUtils.isNotEmpty(queryParams.getSidx())){          switch(queryParams.getSidx()){              case "rq":              hql.append(" order by sj ");              break;              case ...//此处省略          }          hql.append(queryParams.getSord());      }else{          hql.append(" order by xm,sj desc");      }      //统计所有数据条数      String countSql = "select count(*) from ("+hql.toString()+")";      Query countQuery = em.createNativeQuery(countSql);      countQuery.setParameter(1, jzrq);      countQuery.setParameter(2, qsrq);      countQuery.setParameter(3, jzrq);      BigDecimal obj = (BigDecimal) countQuery.getSingleResult();      //得到页数      int totalPage = (int) Math.ceil(obj.divide(new BigDecimal(queryParams.getRows())).doubleValue());      //分页查询      Query query = em.createNativeQuery(hql.toString());      int firstIndex = ((queryParams.getPage()-1) * queryParams.getRows());      query.setFirstResult(firstIndex);      query.setMaxResults(queryParams.getRows());      query.setParameter(1, jzrq);      query.setParameter(2, qsrq);      query.setParameter(3, jzrq);      List<Object> list = query.getResultList();      //设定输出参数      JsonReader jsonReader = new JsonReader();      jsonReader.setPage(queryParams.getPage())      .setTotal(totalPage)      .setRecords(obj.intValue());      jsonReader.setRows(list);      return jsonReader;  }

注意:在使用entitymanager进行自定义sql查询的时候,如果遇到in条件你这样设定参数将是错误的。

query.setParameter(1, jzrq);

javax.persistence.Query会在参数外围用引号包围起来 然后你的查询就会变成 in ( 'xx,xx,xx')而不是你期望的in ('xx','xx','xx').

对于这种情况我们转换思路,特殊处理:

//通过em获取hibernate的session  String sql = "SELECT nf,xx,xx FROM xx WHERE nf in (:nf)";  Session session = em.unwrap(Session.class);  //获取org.hibernate.Query  Query query = session.createSQLQuery(sql+buffer);  if(qjFields.contains(",")){      String[] qjs =  qjFields.split(",");      query.setParameterList("nf", qjs);  }else{      query.setParameter("nf",qjFields);  }  //获取数据  List<Object> rs = query.list();

以上就是关于Specification查询的相关应用,其实 还有很多用法,这就看各位大侠们自行悟道了。欢迎喜欢springDataJpa的同道中人相互指导交流!

来自: http://my.oschina.net/u/2434456/blog/596938