tdd-java-demo

commit efb294395e63bd329ed8e18c91a1eae646446965
parent 7df1586a1e047991d13cbbbb63444fa4104c1797
Author: Jul <jul@9o.is>
Date:   Wed, 31 Mar 2021 16:49:10 +0800

use a hashmap to track amounts before calculating total

Diffstat:
M
src/main/java/takehome/Runner.java
 | 
12
++++++++++--

1 file changed, 10 insertions(+), 2 deletions(-)
diff --git a/src/main/java/takehome/Runner.java b/src/main/java/takehome/Runner.java
@@ -4,6 +4,8 @@ import java.io.InputStream;
 import java.io.OutputStream;
 import java.io.PrintStream;
 import java.text.DecimalFormat;
+import java.util.HashMap;
+import java.util.Map;
 import java.util.Scanner;
 
 public class Runner {
@@ -21,11 +23,17 @@ public class Runner {
                 new Item("bread", 0.80)
         };
 
-        double total = 0;
+        Map<Item, Integer> amounts = new HashMap<>();
         for (Item item : items) {
-            total += getTotal(item);
+            int amount = getItemAmount(item);
+            amounts.put(item, amount);
         }
 
+        double total = 0;
+        for (Item item : amounts.keySet()) {
+            int amount = amounts.get(item);
+            total += item.getTotal(amount);
+        }
 
         DecimalFormat df = new DecimalFormat("0.00");
         out.println("Total: $" + df.format(total));