Added methods to update ID

1 files changed, 8 insertions, 1 deletions

diff --git a/ufund-api/src/main/java/com/ufund/api/ufundapi/Cupboard.java b/ufund-api/src/main/java/com/ufund/api/ufundapi/Cupboard.java
index a11e7ab..420e62e 100644
--- a/ufund-api/src/main/java/com/ufund/api/ufundapi/Cupboard.java
+++ b/ufund-api/src/main/java/com/ufund/api/ufundapi/Cupboard.java
@@ -14,10 +14,17 @@ public class Cupboard {
     }
 
     public void createNeed(String name, double max, Need.GoalType type) {
-        Need need = new Need(name, 0, max, type);
+        int id = needs.size();
+        Need need = new Need(name, id, max, type);
         addNeed(need);
     }
 
+    public void updateID(int id) {
+        for (int i = 0; i < needs.size(); i++) {
+            needs.get(i).setID(i);
+        }
+    }
+
     public void removeNeed(Need need) {
         needs.remove(need);
     }