diff --git a/src/se/edu/inclass/Main.java b/src/se/edu/inclass/Main.java index 3781393..a6325d6 100644 --- a/src/se/edu/inclass/Main.java +++ b/src/se/edu/inclass/Main.java @@ -8,6 +8,8 @@ import java.util.ArrayList; import java.util.stream.Collectors; +import static java.util.stream.Collectors.toList; + public class Main { private TaskNameComparator taskNameComparator; @@ -16,10 +18,15 @@ public static void main(String[] args) { DataManager dm = new DataManager("./data/data.txt"); ArrayList tasksData = dm.loadData(); - System.out.println("Printing deadlines"); - printDeadlines(tasksData); +// System.out.println("Printing deadlines"); +// printDeadlines(tasksData); +// +// System.out.println("Total number of deadlines: " + countDeadlines(tasksData)); + printDeadlinesUsingStreams(tasksData); + for (Task t : filterByString(tasksData, "11")) { + System.out.println(t); + } - System.out.println("Total number of deadlines: " + countDeadlines(tasksData)); } @@ -46,4 +53,21 @@ public static void printDeadlines(ArrayList tasksData) { } } } + + public static void printDeadlinesUsingStreams(ArrayList tasksData) { + tasksData.stream() + .filter(s -> s instanceof Deadline) + .sorted((a, b) -> a.getDescription().toLowerCase().compareTo(b.getDescription().toLowerCase())) + .forEach(System.out::println); + } + + public static ArrayList filterByString(ArrayList tasksData, String filterString) { + ArrayList filteredTaskList = (ArrayList) tasksData.stream() + .filter(s -> s.getDescription().contains(filterString)) + .collect(toList()); + + return filteredTaskList; + } + + //test }