[#38] feat: Guest, Budget validation 추가

codesquad-member-2020 · Jun 5, 2020 · d6a81dc · d6a81dc
1 parent 7be30ad
commit d6a81dc
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 2 deletions.
diff --git a/BE/src/main/java/com/codesquad/airbnb/room/domain/Budget.java b/BE/src/main/java/com/codesquad/airbnb/room/domain/Budget.java
@@ -2,14 +2,23 @@
 
 import lombok.*;
 
+import javax.validation.constraints.AssertTrue;
+import javax.validation.constraints.Min;
+
 @Getter
 @Setter
 @NoArgsConstructor
 @AllArgsConstructor
 @ToString
 public class Budget {
 
+    @Min(0)
     private int lowestPrice;
 
     private int highestPrice = Integer.MAX_VALUE;
+
+    @AssertTrue
+    private boolean validateBudget() {
+        return highestPrice > lowestPrice;
+    }
 }
diff --git a/BE/src/main/java/com/codesquad/airbnb/room/ui/MainController.java b/BE/src/main/java/com/codesquad/airbnb/room/ui/MainController.java
@@ -27,8 +27,8 @@ public class MainController {
 
     @GetMapping("")
     public Main showMain(ReservationDate reservationDate,
-                         Guest guest,
-                         Budget budget) {
+                         @Valid Guest guest,
+                         @Valid Budget budget) {
 
         reservationDate.validateReservationDate();
         return viewDAO.main(utilDAO, reservationDate, guest, budget);