From e8a53c081980ce1a5a9e1a36c822edab15ea42e0 Mon Sep 17 00:00:00 2001 From: wayne <wayne8692wayne8692@gmail.com> Date: 星期四, 02 十二月 2021 14:54:05 +0800 Subject: [PATCH] [update] 標記為已聯繫時紀錄已聯繫時間 --- pamapi/src/main/java/com/pollex/pam/service/ConsultantQuerySpec.java | 95 +++++++++++++++++++++++++++++++++-------------- 1 files changed, 67 insertions(+), 28 deletions(-) diff --git a/pamapi/src/main/java/com/pollex/pam/service/ConsultantQuerySpec.java b/pamapi/src/main/java/com/pollex/pam/service/ConsultantQuerySpec.java index bfefa21..dda2ade 100644 --- a/pamapi/src/main/java/com/pollex/pam/service/ConsultantQuerySpec.java +++ b/pamapi/src/main/java/com/pollex/pam/service/ConsultantQuerySpec.java @@ -2,15 +2,14 @@ import com.pollex.pam.domain.Consultant; import com.pollex.pam.service.dto.FastQueryConsultantParam; +import com.pollex.pam.service.dto.StrictQueryConsultantParam; 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 java.util.HashSet; -import java.util.Set; +import javax.persistence.criteria.*; +import java.util.*; + +import static com.pollex.pam.consts.SeniorityQueryConst.*; public class ConsultantQuerySpec { private ConsultantQuerySpec() { @@ -25,35 +24,75 @@ public Predicate toPredicate(Root<Consultant> root, CriteriaQuery<?> query, CriteriaBuilder criteriaBuilder) { Set<Predicate> set = new HashSet<>(); - if(!StringUtils.isEmpty(param.getCommunicationStyle())) { - set.add(criteriaBuilder.like(root.get("concept"), "%" + param.getCommunicationStyle())); + param.getCommunicationStyles().forEach((communicationStyle) -> { + set.add(criteriaBuilder.like(root.get("communicationStyle"), "%" + communicationStyle + "%")); + }); + + if(Objects.nonNull(param.getGender())) { + set.add(criteriaBuilder.equal(root.get("gender"), param.getGender())); } - set.add(criteriaBuilder.equal(root.get("gender"), param.getGender())); - 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.greaterThanOrEqualTo(root.get("avgScore"), param.getAvgScore().intValue())); + } Predicate[] predicates = new Predicate[set.size()]; predicates = set.toArray(predicates); - return criteriaBuilder.or(predicates); + return criteriaBuilder.and(predicates); } }; } -// public static Specification<Consultant> getStrictQuerySpec(StrictQueryConsultantParam param) { -// return new Specification<Consultant>() { -// @Override -// public Predicate toPredicate(Root<Consultant> root, CriteriaQuery<?> query, CriteriaBuilder criteriaBuilder) { -// Set<Predicate> set = new HashSet<>(); -// -// set.add(criteriaBuilder.equal(root.get("gender"), param.getGender())); -// set.add(criteriaBuilder.between(root.get("avgScore"), param.getAvgScore().intValue() - 1, param.getAvgScore().intValue() + 1)); -// set.add(criteriaBuilder.like(root.get("serve_area"),"%"+param.getArea()+"%")); -// set.add(root.get("").in(param.getRequirements())) -// -// Predicate[] predicates = new Predicate[set.size()]; -// predicates = set.toArray(predicates); -// return criteriaBuilder.or(predicates); -// } -// }; -// } + public static Specification<Consultant> getStrictQuerySpec(StrictQueryConsultantParam param) { + return new Specification<Consultant>() { + @SuppressWarnings("FieldNameHidesFieldInSuperclass") + private static final long serialVersionUID = 664452413090316174L; + + @Override + public Predicate toPredicate(Root<Consultant> root, CriteriaQuery<?> query, CriteriaBuilder criteriaBuilder) { + Set<Predicate> set = new HashSet<>(); + + if(Objects.nonNull(param.getGender())) { + set.add(criteriaBuilder.equal(root.get("gender"), param.getGender())); + } + + if(Objects.nonNull(param.getAvgScore()) && param.getAvgScore().intValue() != 0) { + set.add(criteriaBuilder.greaterThanOrEqualTo(root.get("avgScore"), param.getAvgScore().intValue())); + } + + if(StringUtils.isNotEmpty(param.getSeniority()) && !StringUtils.equals(UNLIMITED, param.getSeniority())) { + if(YOUNG.equals(param.getSeniority())) { + set.add(criteriaBuilder.lessThanOrEqualTo(root.get("seniorityYear"), 5)); + } + else if(SENIOR.equals(param.getSeniority())) { + set.add(criteriaBuilder.greaterThan(root.get("seniorityYear"), 5)); + } + } + + if(StringUtils.isNotEmpty(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(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(StringUtils.isNotEmpty(param.getOtherPopularTags())) { + allPopularTags.add(param.getOtherPopularTags()); + } + allPopularTags.forEach(popularTag -> set.add(criteriaBuilder.like(root.get("concept"), "%" + popularTag + "%"))); + + Predicate[] predicates = new Predicate[set.size()]; + predicates = set.toArray(predicates); + return criteriaBuilder.and(predicates); + } + }; + } } -- Gitblit v1.8.0