Feature: Added filter for doctor and patient in specification.

elwyncrestha · Jul 11, 2020 · 4de112d · 4de112d
1 parent 244b850
commit 4de112d
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/pemits-api/src/main/java/com/pemits/webcare/api/doctor/repository/spec/DoctorSpec.java b/pemits-api/src/main/java/com/pemits/webcare/api/doctor/repository/spec/DoctorSpec.java
@@ -15,6 +15,7 @@
 public class DoctorSpec implements Specification<Doctor> {
 
  private static final String FILTER_BY_NAME = "user.name";
+ private static final String FILTER_BY_USER_ID = "user.id";
  public static final String FILTER_BY_DEPARTMENT_ID = "department.id";
 
  private final String property;
@@ -31,6 +32,8 @@ public Predicate toPredicate(Root<Doctor> root, CriteriaQuery<?> criteriaQuery,
  switch (property) {
  case FILTER_BY_NAME:
  return criteriaBuilder.equal(root.join("user").get("name"), value.toLowerCase());
+ case FILTER_BY_USER_ID:
+ return criteriaBuilder.equal(root.join("user").get("id"), Long.valueOf(value));
  case FILTER_BY_DEPARTMENT_ID:
  return criteriaBuilder.equal(root.join("department").get("id"), Long.valueOf(value));
  default:

diff --git a/pemits-api/src/main/java/com/pemits/webcare/api/patient/repository/spec/PatientSpec.java b/pemits-api/src/main/java/com/pemits/webcare/api/patient/repository/spec/PatientSpec.java
@@ -14,6 +14,8 @@
  */
 public class PatientSpec implements Specification<Patient> {
 
+ private static final String FILTER_BY_USER_ID = "user.id";
+
  private final String property;
  private final String value;
 
@@ -25,6 +27,11 @@ public PatientSpec(String property, String value) {
  @Override
  public Predicate toPredicate(Root<Patient> root, CriteriaQuery<?> criteriaQuery,
  CriteriaBuilder criteriaBuilder) {
- return null;
+ switch (property) {
+ case FILTER_BY_USER_ID:
+ return criteriaBuilder.equal(root.join("user").get("id"), Long.valueOf(value));
+ default:
+ return null;
+ }
  }
 }