JPA 动态查询之AND、OR结合使用
作者:互联网
引子
现在,我负责开发的项目中,使用JPA作为ORM框架。有了JPA,一行SQL都没写过。在昨天,有一个新的需求,需要进行动态查询,这个简单。但是有一个地方需要AND、OR结合使用,这里,我将记录下我的理解与写法,希望能帮助到大家。
问题描述
需要根据条件进行动态查询,实现一条类似下文的语句:
SELECT *
FROM table
WHERE 1 = 1
if (a == 1)
AND table.column1 = a
if (b != null)
AND table.column2 = b
if (cList != null && cList.size() > 0)
AND table.column3 IN cList
if (d == 2 || dd == 2)
AND (table.column4 = d OR table.column5 = dd)
上面是几行伪代码。意思是,几个条件之间是AND连接,但是其中的部分条件,是使用OR连接的。
在我们的实际项目中,这个场景也是很常见的。这里,我将分享下具体的写法。以我们项目中的例子为例。
代码示例
JPA的动态查询,这里我们使用的方式是:实现 Specification 接口,自定义动态查询逻辑。这也是我个人比较推荐的方式。JPA的使用、Specification 接口基础知识这里我就不讲了。有兴趣的朋友可以查阅官方文档学习。
下面,我们首先定义好我们的数据库实体:
@Data
@Entity
@Table(name = "user")
public class User {
@Id
@GeneratedValue(strategy = GenerationType.IDENTITY)
private Integer id;
/**
* 用户名
*/
private String username;
/**
* 年龄
*/
private Integer age;
/**
* 生日
*/
private Date birthDay;
/**
* 删除标识; 0 - 未删除,1 - 已删除
*/
private Integer deleteFlag;
}
然后定义好DAO层接口:
@Repository
public interface UserDAO extends JpaRepository<User, Long> {
/**
* 其实,这个功能一般用作 list 接口使用,一般结合分页查询使用。这里,我不做介绍,看情况要不要后期加上教程
*/
List<User> findAll(Specification<User> querySpec);
}
下面是前端传过来的动态查询的参数对象:
@Data
public class UserDTO {
/**
* 用户名,用于模糊搜索
*/
private String username;
/**
* 用户ID,用于 In 查询
*/
private List<String> userIdList;
/**
* 用户年龄,用于 OR In 查询
*/
private List<Integer> ageList;
/**
* 生日,开始
*/
@JsonFormat(pattern = "yyyy-MM-dd", locale = "zh", timezone = "GMT+8")
private Date birthDayBegin;
/**
* 生日,结束
*/
@JsonFormat(pattern = "yyyy-MM-dd", locale = "zh", timezone = "GMT+8")
private Date birthDayEnd;
}
然后,重要的地方来了,我们实现 Specification 接口,定义查询逻辑:
在实际代码操作中,我会将这部分逻辑抽离为一个单独的方法,使用lambda表达式完成,其实也就是匿名内部类。
private Specification<User> getListSpec(UserDTO userDTO) {
return (root, criteriaQuery, criteriaBuilder) -> {
List<Predicate> predicateList = new ArrayList<>();
// 未删除标识,只查询未删除的数据
predicateList.add(criteriaBuilder.equal(root.get("deleteFlag"), 0));
// 根据 用户名 或 年龄List 查询
List<Predicate> usernameOrAgePredicate = new ArrayList<>();
String username = userDTO.getUsername();
if (!StringUtils.isEmpty(username)) {
// 用户名这里,用模糊匹配
usernameOrAgePredicate.add(criteriaBuilder.like(root.get("username"), "%" + username + "%"));
}
List<Integer> ageList = userDTO.getAgeList();
if (!CollectionUtils.isEmpty(ageList)) {
// 下面是一个 IN查询
CriteriaBuilder.In<Integer> in = criteriaBuilder.in(root.get("age"));
ageList.forEach(in::value);
usernameOrAgePredicate.add(in);
}
/* 下面这一行代码很重要。
* criteriaBuilder.or(Predicate... restrictions) 接收多个Predicate,可变参数;
* 这多个 Predicate条件之间,是使用OR连接的;该方法最终返回 一个Predicate对象;
*/
predicateList.add(criteriaBuilder.or(usernameOrAgePredicate.toArray(new Predicate[0])));
// 用户ID List,IN 查询
List<Integer> userIdList = reqDTO.getUserIdList();
if (!CollectionUtils.isEmpty(userIdList)) {
CriteriaBuilder.In<Integer> in = criteriaBuilder.in(root.get("id"));
userIdList.forEach(in::value);
predicateList.add(in);
}
// 生日时间段查询
Date birthDayBegin = reqDTO.getBirthDayBegin();
Date birthDayEnd = reqDTO.getBirthDayEnd();
if (birthDayBegin != null && birthDayEnd != null) {
// DateUtils 是我自定义的一个工具类
Date begin = DateUtils.startOfDay(birthDayBegin);
Date end = DateUtils.endOfDay(birthDayEnd);
predicateList.add(criteriaBuilder.greaterThanOrEqualTo(root.get("birthDay"), begin));
predicateList.add(criteriaBuilder.lessThanOrEqualTo(root.get("birthDay"), end));
}
// 最终,使用AND 连接 多个 Predicate 查询条件
return criteriaBuilder.and(predicateList.toArray(new Predicate[0]));
};
}
这样,我们的动态查询部分就构建完毕了。具体怎么使用呢?如下:
Specification<User> querySpec = this.getListSpec(userDTO);
List<User> userList = userDAO.findAll(querySpec);
就这样,我们就执行了一次动态查询,并获取到了结果。
上面的动态查询,实际上等价于下面的伪代码:
SELECT *
FROM user
WHERE user.deleteFlag = 0
AND ( user.username like '%{username}%' OR user.age IN ageList )
AND user.id IN userIdList
AND user.birthDay > birthDayBegin AND user.birthDay < birthDayEnd ;
当前,需要对应值不为空,才会拼接相应的AND条件。
至此,JPA 动态查询之 AND OR结合使用,教程到这里就结束了。作者也是在不断学习之中,如果有什么地方没有描述清楚或有问题,欢迎大家指出,一起交流进步!
标签:username,JPA,List,private,查询,user,criteriaBuilder,动态 来源: https://blog.csdn.net/Zereao/article/details/98872092