This commit is contained in:
Sunny Aggarwal 2018-03-25 01:43:36 +01:00
parent 8fe34097eb
commit eac63038e3
1 changed files with 7 additions and 1 deletions

View File

@ -18,6 +18,12 @@ func NewCoinKeeper(am sdk.AccountMapper) CoinKeeper {
return CoinKeeper{am: am} return CoinKeeper{am: am}
} }
// GetCoins returns the coins at the addr.
func (ck CoinKeeper) GetCoins(ctx sdk.Context, addr sdk.Address, amt sdk.Coins) sdk.Coins {
acc := ck.am.GetAccount(ctx, addr)
return acc.GetCoins()
}
// SubtractCoins subtracts amt from the coins at the addr. // SubtractCoins subtracts amt from the coins at the addr.
func (ck CoinKeeper) SubtractCoins(ctx sdk.Context, addr sdk.Address, amt sdk.Coins) (sdk.Coins, sdk.Error) { func (ck CoinKeeper) SubtractCoins(ctx sdk.Context, addr sdk.Address, amt sdk.Coins) (sdk.Coins, sdk.Error) {
acc := ck.am.GetAccount(ctx, addr) acc := ck.am.GetAccount(ctx, addr)
@ -31,7 +37,7 @@ func (ck CoinKeeper) SubtractCoins(ctx sdk.Context, addr sdk.Address, amt sdk.Co
return amt, sdk.ErrInsufficientCoins(fmt.Sprintf("%s < %s", coins, amt)) return amt, sdk.ErrInsufficientCoins(fmt.Sprintf("%s < %s", coins, amt))
} }
// acc.SetCoins(newCoins) acc.SetCoins(newCoins)
ck.am.SetAccount(ctx, acc) ck.am.SetAccount(ctx, acc)
return newCoins, nil return newCoins, nil
} }