Added getNetWorth methods to player and portfolio

IDATT2003-gruppe06 · Apr 8, 2026 · 0771753 · 0771753
1 parent a6c0ce1
commit 0771753
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 0 deletions.
diff --git a/src/main/java/millions/Player.java b/src/main/java/millions/Player.java
@@ -39,6 +39,12 @@ public void withdrawMoney(BigDecimal amount) {
     this.money = this.money.subtract(amount);
   }
 
+  public BigDecimal getNetWorth() {
+    BigDecimal netWorth = this.money;
+    netWorth = netWorth.add(this.portfolio.getNetWorth());
+    return netWorth;
+  }
+
   public String getName() {
     return this.name;
   }

diff --git a/src/main/java/millions/Portfolio.java b/src/main/java/millions/Portfolio.java
@@ -1,5 +1,8 @@
 package millions;
 
+import millions.calculators.SaleCalculator;
+
+import java.math.BigDecimal;
 import java.util.ArrayList;
 import java.util.List;
 
@@ -28,6 +31,14 @@ public List<Share> getShares(String symbol) {
         .toList();
   }
 
+  public BigDecimal getNetWorth() {
+    BigDecimal netWorth = new BigDecimal(0);
+    for (Share share : shares) {
+      netWorth = netWorth.add(new SaleCalculator(share).calculateTotal());
+    }
+    return netWorth;
+  }
+
   public boolean contains(Share share) {
     return this.shares.contains(share);
   }