tdd-java-demo

commit e968ffe2e038f51e5f93383bcd669a22b38dbc33
parent c20d13adcae2a0956385d06a6048d4672f1cd751
Author: Jul <jul@9o.is>
Date:   Wed, 31 Mar 2021 23:20:39 +0800

use Multiset in Basket

Diffstat:
M
src/main/java/takehome/Basket.java
 | 
11
+++++------

1 file changed, 5 insertions(+), 6 deletions(-)
diff --git a/src/main/java/takehome/Basket.java b/src/main/java/takehome/Basket.java
@@ -6,20 +6,19 @@ import com.google.common.collect.Multiset;
 import java.util.Map;
 
 public class Basket {
-    private final Map<Item, Integer> amounts;
+    private final Multiset<Item> amounts;
 
     public Basket(Map<Item, Integer> amounts) {
-        this.amounts = amounts;
+        this.amounts = toMultiset(amounts);
     }
 
     public Multiset<Item> getAmounts() {
-        return toMultiset(amounts);
+        return amounts;
     }
 
     double getTotal() {
-        Multiset<Item> set = toMultiset(amounts);
         double total = 0;
-        for (Item item : set.elementSet()) {
+        for (Item item : amounts.elementSet()) {
             int amount = getAmount(item);
             total += item.getTotal(amount);
         }
@@ -27,7 +26,7 @@ public class Basket {
     }
 
     Integer getAmount(Item item) {
-        return toMultiset(amounts).count(item);
+        return amounts.count(item);
     }
 
     double getTotal(Discount[] discounts) {