loadgen

type CallResult struct {
	Failed     bool          `json:"failed,omitempty"`
	Timeout    bool          `json:"timeout,omitempty"`
	Duration   time.Duration `json:"duration"`
	StartedAt  *time.Time    `json:"started_at,omitempty"`
	FinishedAt *time.Time    `json:"finished_at,omitempty"`
	Data       interface{}   `json:"data,omitempty"`
	Error      string        `json:"error,omitempty"`
}

type Config struct {
	T          *testing.T
	LoadType   string
	Labels     map[string]string
	LokiConfig *client.LokiConfig
	Schedule   []*Segment

	StatsPollInterval time.Duration
	CallTimeout       time.Duration
	Gun               Gun
	Instance          Instance
	Logger            zerolog.Logger
	SharedData        interface{}
	// contains filtered or unexported fields
}

type Generator struct {
	Log zerolog.Logger

	ResponsesWaitGroup *sync.WaitGroup

	ResponsesCtx context.Context

	ResponsesChan chan CallResult
	// contains filtered or unexported fields
}

type Gun interface {
	Call(l *Generator) CallResult
}

type HTTPMockServer struct {
	Sleep time.Duration
	// contains filtered or unexported fields
}

type Instance interface {
	Run(l *Generator)
	Stop(l *Generator)
	Clone(l *Generator) Instance
}

type MockGun struct {
	Data []string
	// contains filtered or unexported fields
}

type MockGunConfig struct {
	// FailRatio in percentage, 0-100
	FailRatio int
	// TimeoutRatio in percentage, 0-100
	TimeoutRatio int
	// CallSleep time spent waiting inside a call
	CallSleep time.Duration
}

type MockHTTPGun struct {
	Data []string
	// contains filtered or unexported fields
}

type MockHTTPGunConfig struct {
	TargetURL string
}

type MockInstance struct {
	Data []string
	// contains filtered or unexported fields
}

type MockInstanceConfig struct {
	// FailRatio in percentage, 0-100
	FailRatio int
	// TimeoutRatio in percentage, 0-100
	TimeoutRatio int
	// CallSleep time spent waiting inside a call
	CallSleep time.Duration
}

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

type ResponseData struct {
	OKData []interface{}

	OKResponses []CallResult

	FailResponses []CallResult
	// contains filtered or unexported fields
}

type Segment struct {
	From         int64
	Increase     int64
	Steps        int64
	StepDuration time.Duration
	// contains filtered or unexported fields
}

type Stats struct {
	CurrentRPS       atomic.Int64 `json:"currentRPS"`
	CurrentInstances atomic.Int64 `json:"currentInstances"`
	LastSegment      atomic.Int64 `json:"last_segment"`
	CurrentSegment   atomic.Int64 `json:"current_schedule_segment"`
	CurrentStep      atomic.Int64 `json:"current_schedule_step"`
	RunStopped       atomic.Bool  `json:"runStopped"`
	RunFailed        atomic.Bool  `json:"runFailed"`
	Success          atomic.Int64 `json:"success"`
	Failed           atomic.Int64 `json:"failed"`
	CallTimeout      atomic.Int64 `json:"callTimeout"`
}

type WSMockConfig struct {
	TargetURl string
}

type WSMockInstance struct {
	Data []string
	// contains filtered or unexported fields
}