bencher

type Bencher struct {
	//Some sort of Logger/Writer
	//Worker pool
	Work Workload

	Strict bool
	// contains filtered or unexported fields
}

type BenchmarkConfig struct {
	OutputFile string         `json:"output" yaml:"output"`
	Workload   WorkloadConfig `json:"workload" yaml:"workload"`
}

type ConstantRate struct {
	TotalRequests uint64

	sync.RWMutex
	// contains filtered or unexported fields
}

type FixedRPSRate struct {
	RPS             int64
	BypassAtFailure bool
	// contains filtered or unexported fields
}

type FunctionAPIInvoker interface {
	Invoker
}

type HTTPInvoker struct {
	Request     *http.Request
	RequestBody []byte

	// DisableCompression is an option to disable compression in response
	DisableCompression bool

	// DisableKeepAlive is an option to prevents re-use of TCP connections between different HTTP requests
	DisableKeepAlive bool

	// DisableRedirects is an option to prevent the following of HTTP redirects
	DisableRedirects bool

	// H2 is an option to make HTTP/2 requests
	H2 bool

	// Timeout in seconds.
	Timeout int
	// contains filtered or unexported fields
}

type HatchRate interface {
	//function is called once before starting the phase
	Setup(context.Context, *Phase) (*sync.Cond, error)
	//function should block until the next request should be made
	Take() error
	//call if the invocation was successful
	OnSuccess()
	//call if the invocation failed
	OnFailed()
	//call if the invocation was queued
	OnQueued()

	Close() error
}

type HatchRateConfig struct {
	Type    string                 `yaml:"type"`
	Options map[string]interface{} `yaml:",inline"`
}

type HatchRateConstructor func(config HatchRateConfig) (HatchRate, error)

type Invoker interface {
	Setup(context.Context, *Phase, *Bencher) error
	Exec(rate HatchRate) error
}

type InvokerConfig struct {
	Options map[string]interface{} `yaml:",inline"`
	Type    string                 `yaml:"type"`
}

type InvokerConstructor func(config InvokerConfig) (Invoker, error)

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

type PayloadFunc func(Invoker) []byte

type Phase struct {
	Name        string        //unique name, might be used to identify this phase
	Threads     int           //number of threads that send the invocation
	HatchRate   HatchRate     //hatch rate, that governs how often new invocations are performed
	Timeout     time.Duration //max duration of this phase
	Target      string        //the target of this workload, can be an url or platfrom identifier (e.g. function name)
	PayloadFunc PayloadFunc   //if set this function is called during _each_ invocation to generate a payload, use for authentication
	PreRun      PreRunFunc    //if set this function will be called _once_ before running the phase
	PostRun     PostRunFunc   //if set this function will be called _once_ after running the phase
	Invocation  Invoker       //the invocation of this phase, e.g. HTTP or CLI
}

type PhaseConfig struct {
	Name      string          `json:"name" yaml:"name"`
	Threads   int             `json:"threads" yaml:"threads"`
	HatchRate HatchRateConfig `json:"hatchRate" yaml:"hatchRate"`
	Timeout   time.Duration   `json:"timeout" yaml:"timeout"`
}

type PostRunFunc func() error

type PreRunFunc func() error

type SlopingRate struct {
	StartRate       int64
	HatchRate       float64
	BypassAtFailure bool
	// contains filtered or unexported fields
}

type WhiskInvoker struct {
	FunctionName string

	RequestPerMinute int64

	Host  string
	Token string

	Request interface{}
	// contains filtered or unexported fields
}

type Workload struct {
	Name    string
	Target  string      //the target of this workload, can be an url or platfrom identifier (e.g. function name)
	PreRun  PreRunFunc  //if set this function will be called once before the benchmark
	Phases  []Phase     //phases of the workload
	PostRun PostRunFunc //if set this function will be called once after the benchmark
}

type WorkloadConfig struct {
	Name       string        `json:"name" yaml:"name"`
	Target     string        `json:"target" yaml:"target"`
	Phases     []PhaseConfig `json:"phases" yaml:"phases"`
	Invocation InvokerConfig `json:"invoker" yaml:"invoker"`
}