package design.persistence; import com.fasterxml.jackson.annotation.*; import com.fasterxml.jackson.databind.*; import com.fasterxml.jackson.databind.module.SimpleModule; import com.fasterxml.jackson.datatype.jsr310.JavaTimeModule; import com.fasterxml.jackson.module.paramnames.ParameterNamesModule; import design.model.Course; import design.model.Golfer; import design.model.League; import java.io.File; import java.io.IOException; import java.util.HashMap; import java.util.List; import java.util.Map; public class JSONPersonalDatabase implements PersonalDatabase { private static JSONPersonalDatabase INSTANCE; public static JSONPersonalDatabase instance() { if (INSTANCE == null) { INSTANCE = new JSONPersonalDatabase("data/personaldb.json"); } return INSTANCE; } private final Map cache; private final ObjectMapper mapper; private final File file; private JSONPersonalDatabase(String filename) { this.file = new File(filename); this.cache = new HashMap<>(); this.mapper = new ObjectMapper(); SimpleModule module = new SimpleModule(); module.addDeserializer(Course.class, new Serializers.CourseIdDeserializer()); module.addSerializer(Course.class, new Serializers.CourseIdSerializer()); module.addSerializer(League.class, new Serializers.LeagueIDSerializer()); module.addDeserializer(League.class, new Serializers.LeagueIDDeserializer()); mapper.registerModule(module); mapper.configure(DeserializationFeature.FAIL_ON_UNKNOWN_PROPERTIES, false); mapper.setVisibility(PropertyAccessor.FIELD, JsonAutoDetect.Visibility.ANY); mapper.registerModule(new ParameterNamesModule(JsonCreator.Mode.PROPERTIES)); mapper.registerModule(new JavaTimeModule()); mapper.configOverride(List.class).setSetterInfo(JsonSetter.Value.forValueNulls(Nulls.AS_EMPTY)); try { load(); } catch (IOException ex) { throw new RuntimeException(ex); } } private void load() throws IOException { Golfer[] data = mapper.readValue(file, Golfer[].class); cache.clear(); for (Golfer i : data) { cache.put(i.getUsername(), i); } } private void save() throws IOException { Golfer[] data = cache.values().toArray(Golfer[]::new); mapper.writer(new Serializers.CustomPrettyPrinter()).writeValue(file, data); } // turns that collection into a real array of golfers @Override public Golfer[] getGolfers() { return cache.values().toArray(Golfer[]::new); } @Override public Golfer getGolfer(String username) { return cache.get(username); } @Override public void addGolfer(Golfer golfer) throws IOException { cache.putIfAbsent(golfer.getUsername(), golfer); save(); } @Override public void removeGolfer(Golfer golfer) throws IOException { cache.remove(golfer.getUsername()); save(); } @Override public void updateGolfer(Golfer golfer) throws IOException { cache.put(golfer.getUsername(), golfer); save(); } @Override public void importData(JsonNode tree) throws IOException { Golfer[] data = mapper.treeToValue(tree, Golfer[].class); cache.clear(); for (Golfer golfer : data) { cache.put(golfer.getUsername(), golfer); } save(); } @Override public JsonNode exportData() { Object[] data = cache.values().toArray(); return mapper.valueToTree(data); } }