diff --git a/src/main/java/Portfolio.java b/src/main/java/Portfolio.java new file mode 100644 index 0000000..a25561f --- /dev/null +++ b/src/main/java/Portfolio.java @@ -0,0 +1,38 @@ +import java.util.ArrayList; +import java.util.List; + +public class Portfolio { + + private final List shares; + + public Portfolio() { + this.shares = new ArrayList<>(); + } + + public boolean addShare(Share share) { + return shares.add(share); + } + + public boolean removeShare(Share share) { + return shares.remove(share); + } + + public List getShares() { + return new ArrayList<>(shares); + } + + public List getShares(String symbol) { + List result = new ArrayList<>(); + for (Share share : shares) { + if (share.getStock().getSymbol().equals(symbol)) { + result.add(share); + } + } + return result; + } + + public boolean contains(Share share) { + return shares.contains(share); + } + +} diff --git a/target/classes/Portfolio.class b/target/classes/Portfolio.class new file mode 100644 index 0000000..2dc1492 Binary files /dev/null and b/target/classes/Portfolio.class differ