| | |
| | | import org.apache.commons.lang3.StringUtils; |
| | | import org.springframework.data.jpa.domain.Specification; |
| | | |
| | | import javax.persistence.criteria.CriteriaBuilder; |
| | | import javax.persistence.criteria.CriteriaQuery; |
| | | import javax.persistence.criteria.Predicate; |
| | | import javax.persistence.criteria.Root; |
| | | import javax.persistence.criteria.*; |
| | | import java.util.*; |
| | | |
| | | public class ConsultantQuerySpec { |
| | |
| | | Set<Predicate> set = new HashSet<>(); |
| | | |
| | | param.getCommunicationStyles().forEach((communicationStyle) -> { |
| | | set.add(criteriaBuilder.like(root.get("concept"), "%" + param.getCommunicationStyles())); |
| | | set.add(criteriaBuilder.like(root.get("communicationStyle"), "%" + communicationStyle + "%")); |
| | | }); |
| | | |
| | | if(Objects.nonNull(param.getGender())) { |
| | | set.add(criteriaBuilder.equal(root.get("gender"), param.getGender())); |
| | | } |
| | | |
| | | if(Objects.nonNull(param.getAvgScore())) { |
| | | set.add(criteriaBuilder.between(root.get("avgScore"), param.getAvgScore().intValue() - 1, param.getAvgScore().intValue() + 1)); |
| | | if(Objects.nonNull(param.getAvgScore()) && param.getAvgScore().intValue() != 0) { |
| | | set.add(criteriaBuilder.equal(root.get("avgScore"), param.getAvgScore())); |
| | | } |
| | | |
| | | Predicate[] predicates = new Predicate[set.size()]; |
| | | predicates = set.toArray(predicates); |
| | | return criteriaBuilder.or(predicates); |
| | | return criteriaBuilder.and(predicates); |
| | | } |
| | | }; |
| | | } |
| | |
| | | set.add(criteriaBuilder.equal(root.get("gender"), param.getGender())); |
| | | } |
| | | |
| | | if(Objects.nonNull(param.getAvgScore())) { |
| | | set.add(criteriaBuilder.between(root.get("avgScore"), param.getAvgScore().intValue() - 1, param.getAvgScore().intValue() + 1)); |
| | | if(Objects.nonNull(param.getAvgScore()) && param.getAvgScore().intValue() != 0) { |
| | | set.add(criteriaBuilder.equal(root.get("avgScore"), param.getAvgScore())); |
| | | } |
| | | |
| | | if(StringUtils.isNotEmpty(param.getArea())) { |
| | | set.add(criteriaBuilder.like(root.get("serveArea"), "%" + param.getArea() + "%")); |
| | | Predicate predicate1 = criteriaBuilder.like(root.get("serveArea"), "%" + param.getArea() + "%"); |
| | | Predicate predicate2 = criteriaBuilder.like(root.get("serveArea"), "%全台%"); |
| | | |
| | | Predicate finalPredicate = criteriaBuilder.or(predicate1, predicate2); |
| | | set.add(finalPredicate); |
| | | } |
| | | |
| | | List<String> allRequirements = Objects.nonNull(param.getRequirements()) ? param.getRequirements() : new ArrayList<>(); |
| | | if(Objects.nonNull(param.getOtherRequirement())) { |
| | | if(StringUtils.isNotEmpty(param.getOtherPopularTags())) { |
| | | allRequirements.add(param.getOtherRequirement()); |
| | | } |
| | | allRequirements.forEach(requirement -> set.add(criteriaBuilder.like(root.get("expertise"), "%" + requirement + "%"))); |
| | | |
| | | List<String> allPopularTags = Objects.nonNull(param.getPopularTags()) ? param.getPopularTags() : new ArrayList<>(); |
| | | if(Objects.nonNull(param.getOtherPopularTags())) { |
| | | if(StringUtils.isNotEmpty(param.getOtherPopularTags())) { |
| | | allPopularTags.add(param.getOtherPopularTags()); |
| | | } |
| | | allPopularTags.forEach(popularTag -> set.add(criteriaBuilder.like(root.get("concept"), "%" + popularTag + "%"))); |