1 files changed, 37 insertions, 3 deletions
diff --git a/src/main/java/design/persistence/JSONPersonalDatabase.java b/src/main/java/design/persistence/JSONPersonalDatabase.java
index 3ec0012..511d962 100644
--- a/src/main/java/design/persistence/JSONPersonalDatabase.java
+++ b/src/main/java/design/persistence/JSONPersonalDatabase.java
@@ -1,6 +1,12 @@
 package design.persistence;
 
-import com.fasterxml.jackson.databind.ObjectMapper;
+import com.fasterxml.jackson.annotation.JsonAutoDetect;
+import com.fasterxml.jackson.annotation.PropertyAccessor;
+import com.fasterxml.jackson.core.JsonGenerator;
+import com.fasterxml.jackson.core.JsonParser;
+import com.fasterxml.jackson.databind.*;
+import com.fasterxml.jackson.databind.module.SimpleModule;
+import design.model.Course;
 import design.model.Golfer;
 
 import java.io.File;
@@ -14,12 +20,23 @@ public class JSONPersonalDatabase implements PersonalDatabase {
     private final ObjectMapper mapper;
     private final File file;
 
-    public JSONPersonalDatabase(String filename) throws IOException {
+    public JSONPersonalDatabase(String filename) {
         this.file = new File(filename);
         this.cache = new HashMap<>();
         this.mapper = new ObjectMapper();
 
-        load();
+        SimpleModule module = new SimpleModule();
+        module.addDeserializer(Course.class, new CourseIdDeserializer());
+        module.addSerializer(Course.class, new CourseIdSerializer());
+        mapper.registerModule(module);
+        mapper.setVisibility(PropertyAccessor.FIELD, JsonAutoDetect.Visibility.ANY);
+        mapper.writerWithDefaultPrettyPrinter();
+
+        try {
+            load();
+        } catch (IOException ex) {
+            throw new RuntimeException(ex);
+        }
     }
 
     private void load() throws IOException {
@@ -62,4 +79,21 @@ public class JSONPersonalDatabase implements PersonalDatabase {
         cache.put(golfer.getUsername(), golfer);
         save();
     }
+
+    private static class CourseIdSerializer extends JsonSerializer<Course> {
+        @Override
+        public void serialize(Course course, JsonGenerator gen, SerializerProvider serializers) throws IOException {
+            gen.writeNumber(course.getId());
+        }
+    }
+
+    private static class CourseIdDeserializer extends JsonDeserializer<Course> {
+        MasterDatabase masterDB = MasterDatabase.INSTANCE;
+
+        @Override
+        public Course deserialize(JsonParser p, DeserializationContext context) throws IOException {
+            int id = p.getValueAsInt();
+            return masterDB.getCourse(id);
+        }
+    }
 }