Implement getBalances

ericgrandt · Aug 5, 2017 · 988c295 · 988c295
1 parent b4fd7cc
commit 988c295
Showing 1 changed file with 14 additions and 2 deletions.
diff --git a/src/main/java/com/erigitic/config/TEAccount.java b/src/main/java/com/erigitic/config/TEAccount.java
@@ -151,9 +151,21 @@ public BigDecimal getBalance(Currency currency, Set<Context> contexts) {
         return BigDecimal.ZERO;
     }
 
+    /**
+     * Get a player's balance for each currency type
+     *
+     * @param contexts
+     * @return Map A map of the balances of each currency
+     */
     @Override
     public Map<Currency, BigDecimal> getBalances(Set<Context> contexts) {
-        return new HashMap<>();
+        HashMap<Currency, BigDecimal> balances = new HashMap<>();
+
+        for (Currency currency : totalEconomy.getCurrencies()) {
+            balances.put(currency, getBalance(currency, contexts));
+        }
+
+        return balances;
     }
 
     /**
@@ -163,7 +175,7 @@ public Map<Currency, BigDecimal> getBalances(Set<Context> contexts) {
      * @param amount Amount to set the balance to
      * @param cause
      * @param contexts
-     * @return
+     * @return TransactionResult Result of the transaction
      */
     @Override
     public TransactionResult setBalance(Currency currency, BigDecimal amount, Cause cause, Set<Context> contexts) {