Added methods to update ID

2 files changed, 13 insertions, 2 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);
     }
diff --git a/ufund-api/src/main/java/com/ufund/api/ufundapi/Need.java b/ufund-api/src/main/java/com/ufund/api/ufundapi/Need.java
index f6ef720..ea294bf 100644
--- a/ufund-api/src/main/java/com/ufund/api/ufundapi/Need.java
+++ b/ufund-api/src/main/java/com/ufund/api/ufundapi/Need.java
@@ -8,7 +8,7 @@ public class Need {
     }
 
     private String name;
-    final private int id;
+    private int id;
     private String[] filterAttributes;
     final private GoalType type;
     private double maxGoal;
@@ -60,4 +60,8 @@ public class Need {
     public void setName(String name) {
         this.name = name;
     }
+
+    public void setID(int id){
+        this.id = id;
+    }
 }
\ No newline at end of file