mirror of
https://github.com/0glabs/0g-chain.git
synced 2024-12-26 00:05:18 +00:00
49d62dd076
* initial feature scaffolding * implement interest keeper logic * basic AccrueInterest * accrue interest on borrow * update borrow index formula * update sample reserve factor * move AccrueInterest to begin blocker * refactor interest rate updates for accrue interest * use interest rate model from store * refactor begin blocker state machine * add reserve factor to interest model params * update comment * store money market instead of interest rate models * update test suite * use BorrowedCoins store key * update public functions and alias * unit tests, keeper test scaffolding * demo panic * address revisions * add 'normal no jump' test case * spy = 1 + borrow rate * update comment * APYToSPY unit test * per user borrow index list * interest keeper test * test: interest applied on successive borrows * varied snapshot times * test: multiple, varied snapshots * address revisions * add pending interest before validating new borrow * update makefile * address revisions * fix test
104 lines
3.1 KiB
Go
104 lines
3.1 KiB
Go
package keeper
|
|
|
|
import (
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
sdkerrors "github.com/cosmos/cosmos-sdk/types/errors"
|
|
supplyExported "github.com/cosmos/cosmos-sdk/x/supply/exported"
|
|
|
|
"github.com/kava-labs/kava/x/harvest/types"
|
|
)
|
|
|
|
// Deposit deposit
|
|
func (k Keeper) Deposit(ctx sdk.Context, depositor sdk.AccAddress, amount sdk.Coin) error {
|
|
err := k.ValidateDeposit(ctx, amount)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
err = k.supplyKeeper.SendCoinsFromAccountToModule(ctx, depositor, types.ModuleAccountName, sdk.NewCoins(amount))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
deposit, found := k.GetDeposit(ctx, depositor, amount.Denom)
|
|
if !found {
|
|
deposit = types.NewDeposit(depositor, amount)
|
|
} else {
|
|
deposit.Amount = deposit.Amount.Add(amount)
|
|
}
|
|
|
|
k.SetDeposit(ctx, deposit)
|
|
|
|
ctx.EventManager().EmitEvent(
|
|
sdk.NewEvent(
|
|
types.EventTypeHarvestDeposit,
|
|
sdk.NewAttribute(sdk.AttributeKeyAmount, amount.String()),
|
|
sdk.NewAttribute(types.AttributeKeyDepositor, deposit.Depositor.String()),
|
|
sdk.NewAttribute(types.AttributeKeyDepositDenom, deposit.Amount.Denom),
|
|
),
|
|
)
|
|
|
|
return nil
|
|
}
|
|
|
|
// ValidateDeposit validates a deposit
|
|
func (k Keeper) ValidateDeposit(ctx sdk.Context, amount sdk.Coin) error {
|
|
params := k.GetParams(ctx)
|
|
for _, lps := range params.LiquidityProviderSchedules {
|
|
if lps.DepositDenom == amount.Denom {
|
|
return nil
|
|
}
|
|
}
|
|
return sdkerrors.Wrapf(types.ErrInvalidDepositDenom, "liquidity provider denom %s not found", amount.Denom)
|
|
}
|
|
|
|
// Withdraw returns some or all of a deposit back to original depositor
|
|
func (k Keeper) Withdraw(ctx sdk.Context, depositor sdk.AccAddress, amount sdk.Coin) error {
|
|
deposit, found := k.GetDeposit(ctx, depositor, amount.Denom)
|
|
if !found {
|
|
return sdkerrors.Wrapf(types.ErrDepositNotFound, "no %s deposit found for %s", amount.Denom, depositor)
|
|
}
|
|
|
|
if !deposit.Amount.IsGTE(amount) {
|
|
return sdkerrors.Wrapf(types.ErrInvalidWithdrawAmount, "%s>%s", amount, deposit.Amount)
|
|
}
|
|
|
|
err := k.supplyKeeper.SendCoinsFromModuleToAccount(ctx, types.ModuleAccountName, depositor, sdk.NewCoins(amount))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
ctx.EventManager().EmitEvent(
|
|
sdk.NewEvent(
|
|
types.EventTypeHarvestWithdrawal,
|
|
sdk.NewAttribute(sdk.AttributeKeyAmount, amount.String()),
|
|
sdk.NewAttribute(types.AttributeKeyDepositor, depositor.String()),
|
|
sdk.NewAttribute(types.AttributeKeyDepositDenom, amount.Denom),
|
|
),
|
|
)
|
|
|
|
if deposit.Amount.IsEqual(amount) {
|
|
ctx.EventManager().EmitEvent(
|
|
sdk.NewEvent(
|
|
types.EventTypeDeleteHarvestDeposit,
|
|
sdk.NewAttribute(types.AttributeKeyDepositor, depositor.String()),
|
|
sdk.NewAttribute(types.AttributeKeyDepositDenom, amount.Denom),
|
|
),
|
|
)
|
|
k.DeleteDeposit(ctx, deposit)
|
|
return nil
|
|
}
|
|
|
|
deposit.Amount = deposit.Amount.Sub(amount)
|
|
k.SetDeposit(ctx, deposit)
|
|
|
|
return nil
|
|
}
|
|
|
|
// GetTotalDeposited returns the total amount deposited for the input deposit type and deposit denom
|
|
func (k Keeper) GetTotalDeposited(ctx sdk.Context, depositDenom string) (total sdk.Int) {
|
|
var macc supplyExported.ModuleAccountI
|
|
macc = k.supplyKeeper.GetModuleAccount(ctx, types.ModuleAccountName)
|
|
return macc.GetCoins().AmountOf(depositDenom)
|
|
}
|