Push current changes (working on documentation and tests)

1 files changed, 126 insertions, 0 deletions

diff --git a/ufund-api/src/main/java/com/ufund/api/ufundapi/persistence/CupboardFileDAO.java b/ufund-api/src/main/java/com/ufund/api/ufundapi/persistence/CupboardFileDAO.java
new file mode 100644
index 0000000..c4aaca3
--- /dev/null
+++ b/ufund-api/src/main/java/com/ufund/api/ufundapi/persistence/CupboardFileDAO.java
@@ -0,0 +1,126 @@
+package com.ufund.api.ufundapi.persistence;
+
+import com.fasterxml.jackson.databind.ObjectMapper;
+import com.ufund.api.ufundapi.model.Need;
+import org.springframework.beans.factory.annotation.Value;
+import org.springframework.stereotype.Component;
+
+import java.io.File;
+import java.io.IOException;
+import java.util.Map;
+import java.util.TreeMap;
+
+@Component
+public class CupboardFileDAO implements CupboardDAO {
+
+    private final Map<Integer, Need> needs; // cache
+    private final ObjectMapper objectMapper;
+    private static int nextId;
+    private final String filename;
+
+    public CupboardFileDAO(@Value("${cupboard.file}") String filename, ObjectMapper objectMapper) throws IOException {
+        this.filename = filename;
+        this.objectMapper = objectMapper;
+        needs = new TreeMap<>();
+        load();  // load the heroes from the file
+    }
+
+    private synchronized static int nextId() {
+        int id = nextId;
+        nextId++;
+        return id;
+    }
+
+    /**
+     * Load changes from the json file
+     *
+     * @throws IOException Any IO issue with the file
+     */
+    private void load() throws IOException {
+        needs.clear();
+        nextId = 0;
+
+        Need[] needsArray = objectMapper.readValue(new File(filename), Need[].class);
+
+        for (Need need : needsArray) {
+            needs.put(need.getId(), need);
+            if (need.getId() > nextId()) {
+                nextId = need.getId();
+            }
+        }
+        nextId++;
+    }
+
+    /**
+     * Return an array of the needs
+     *
+     * @return An array of all the needs
+     */
+    private Need[] getNeedsArray() {
+        return needs.values().toArray(Need[]::new);
+    }
+
+    /**
+     * Saves the needs to json
+     *
+     * @return True if the save was successful, false otherwise
+     * @throws IOException If there was an IO issue saving the file
+     */
+    private boolean save() throws IOException {
+        Need[] needArray = getNeedsArray();
+
+        objectMapper.writeValue(new File(filename), needArray);
+        return true;
+    }
+
+    @Override
+    public Need[] getNeeds() {
+        synchronized (needs) {
+            return getNeedsArray();
+        }
+    }
+
+    @Override
+    public Need getNeed(int id) {
+        synchronized (needs) {
+            return needs.getOrDefault(id, null);
+        }
+    }
+
+    @Override
+    public Need addNeed(Need need) throws IOException {
+        synchronized (needs) {
+            Need newNeed = new Need(need);
+            newNeed.setID(nextId());
+            needs.put(newNeed.getId(), newNeed);
+            save();
+            return newNeed;
+        }
+    }
+
+    @Override
+    public Need updateNeed(Need need) throws IOException {
+        synchronized (needs) {
+            if (needs.containsKey(need.getId())) {
+                needs.put(need.getId(), need);
+                save();
+                return need;
+            } else {
+                return null;
+            }
+
+        }
+    }
+
+    @Override
+    public boolean deleteNeed(int id) throws IOException {
+        synchronized (needs) {
+            if (needs.containsKey(id)) {
+                needs.remove(id);
+                return save();
+            } else {
+                return false;
+            }
+        }
+    }
+}
\ No newline at end of file