Spring-Data-JPA中使用Specification实现动态查询

      最近项目技术选型db框架选择了使用JPA,刚开始时,使用jpa进行一些单表简单的查询非常轻松,大家写的不亦乐乎,后来在遇到多条件动态查询的业务场景时,发现现有的JpaRepository提供的方法和自己写@Query已经满足了不了需求,难不成要对所有的条件和字段进行判断,再写很多个dao方法?后面查到jpa提供了围绕Specification这个类的一系列类,来用于实现动态查询。

       首先,毫无疑问需要定义一个dao接口,这个接口除了继承JpaRepository之外,还需要继承JpaSpecificationExecutor。

public interface FileInfoDao extends JpaRepository, JpaSpecificationExecutor{
}

       这是我用于查询文件信息的dao,我们可以看到JpaSpecificationExecutor有以下这些方法:

       T findOne(Specification spec);
	List findAll(Specification spec);
	Page findAll(Specification spec, Pageable pageable);
	List findAll(Specification spec, Sort sort);
	long count(Specification spec);

findOne(spec):根据条件查询获取一条数据;

findAll(spec) :根据条件查询获取全部数据;

findAll(specification, pageable) : 根据条件分页查询数据,pageable设定页码、一页数据量,同时返回的是Page类对象,可以通过getContent()方法拿到List集合数据;

findAll(specification, sort) : 根据条件查询并返回排序后的数据;

count(spec) : 获取满足当前条件查询的数据总数;

       现在光定义了dao没用,人家就是需要你往方法里传Specification的对象,Specification是一个接口,肯定没法直接获取对象。

public interface Specification {
	Predicate toPredicate(Root root, CriteriaQuery query, CriteriaBuilder cb);
}

        所以我自定义了一个类实现类这个接口。

public class SinoSpecification implements Specification {
	
	private TableQueryParam param;
	
	private Class clazz;
	
	public SinoSpecification(TableQueryParam param, Class clazz) {
		this.param = param;
		this.clazz = clazz;
	}
	
	@Override
	public Predicate toPredicate(Root root, CriteriaQuery query, CriteriaBuilder cb) {
		Field[] fields =  clazz.getDeclaredFields();
		Map conditions = param.getCondition();
		List predicates = new ArrayList<>();
		for (int i = 0; i < fields.length; i++) {
			fields[i].setAccessible(true);
			String name = fields[i].getName();
			if(conditions.containsKey(name)) {
				if(ObjectUtils.isEmpty(conditions.get(name))) {
					continue;
				}
				predicates.add(cb.like(root.get(name), "%"+conditions.get(name)+"%"));
			}
		}
		return cb.and(predicates.toArray(new Predicate[predicates.size()]));
	}

}
        TableQueryParam是我自定义的用于封装查询条件的类,这个类有一个condition属性是条件集合。在这个类中,我是通过反射获取实体类的字段属性,再去和TableQueryParam中的conditon里的条件进行比对,如果条件中包含,则将这个条件以及查询的值一起构造加入进Predicate,cb.like(arg1, arg2)这个方法,就代表模糊查询,sql中的Like,arg1参数代表要查询的字段,arg2代表查询的字段的值,转换成sql就是 arg1 LIKE arg2。当然,我这个自定义的Specification主要是用于模糊查询的,而它不仅仅支持模糊,还可以通过CriteriaBuilder里的方法来实现各种组合查询,例如cb.equal()进行等于查询,cb.greaterThan()、cb.lessThan()来实现大于小于,等等,可以说非常丰富。

        定义好Specification之后,就可以调用前面说的findAll方法,将自定义的specification对象放进去即可。

你可能感兴趣的:(Web随笔,JPA)