Accept and Decline restaurant

KU-Wongnai · Nov 4, 2023 · 1e36477 · 1e36477
1 parent e8bffa4
commit 1e36477
Show file tree

Hide file tree

Showing 2 changed files with 26 additions and 0 deletions.
diff --git a/src/main/java/ku/cs/kuwongnai/restaurant/controller/RestaurantController.java b/src/main/java/ku/cs/kuwongnai/restaurant/controller/RestaurantController.java
@@ -58,6 +58,20 @@ public ResponseEntity<Restaurant> update(@PathVariable Long id, @Valid @RequestB
         return new ResponseEntity<>(updatedRestaurant, HttpStatus.OK);
     }
 
+    @PutMapping("/{id}/accept")
+    public ResponseEntity<Restaurant> accept(@PathVariable Long id) {
+        Restaurant acceptedRestaurant = service.acceptRestaurant(id);
+        publisher.publishJson("events.restaurant", "restaurant.updated", acceptedRestaurant);
+        return new ResponseEntity<>(acceptedRestaurant, HttpStatus.OK);
+    }
+
+    @PutMapping("/{id}/accept")
+    public ResponseEntity<Restaurant> decline(@PathVariable Long id) {
+        Restaurant declinedRestaurant = service.declineRestaurant(id);
+        publisher.publishJson("events.restaurant", "restaurant.updated", declinedRestaurant);
+        return new ResponseEntity<>(declinedRestaurant, HttpStatus.OK);
+    }
+
     @DeleteMapping("/{id}")
     public ResponseEntity<Restaurant> delete(@PathVariable Long id,
                                              @AuthenticationPrincipal Jwt jwt) {

diff --git a/src/main/java/ku/cs/kuwongnai/restaurant/service/RestaurantService.java b/src/main/java/ku/cs/kuwongnai/restaurant/service/RestaurantService.java
@@ -76,6 +76,18 @@ public Restaurant updateRestaurant(Long id, RestaurantRequest requestBody) {
         return restaurantRepository.save(record);
     }
 
+    public Restaurant acceptRestaurant(Long id) {
+        Restaurant record = getRestaurantById(id);
+        record.setStatus(RestaurantStatus.ACCEPTED);
+        return restaurantRepository.save(record);
+    }
+
+    public Restaurant declineRestaurant(Long id) {
+        Restaurant record = getRestaurantById(id);
+        record.setStatus(RestaurantStatus.DECLINED);
+        return restaurantRepository.save(record);
+    }
+
     public Restaurant deleteRestaurant(Long id, Long userId) {
 
         // Get the restaurant by its ID