mirror of
				https://github.com/0glabs/0g-chain.git
				synced 2025-11-04 00:27:41 +00:00 
			
		
		
		
	* 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
		
			
				
	
	
		
			67 lines
		
	
	
		
			1.9 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
			
		
		
	
	
			67 lines
		
	
	
		
			1.9 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
package simulation
 | 
						|
 | 
						|
import (
 | 
						|
	"fmt"
 | 
						|
	"math/rand"
 | 
						|
	"strings"
 | 
						|
	"time"
 | 
						|
 | 
						|
	"github.com/cosmos/cosmos-sdk/codec"
 | 
						|
	sdk "github.com/cosmos/cosmos-sdk/types"
 | 
						|
	"github.com/cosmos/cosmos-sdk/types/module"
 | 
						|
	"github.com/cosmos/cosmos-sdk/x/simulation"
 | 
						|
 | 
						|
	"github.com/kava-labs/kava/x/issuance/types"
 | 
						|
)
 | 
						|
 | 
						|
var (
 | 
						|
	accs []simulation.Account
 | 
						|
)
 | 
						|
 | 
						|
// RandomizedGenState generates a random GenesisState for the module
 | 
						|
func RandomizedGenState(simState *module.SimulationState) {
 | 
						|
	accs = simState.Accounts
 | 
						|
	params := randomizedParams(simState.Rand)
 | 
						|
	gs := types.NewGenesisState(params, types.AssetSupplies{})
 | 
						|
	fmt.Printf("Selected randomly generated %s parameters:\n%s\n", types.ModuleName, codec.MustMarshalJSONIndent(simState.Cdc, gs))
 | 
						|
	simState.GenState[types.ModuleName] = simState.Cdc.MustMarshalJSON(gs)
 | 
						|
}
 | 
						|
 | 
						|
func randomizedParams(r *rand.Rand) types.Params {
 | 
						|
	assets := randomizedAssets(r)
 | 
						|
	return types.NewParams(assets)
 | 
						|
}
 | 
						|
 | 
						|
func randomizedAssets(r *rand.Rand) types.Assets {
 | 
						|
	randomAssets := types.Assets{}
 | 
						|
	numAssets := Max(1, r.Intn(5))
 | 
						|
	for i := 0; i < numAssets; i++ {
 | 
						|
		denom := strings.ToLower(simulation.RandStringOfLength(r, (r.Intn(3) + 3)))
 | 
						|
		owner := randomOwner(r)
 | 
						|
		paused := r.Intn(2) == 0
 | 
						|
		rateLimited := r.Intn(2) == 0
 | 
						|
		rateLimit := types.NewRateLimit(false, sdk.ZeroInt(), time.Duration(0))
 | 
						|
		if rateLimited {
 | 
						|
			timeLimit := time.Duration(3600000000000 + (r.Intn(24) + 1))
 | 
						|
			assetLimit := simulation.RandIntBetween(r, 100000000000, 1000000000000)
 | 
						|
			rateLimit = types.NewRateLimit(true, sdk.NewInt(int64(assetLimit)), timeLimit)
 | 
						|
		}
 | 
						|
		randomAsset := types.NewAsset(owner.Address, denom, []sdk.AccAddress{}, paused, true, rateLimit)
 | 
						|
		randomAssets = append(randomAssets, randomAsset)
 | 
						|
	}
 | 
						|
	return randomAssets
 | 
						|
}
 | 
						|
 | 
						|
func randomOwner(r *rand.Rand) simulation.Account {
 | 
						|
	acc, _ := simulation.RandomAcc(r, accs)
 | 
						|
	return acc
 | 
						|
}
 | 
						|
 | 
						|
// Max return max of two ints
 | 
						|
func Max(x, y int) int {
 | 
						|
	if x > y {
 | 
						|
		return x
 | 
						|
	}
 | 
						|
	return y
 | 
						|
}
 |