145 lines
4.1 KiB
Go
145 lines
4.1 KiB
Go
package simulation
|
|
|
|
import (
|
|
"fmt"
|
|
"math/rand"
|
|
|
|
"github.com/cosmos/cosmos-sdk/types/simulation"
|
|
)
|
|
|
|
const (
|
|
// Minimum time per block
|
|
minTimePerBlock int64 = 10000 / 2
|
|
|
|
// Maximum time per block
|
|
maxTimePerBlock int64 = 10000
|
|
)
|
|
|
|
// TODO: explain transitional matrix usage
|
|
var (
|
|
// Currently there are 3 different liveness types,
|
|
// fully online, spotty connection, offline.
|
|
defaultLivenessTransitionMatrix, _ = CreateTransitionMatrix([][]int{
|
|
{90, 20, 1},
|
|
{10, 50, 5},
|
|
{0, 10, 1000},
|
|
})
|
|
|
|
// 3 states: rand in range [0, 4*provided blocksize],
|
|
// rand in range [0, 2 * provided blocksize], 0
|
|
defaultBlockSizeTransitionMatrix, _ = CreateTransitionMatrix([][]int{
|
|
{85, 5, 0},
|
|
{15, 92, 1},
|
|
{0, 3, 99},
|
|
})
|
|
)
|
|
|
|
// Params define the parameters necessary for running the simulations
|
|
type Params struct {
|
|
pastEvidenceFraction float64
|
|
numKeys int
|
|
evidenceFraction float64
|
|
initialLivenessWeightings []int
|
|
livenessTransitionMatrix simulation.TransitionMatrix
|
|
blockSizeTransitionMatrix simulation.TransitionMatrix
|
|
}
|
|
|
|
func (p Params) PastEvidenceFraction() float64 {
|
|
return p.pastEvidenceFraction
|
|
}
|
|
|
|
func (p Params) NumKeys() int {
|
|
return p.numKeys
|
|
}
|
|
|
|
func (p Params) EvidenceFraction() float64 {
|
|
return p.evidenceFraction
|
|
}
|
|
|
|
func (p Params) InitialLivenessWeightings() []int {
|
|
return p.initialLivenessWeightings
|
|
}
|
|
|
|
func (p Params) LivenessTransitionMatrix() simulation.TransitionMatrix {
|
|
return p.livenessTransitionMatrix
|
|
}
|
|
|
|
func (p Params) BlockSizeTransitionMatrix() simulation.TransitionMatrix {
|
|
return p.blockSizeTransitionMatrix
|
|
}
|
|
|
|
// RandomParams returns random simulation parameters
|
|
func RandomParams(r *rand.Rand) Params {
|
|
return Params{
|
|
pastEvidenceFraction: r.Float64(),
|
|
numKeys: simulation.RandIntBetween(r, 2, 2500), // number of accounts created for the simulation
|
|
evidenceFraction: r.Float64(),
|
|
initialLivenessWeightings: []int{simulation.RandIntBetween(r, 1, 80), r.Intn(10), r.Intn(10)},
|
|
livenessTransitionMatrix: defaultLivenessTransitionMatrix,
|
|
blockSizeTransitionMatrix: defaultBlockSizeTransitionMatrix,
|
|
}
|
|
}
|
|
|
|
//-----------------------------------------------------------------------------
|
|
// Param change proposals
|
|
|
|
// ParamChange defines the object used for simulating parameter change proposals
|
|
type ParamChange struct {
|
|
subspace string
|
|
key string
|
|
simValue simulation.SimValFn
|
|
}
|
|
|
|
func (spc ParamChange) Subspace() string {
|
|
return spc.subspace
|
|
}
|
|
|
|
func (spc ParamChange) Key() string {
|
|
return spc.key
|
|
}
|
|
|
|
func (spc ParamChange) SimValue() simulation.SimValFn {
|
|
return spc.simValue
|
|
}
|
|
|
|
// NewSimParamChange creates a new ParamChange instance
|
|
func NewSimParamChange(subspace, key string, simVal simulation.SimValFn) simulation.ParamChange {
|
|
return ParamChange{
|
|
subspace: subspace,
|
|
key: key,
|
|
simValue: simVal,
|
|
}
|
|
}
|
|
|
|
// ComposedKey creates a new composed key for the param change proposal
|
|
func (spc ParamChange) ComposedKey() string {
|
|
return fmt.Sprintf("%s/%s", spc.Subspace(), spc.Key())
|
|
}
|
|
|
|
//-----------------------------------------------------------------------------
|
|
// Proposal Contents
|
|
|
|
// WeightedProposalContent defines a common struct for proposal contents defined by
|
|
// external modules (i.e outside gov)
|
|
type WeightedProposalContent struct {
|
|
appParamsKey string // key used to retrieve the value of the weight from the simulation application params
|
|
defaultWeight int // default weight
|
|
contentSimulatorFn simulation.ContentSimulatorFn // content simulator function
|
|
}
|
|
|
|
func NewWeightedProposalContent(appParamsKey string, defaultWeight int, contentSimulatorFn simulation.ContentSimulatorFn) simulation.WeightedProposalContent {
|
|
return &WeightedProposalContent{appParamsKey: appParamsKey, defaultWeight: defaultWeight, contentSimulatorFn: contentSimulatorFn}
|
|
}
|
|
|
|
func (w WeightedProposalContent) AppParamsKey() string {
|
|
return w.appParamsKey
|
|
}
|
|
|
|
func (w WeightedProposalContent) DefaultWeight() int {
|
|
return w.defaultWeight
|
|
}
|
|
|
|
func (w WeightedProposalContent) ContentSimulatorFn() simulation.ContentSimulatorFn {
|
|
return w.contentSimulatorFn
|
|
}
|