Added getNetWorth() method to Portfolio class

solvena · Mar 24, 2026 · 4a105f8 · 4a105f8
1 parent d2c472a
commit 4a105f8
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git 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<Share> 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