diff --git a/src/main/java/temppackage/TransactionArchive.java b/src/main/java/temppackage/TransactionArchive.java index 9888c8c..8b257b2 100644 --- a/src/main/java/temppackage/TransactionArchive.java +++ b/src/main/java/temppackage/TransactionArchive.java @@ -29,17 +29,20 @@ public List getTransactions(int week) { } public List getPurchases(int week) { - // TODO - return null; + return transactions.stream() + .filter(t -> t.getWeek() == week && t instanceof Purchase) + .map(t -> (Purchase) t) + .collect(Collectors.toList()); } public List getSales(int week) { - // TODO - return null; + return transactions.stream() + .filter(t -> t.getWeek() == week && t instanceof Sale) + .map(t -> (Sale) t) + .collect(Collectors.toList()); } public int countDistinctWeeks() { - // TODO - return 0; + return (int) transactions.stream().map(Transaction::getWeek).distinct().count(); } }