| | |
| | | package com.pollex.pam.service; |
| | | |
| | | import com.pollex.pam.domain.Consultant; |
| | | import com.pollex.pam.enums.ConsultantStatusEnum; |
| | | import com.pollex.pam.service.dto.FastQueryConsultantParam; |
| | | import com.pollex.pam.service.dto.StrictQueryConsultantParam; |
| | | import org.apache.commons.lang3.StringUtils; |
| | |
| | | return new Specification<Consultant>() { |
| | | @SuppressWarnings("FieldNameHidesFieldInSuperclass") |
| | | private static final long serialVersionUID = 6644524130903161741L; |
| | | private static final float QUERY_AVG_SCORE_OFFSET = 0.5f; |
| | | |
| | | @Override |
| | | public Predicate toPredicate(Root<Consultant> root, CriteriaQuery<?> query, CriteriaBuilder criteriaBuilder) { |
| | | Set<Predicate> set = new HashSet<>(); |
| | | |
| | | set.add(criteriaBuilder.equal(root.get("status"), ConsultantStatusEnum.AVAILABLE)); |
| | | |
| | | param.getCommunicationStyles().forEach((communicationStyle) -> { |
| | | set.add(criteriaBuilder.like(root.get("communicationStyle"), "%" + communicationStyle + "%")); |
| | |
| | | } |
| | | |
| | | if(Objects.nonNull(param.getAvgScore()) && param.getAvgScore().intValue() != 0) { |
| | | set.add(criteriaBuilder.greaterThanOrEqualTo(root.get("avgScore"), param.getAvgScore().intValue())); |
| | | set.add(criteriaBuilder.greaterThanOrEqualTo(root.get("avgScore"), param.getAvgScore().intValue() - QUERY_AVG_SCORE_OFFSET)); |
| | | } |
| | | |
| | | if(StringUtils.isNotEmpty(param.getSeniority()) && !StringUtils.equals(UNLIMITED, param.getSeniority())) { |
| | |
| | | public Predicate toPredicate(Root<Consultant> root, CriteriaQuery<?> query, CriteriaBuilder criteriaBuilder) { |
| | | Set<Predicate> set = new HashSet<>(); |
| | | |
| | | set.add(criteriaBuilder.equal(root.get("status"), ConsultantStatusEnum.AVAILABLE)); |
| | | |
| | | if(Objects.nonNull(param.getGender())) { |
| | | set.add(criteriaBuilder.equal(root.get("gender"), param.getGender())); |
| | | } |
| | |
| | | if(StringUtils.isNotEmpty(param.getArea())) { |
| | | set.add(criteriaBuilder.like(root.get("serveArea"), "%" + param.getArea() + "%")); |
| | | } |
| | | |
| | | |
| | | Predicate[] predicates = new Predicate[set.size()]; |
| | | predicates = set.toArray(predicates); |
| | | return criteriaBuilder.and(predicates); |