SpringBoot JPA 条件分页

太过爱你忘了你带给我的痛 2022-06-07 08:45 260阅读 0赞

一点点基础知识

public interface UserRepository extends JpaRepository<User, String> { 
        Page<User> findAll(Specification<User> specification, Pageable pageable);
    }

只有一个查询条件的分页处理

public Users findUserInfo(Integer page, Integer size, String userId) {
            if (StringUtils.isEmpty(userId)) {
                Pageable pageable = new PageRequest(page, size, Sort.Direction.ASC, "id");
                return getUsersFromPage(userRepository.findAll(pageable), page, size);
            } else {
                UserQuery userQuery = UserQuery.builder().userId(userId).build();
                Pageable pageable = new PageRequest(page, size, Sort.Direction.ASC, "id");
                Page<User> userInfoPage = userRepository.findAll(new Specification<User>() {
                    @Override
                    public Predicate toPredicate(Root<User> root,
                                                 CriteriaQuery<?> query, CriteriaBuilder criteriaBuilder) {
                        Predicate predicate = criteriaBuilder.equal(root.get("id").as(String.class), userQuery.getUserId());
                        return criteriaBuilder.and(predicate);
                    }
                }, pageable);
                return getUsersFromPage(userInfoPage, page, size);
            }
        }

多个查询条件的处理

if (!StringUtils.isEmpty(userId)) {
                        predicateUserName = criteriaBuilder.equal(root.get("userId").as(String.class), userId);
                    }
                    if (!StringUtils.isEmpty(result)) {
                        predicateResult = criteriaBuilder.equal(root.get("result").as(String.class), result);
                    }
                    if (!StringUtils.isEmpty(startTime) && !StringUtils.isEmpty(endTime)) {
                        predicateCreateTime = criteriaBuilder.between(root.get("createTime").as(String.class), startTime, endTime);
                    }
                    if (!StringUtils.isEmpty(startTime) && StringUtils.isEmpty(endTime)) {
                        predicateCreateTime = criteriaBuilder.greaterThanOrEqualTo(root.get("createTime").as(String.class), startTime);
                    }
                    if (StringUtils.isEmpty(startTime) && !StringUtils.isEmpty(endTime)) {
                        predicateCreateTime = criteriaBuilder.lessThanOrEqualTo(root.get("createTime").as(String.class), endTime);
                    }
                    return criteriaBuilder.and(predicateUserName, predicateResult, predicateCreateTime);