mirror of
https://github.com/0glabs/0g-chain.git
synced 2024-11-14 20:15:18 +00:00
4cf41d18c2
Implement GetBalance for extended balances which passes through to `x/bank` for non-extended denoms. This diverges from `x/evmutil` behavior which will panic on non-"akava" calls. Add bank / account keeper mocks for testing, with mockery config for [mockery package setup](https://vektra.github.io/mockery/latest/migrating_to_packages/)
38 lines
1.0 KiB
Go
38 lines
1.0 KiB
Go
package keeper
|
|
|
|
import (
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
"github.com/kava-labs/kava/x/precisebank/types"
|
|
)
|
|
|
|
// GetBalance returns the balance of a specific denom for an address. This will
|
|
// return the extended balance for the ExtendedCoinDenom, and the regular
|
|
// balance for all other denoms.
|
|
func (k Keeper) GetBalance(
|
|
ctx sdk.Context,
|
|
addr sdk.AccAddress,
|
|
denom string,
|
|
) sdk.Coin {
|
|
// Pass through to x/bank for denoms except ExtendedCoinDenom
|
|
if denom != types.ExtendedCoinDenom {
|
|
return k.bk.GetBalance(ctx, addr, denom)
|
|
}
|
|
|
|
// x/bank for integer balance - spendable balance only
|
|
spendableCoins := k.bk.SpendableCoins(ctx, addr)
|
|
integerAmount := spendableCoins.AmountOf(types.IntegerCoinDenom)
|
|
|
|
// x/precisebank for fractional balance
|
|
fractionalAmount, found := k.GetFractionalBalance(ctx, addr)
|
|
if !found {
|
|
fractionalAmount = sdk.ZeroInt()
|
|
}
|
|
|
|
// (Integer * ConversionFactor) + Fractional
|
|
fullAmount := integerAmount.
|
|
Mul(types.ConversionFactor()).
|
|
Add(fractionalAmount)
|
|
|
|
return sdk.NewCoin(types.ExtendedCoinDenom, fullAmount)
|
|
}
|