package dao; import domain.Student; import java.util.Collection; import java.util.SortedMap; import java.util.TreeMap; /** * * @author Mark George */ public final class StudentDao { private final static SortedMap<String, Student> students = new TreeMap<>(); public StudentDao() { // add some dummy data for testing if (students.isEmpty()) { addStudent(new Student("1234", "Boris")); addStudent(new Student("4321", "Doris")); } } public void addStudent(Student studentToAdd) { students.put(studentToAdd.getId(), studentToAdd); } public void replaceStudent(String id, Student studentToAdd) { removeStudent(id); students.put(studentToAdd.getId(), studentToAdd); } public void removeStudent(String id) { students.remove(id); } public Collection<Student> getStudents() { return students.values(); } public Student getStudentById(String id) { return students.get(id); } }