progress on implementing strategy pattern

1 files changed, 51 insertions, 0 deletions

diff --git a/src/model/course-search/CourseList.java b/src/model/course-search/CourseList.java
new file mode 100644
index 0000000..2766e11
--- /dev/null
+++ b/src/model/course-search/CourseList.java
@@ -0,0 +1,51 @@
+package design.model;
+
+import java.util.ArrayList;
+import java.util.List;
+
+import Course;
+
+/*
+ * Defines a list of courses that can be sorted, added to or removed from.
+ * Created by Willem Dalton
+ */
+public class CourseList {
+    private CourseSorter sorter;
+    private List<Course> courses;
+
+    public CourseList()
+    {
+        courses = new ArrayList<>();
+        sorter = new CourseSorter();
+    }
+
+    public List<Course> getCourses()
+    {
+        return courses;
+    }
+
+    public setCourses(List<Course> courses)
+    {
+        this.courses = courses;
+    }
+
+    setSorter(CourseSorter courseSorter)
+    {
+        sorter = courseSorter;
+    }
+
+    add(Course course)
+    {
+        courses.add(course);
+    }
+
+    remove(Course course)
+    {
+        courses.remove(course);
+    }
+
+    sort()
+    {
+        sorter.sortCourses(courses);
+    }
+}