diff --git a/src/main/java/temppackage/Portfolio.java b/src/main/java/temppackage/Portfolio.java index ccadd19..89c19c2 100644 --- a/src/main/java/temppackage/Portfolio.java +++ b/src/main/java/temppackage/Portfolio.java @@ -22,6 +22,12 @@ public List getShares() { return this.shares; } + public List getShares(String symbol) { + return this.shares.stream() + .filter(share -> share.getStock().getSymbol().equals(symbol)) + .toList(); + } + public boolean contains(Share share) { return this.shares.contains(share); }