diff --git a/src/main/java/Portfolio.java b/src/main/java/Portfolio.java index a25561f..8b8e58c 100644 --- a/src/main/java/Portfolio.java +++ b/src/main/java/Portfolio.java @@ -1,3 +1,4 @@ +import java.math.BigDecimal; import java.util.ArrayList; import java.util.List; @@ -34,5 +35,14 @@ public List getShares(String symbol) { public boolean contains(Share share) { return shares.contains(share); } + + public BigDecimal getNetWorth() { + BigDecimal total = BigDecimal.ZERO; + for (Share share : shares) { + SaleCalculator calc = new SaleCalculator(share); + total = total.add(calc.calculateTotal()); + } + return total; + } } diff --git a/target/classes/Portfolio.class b/target/classes/Portfolio.class index 2dc1492..05b636c 100644 Binary files a/target/classes/Portfolio.class and b/target/classes/Portfolio.class differ