ultron

type AttackStrategy interface {
	Spawn() []*RampUpStep
	Switch(next AttackStrategy) []*RampUpStep
	Split(int) []AttackStrategy
	Name() string
}

type AttackStrategyCommander interface {
	Open(context.Context, Task) <-chan statistics.AttackResult
	Command(AttackStrategy, Timer)
	ConcurrentUsers() int
	Close()
}

type Attacker interface {
	Name() string
	Fire(context.Context) error
}

type ExitConditions interface {
	NeverStop() bool
	Check(ExitConditions) bool
}

type FixedConcurrentUsers struct {
	ConcurrentUsers int `json:"concurrent_users"`         // 并发用户数
	RampUpPeriod    int `json:"ramp_up_period,omitempty"` // 增压周期时长
}

type GaussianRandomTimer struct {
	StdDev      float64 `json:"std_dev"`      // 标准差
	DesiredMean float64 `json:"desired_mean"` // 期望均值
}

type HTTPAttacker struct {
	// contains filtered or unexported fields
}

type HTTPAttackerOption func(*HTTPAttacker)

type HTTPCheckFunc func(context.Context, *http.Response, []byte) error

type HTTPPrepareFunc func(context.Context) (*http.Request, error)

type LocalRunner interface {
	Launch() error
	Assign(Task)
	SubscribeReport(...ReportHandleFunc)
	SubscribeResult(...ResultHandleFunc)
	StartPlan(Plan) error
	StopPlan()
}

type LoggerOption struct {
	Level      string `default:"info" yaml:"level,omitempty" json:"level,omitempty" toml:"level"`
	FileName   string `yaml:"filename,omitempty" json:"filename,omitempty" toml:"filename"`
	MaxSize    int    `default:"100" yaml:"max_size,omitempty" json:"max_size,omitempty" toml:"max_size"`
	MaxBackups int    `default:"30" yaml:"max_backups,omitempty" json:"max_backups,omitempty" toml:"max_backups"`
}

type MasterRunner interface {
	Launch(...grpc.ServerOption) error   // 服务启动
	StartPlan(Plan) error                // 开始执行某个测试计划
	StopPlan()                           // 停止当前计划
	SubscribeReport(...ReportHandleFunc) // 订阅聚合报告
}

type NonstopTimer struct{}

type Option struct {
	Server ServerOption
	Logger LoggerOption
}

type Plan interface {
	Name() string
	AddStages(...Stage)
	Stages() []Stage
	Current() (int, Stage)
	Status() PlanStatus
}

type PlanStatus int

type RampUpStep struct {
	N        int           // 增、降的数量，>0 为加压， <0为降压
	Interval time.Duration // 间隔时间
}

type ReportHandleFunc func(context.Context, statistics.SummaryReport)

type ResultHandleFunc func(context.Context, statistics.AttackResult)

type ServerOption struct {
	HTTPAddr string `default:":2017" yaml:"http_addr,omitempty" json:"http_addr,omitempty" toml:"http_addr"`
	GRPCAddr string `default:":2021" yaml:"grpc_addr,omitempty" json:"grpc_addr,omitempty" toml:"grpc_addr"`
}

type SlaveAgent interface {
	ID() string
	Extras() map[string]string
}

type SlaveRunner interface {
	Connect(string, ...grpc.DialOption) error // 连接master
	SubscribeResult(...ResultHandleFunc)      // 订阅Attacker的执行结果
	Assign(Task)                              // 指派压测任务
}

type Stage interface {
	GetTimer() Timer
	GetExitConditions() ExitConditions
	GetStrategy() AttackStrategy
}

type Task interface {
	Add(Attacker, uint32)
	PickUp() Attacker
}

type Timer interface {
	Sleep()
}

type UniformRandomTimer struct {
	MinWait time.Duration `json:"min_wait,omitempty"`
	MaxWait time.Duration `json:"max_wait,omitempty"`
}

type UniversalExitConditions struct {
	Requests uint64        `json:"requests,omitempty"` // 请求总数，不严格控制
	Duration time.Duration `json:"duration,omitempty"` // 持续时长，不严格控制
}

type V1StageConfig struct {
	Requests        uint64        `json:"requests,omitempty"`
	Duration        time.Duration `json:"duration,omitempty"`
	ConcurrentUsers int           `json:"concurrent_users"`
	RampUpPeriod    int           `json:"ramp_up_period"` // 单位秒
	MinWait         time.Duration `json:"min_wait,omitempty"`
	MaxWait         time.Duration `json:"max_wait,omitempty"`
}