updateNeed Controller and Cupboard functionality

2 files changed, 10 insertions, 4 deletions

diff --git a/ufund-api/src/main/java/com/ufund/api/ufundapi/controller/CupboardController.java b/ufund-api/src/main/java/com/ufund/api/ufundapi/controller/CupboardController.java
index 106b2e0..0de3403 100644
--- a/ufund-api/src/main/java/com/ufund/api/ufundapi/controller/CupboardController.java
+++ b/ufund-api/src/main/java/com/ufund/api/ufundapi/controller/CupboardController.java
@@ -41,8 +41,14 @@ public class CupboardController {
     }
 
     @PutMapping("")
-    public void updateNeed(@RequestBody Need need) {
-        cupboard.updateNeed(need);
+    public ResponseEntity<Need> updateNeed(@RequestBody Need need) {
+        try {
+            need = cupboard.updateNeed(need);
+            return new ResponseEntity<>(need, HttpStatus.OK);
+        } catch (IOException e) {
+            return new ResponseEntity<>(HttpStatus.INTERNAL_SERVER_ERROR);
+        }
+       
     }
 
     @DeleteMapping("/{id}")
diff --git a/ufund-api/src/main/java/com/ufund/api/ufundapi/model/Cupboard.java b/ufund-api/src/main/java/com/ufund/api/ufundapi/model/Cupboard.java
index a626561..0ce015c 100644
--- a/ufund-api/src/main/java/com/ufund/api/ufundapi/model/Cupboard.java
+++ b/ufund-api/src/main/java/com/ufund/api/ufundapi/model/Cupboard.java
@@ -23,8 +23,8 @@ public class Cupboard {
         return dao.findNeeds(name);
     }
 
-    public void updateNeed(Need need) throws IOException {
-        dao.updateNeed(need);
+    public Need updateNeed(Need need) throws IOException {
+        return dao.updateNeed(need);
     }
 
     public void removeNeed(int id) throws IOException {