Jpa Specification如何实现and和or同时使用查询


Posted in Java/Android onNovember 23, 2021

同时使用and和or的查询

UserServiceImpl 类,service实现类

import org.springframework.beans.BeanUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.jpa.domain.Specification;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
 
import javax.persistence.criteria.CriteriaBuilder;
import javax.persistence.criteria.CriteriaQuery;
import javax.persistence.criteria.Predicate;
import javax.persistence.criteria.Root;
import java.util.ArrayList;
import java.util.List;
 
@Service
@Transactional
public class UserServiceImpl implements UserService {
    @Autowired
    private RongUserRepository rongUserRepository;
     //FriendNumResult  自定的返回类型
    //FriendNumParam  自定义的封装参数的类型
    //RongUser  实体类型
    @Override
    public FriendNumResult friendNum(FriendNumParam friendNumParam) {
        FriendNumResult friendNumResult=new FriendNumResult();
 
        Specification<RongUser> specification = new Specification<RongUser>(){
 
            @Override
            public Predicate toPredicate(Root<RongUser> root, CriteriaQuery<?> criteriaQuery, CriteriaBuilder criteriaBuilder) {
                //封装and语句
                List<Predicate> listAnd = new ArrayList<Predicate>();
                //这里是hql,所以root.get(),方法里面必须是对应的实体属性
                listAnd.add(criteriaBuilder.equal(root.get("perLevel").as(Integer.class), friendNumParam.getPerLevel()));
                Predicate[] array_and=new Predicate[listAnd.size()];
                Predicate Pre_And = criteriaBuilder.and(listAnd.toArray(array_and));
 
                //封装or语句
                List<Predicate> listOr = new ArrayList<Predicate>();
                listOr.add(criteriaBuilder.equal(root.get("fId").as(Integer.class), friendNumParam.getUid()));
                listOr.add(criteriaBuilder.equal(root.get("fId2").as(Integer.class), friendNumParam.getUid()));
                Predicate[] arrayOr = new Predicate[listOr.size()];
                Predicate Pre_Or = criteriaBuilder.or(listOr.toArray(arrayOr));
 
                return criteriaQuery.where(Pre_And,Pre_Or).getRestriction();
                //单独使用  and 或者  or 时 返回
                //return criteriaBuilder.and(list.toArray());
            }
        };
        long num=this.rongUserRepository.count(specification);
        friendNumResult.setFriendNum(Integer.valueOf((int)num));
        return friendNumResult;
    }
}

RongUserRepository接口

import org.springframework.data.jpa.repository.JpaRepository;
import org.springframework.data.jpa.repository.JpaSpecificationExecutor;
//RongUser  自己的实体类型
public interface RongUserRepository extends JpaRepository<RongUser,Integer> , JpaSpecificationExecutor<RongUser> {
}

注意:使用Specification之前,RongUserRepository接口必须实现JpaSpecificationExecutor<RongUser>,RongUser对应表的实体类

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结合使用,教程到这里就结束了。以上为个人经验,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Java/Android 相关文章推荐
Springboot如何使用logback实现多环境配置?
Jun 16 Java/Android
Java基础之详解HashSet的使用方法
Jun 30 Java/Android
dubbo服务整合zipkin详解
Jul 26 Java/Android
Java tomcat手动配置servlet详解
Nov 27 Java/Android
SpringDataJPA实体类关系映射配置方式
Dec 06 Java/Android
Java如何实现通过键盘输入一个数组
Feb 15 Java/Android
Java 超详细讲解hashCode方法
Apr 07 Java/Android
零基础学java之带返回值的方法的定义和调用
Apr 10 Java/Android
Springboot-cli 开发脚手架,权限认证,附demo演示
Apr 28 Java/Android
Java 死锁解决方案
May 11 Java/Android
Java中生成微信小程序太阳码的实现方案
Jun 01 Java/Android
Java Spring Boot请求方式与请求映射过程分析
Jun 25 Java/Android
解析mybatis-plus中的resultMap简单使用
Nov 23 #Java/Android
JPA 通过Specification如何实现复杂查询
Java使用JMeter进行高并发测试
Java 在线考试云平台的实现
OpenCV实现反阈值二值化
聊聊SpringBoot自动装配的魔力
Nov 17 #Java/Android
Springboot如何同时装配两个相同类型数据库
Nov 17 #Java/Android
You might like
德生1994机评
2021/03/02 无线电
php实现压缩合并js的方法【附demo源码下载】
2016/09/22 PHP
tp5(thinkPHP5框架)captcha验证码配置及验证操作示例
2019/05/28 PHP
Javascript学习笔记二 之 变量
2010/12/15 Javascript
JS 获取select(多选下拉)中所选值的示例代码
2013/08/02 Javascript
什么是cookie?js手动创建和存储cookie
2014/05/27 Javascript
jQuery通过Ajax返回JSON数据
2015/04/28 Javascript
javascript实现捕捉键盘上按下的键
2015/05/05 Javascript
学习JavaScript设计模式(多态)
2015/11/25 Javascript
jquery 动态增加,减少input表单的简单方法(必看)
2016/10/12 Javascript
JS及JQuery对Html内容编码,Html转义
2017/02/17 Javascript
JavaScript实现简单精致的图片左右无缝滚动效果
2017/03/16 Javascript
利用nvm管理多个版本的node.js与npm详解
2017/11/02 Javascript
微信小程序实现手指触摸画板
2018/07/09 Javascript
node中的session的具体使用
2018/09/14 Javascript
vue项目使用微信公众号支付总结及遇到的坑
2018/10/23 Javascript
详解react阻止无效重渲染的多种方式
2018/12/11 Javascript
Vue指令指令大全
2019/02/09 Javascript
vue element自定义表单验证请求后端接口验证
2019/12/11 Javascript
jQuery操作元素的内容和样式完整实例分析
2020/01/10 jQuery
python写xml文件的操作实例
2014/10/05 Python
Python中几个比较常见的名词解释
2015/07/04 Python
Python实现图片转字符画的示例
2017/08/22 Python
python微信好友数据分析详解
2018/11/19 Python
python SVD压缩图像的实现代码
2019/11/05 Python
Python实现括号匹配方法详解
2020/02/10 Python
python离线安装外部依赖包的实现
2020/02/13 Python
使用python的turtle函数绘制一个滑稽表情
2020/02/28 Python
Python datetime 格式化 明天,昨天实例
2020/03/02 Python
Python生成器next方法和send方法区别详解
2020/05/30 Python
Python系统公网私网流量监控实现流程
2020/11/23 Python
使用HTML5 Canvas API中的clip()方法裁剪区域图像
2016/03/25 HTML / CSS
廉洁自律承诺书范文
2015/04/28 职场文书
幼儿园门卫安全责任书
2015/05/08 职场文书
什么是求职信?求职信应包含哪些内容?
2019/08/14 职场文书
各种货币符号快捷输入
2022/02/17 杂记