mirror of
https://github.com/0glabs/0g-chain.git
synced 2024-11-14 12:05:18 +00:00
b356309d90
* add rate-limiting and optional blocklists * fix: check account is not nil * add tests for rate-limiting * update simulations * fix typos * remove unsued function arg
63 lines
1.7 KiB
Go
63 lines
1.7 KiB
Go
package keeper
|
|
|
|
import (
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
sdkerrors "github.com/cosmos/cosmos-sdk/types/errors"
|
|
|
|
"github.com/kava-labs/kava/x/issuance/types"
|
|
)
|
|
|
|
// GetParams returns the params from the store
|
|
func (k Keeper) GetParams(ctx sdk.Context) types.Params {
|
|
var p types.Params
|
|
k.paramSubspace.GetParamSet(ctx, &p)
|
|
return p
|
|
}
|
|
|
|
// SetParams sets params on the store
|
|
func (k Keeper) SetParams(ctx sdk.Context, params types.Params) {
|
|
k.paramSubspace.SetParamSet(ctx, ¶ms)
|
|
}
|
|
|
|
// GetAsset returns an asset from the params and a boolean for if it was found
|
|
func (k Keeper) GetAsset(ctx sdk.Context, denom string) (types.Asset, bool) {
|
|
params := k.GetParams(ctx)
|
|
for _, asset := range params.Assets {
|
|
if asset.Denom == denom {
|
|
return asset, true
|
|
}
|
|
}
|
|
return types.Asset{}, false
|
|
}
|
|
|
|
// SetAsset sets an asset in the params
|
|
func (k Keeper) SetAsset(ctx sdk.Context, asset types.Asset) {
|
|
params := k.GetParams(ctx)
|
|
for i := range params.Assets {
|
|
if params.Assets[i].Denom == asset.Denom {
|
|
params.Assets[i] = asset
|
|
}
|
|
}
|
|
k.SetParams(ctx, params)
|
|
}
|
|
|
|
// GetRateLimit returns the rete-limit parameters for the input denom
|
|
func (k Keeper) GetRateLimit(ctx sdk.Context, denom string) (types.RateLimit, error) {
|
|
asset, found := k.GetAsset(ctx, denom)
|
|
if !found {
|
|
sdkerrors.Wrap(types.ErrAssetNotFound, denom)
|
|
}
|
|
return asset.RateLimit, nil
|
|
}
|
|
|
|
// SynchronizeBlockList resets the block list to empty for any asset that is not blockable - could happen if this value is changed via governance
|
|
func (k Keeper) SynchronizeBlockList(ctx sdk.Context) {
|
|
params := k.GetParams(ctx)
|
|
for _, asset := range params.Assets {
|
|
if !asset.Blockable && len(asset.BlockedAddresses) > 0 {
|
|
asset.BlockedAddresses = []sdk.AccAddress{}
|
|
k.SetAsset(ctx, asset)
|
|
}
|
|
}
|
|
}
|