configuration

type ActionsConfig struct {
	JobSetReprioritisations []JobSetReprioritisation `yaml:"jobSetReprioritisations"`
	JobSetCancellations     []JobSetCancellation     `yaml:"jobSetCancellations"`
}

type DatabaseConfig struct {
	Postgres                map[string]string `yaml:"postgres,omitempty"`
	ClickHouse              map[string]string `yaml:"clickHouse,omitempty"`
	InMemory                bool              `yaml:"inMemory,omitempty"`
	PostgresTuningSQL       []string          `yaml:"postgresTuningSQL,omitempty"`
	PostgresTuningRevertSQL []string          `yaml:"postgresTuningRevertSQL,omitempty"`
}

type FeatureToggles struct {
	HotColdSplit bool `yaml:"hotColdSplit,omitempty"`
}

type HistoricalJobsConfig struct {
	NumberOfJobs        int     `yaml:"numberOfJobs"`
	ProportionSucceeded float64 `yaml:"proportionSucceeded"`
	ProportionErrored   float64 `yaml:"proportionErrored"`
	ProportionCancelled float64 `yaml:"proportionCancelled"`
	ProportionPreempted float64 `yaml:"proportionPreempted"`
	JobAgeDays          []int   `yaml:"jobAgeDays"`
}

type IngestionConfig struct {
	BatchSize                   int                      `yaml:"batchSize"`
	NumWorkers                  int                      `yaml:"numWorkers"`
	SubmissionsPerHour          int                      `yaml:"submissionsPerHour"`
	ChannelBufferSizeMultiplier int                      `yaml:"channelBufferSizeMultiplier,omitempty"`
	MaxBacklogSize              int                      `yaml:"maxBacklogSize,omitempty"`
	BacklogDropStrategy         string                   `yaml:"backlogDropStrategy,omitempty"`
	BatchTimeout                time.Duration            `yaml:"batchTimeout,omitempty"`
	HistoricalJobChunkSize      int                      `yaml:"historicalJobChunkSize,omitempty"`
	HistoricalJobWorkers        int                      `yaml:"historicalJobWorkers,omitempty"`
	JobStateTransitionConfig    JobStateTransitionConfig `yaml:"jobStateTransitionConfig"`
}

type JobSetCancellation struct {
	PerformAfterTestStartTime time.Duration `yaml:"performAfterTestStartTime"`
	Queue                     string        `yaml:"queue"`
	JobSet                    string        `yaml:"jobSet"`
}

type JobSetConfig struct {
	Name                 string               `yaml:"name"`
	Proportion           float64              `yaml:"proportion"`
	HistoricalJobsConfig HistoricalJobsConfig `yaml:"historicalJobsConfig"`
}

type JobSetReprioritisation struct {
	PerformAfterTestStartTime time.Duration `yaml:"performAfterTestStartTime"`
	Queue                     string        `yaml:"queue"`
	JobSet                    string        `yaml:"jobSet"`
}

type JobStateTransitionConfig struct {
	QueueingDuration         time.Duration `yaml:"queueingDuration"`
	LeasedDuration           time.Duration `yaml:"leasedDuration"`
	PendingDuration          time.Duration `yaml:"pendingDuration"`
	ProportionSucceed        float64       `yaml:"proportionSucceed"`
	RunningToSuccessDuration time.Duration `yaml:"runningToSuccessDuration"`
	ProportionFail           float64       `yaml:"proportionFail"`
	RunningToFailureDuration time.Duration `yaml:"runningToFailureDuration"`
}

type QueryConfig struct {
	GetJobRunDebugMessageQueriesPerHour int `yaml:"getJobRunDebugMessageQueriesPerHour"`
	GetJobRunErrorQueriesPerHour        int `yaml:"getJobRunErrorQueriesPerHour"`
	GetJobSpecQueriesPerHour            int `yaml:"getJobSpecQueriesPerHour"`
	GetJobsQueriesPerHour               int `yaml:"getJobsQueriesPerHour"`
	GetJobsPageSize                     int `yaml:"getJobsPageSize"`
	GetJobGroupsQueriesPerHour          int `yaml:"getJobGroupsQueriesPerHour"`
	GetJobGroupsPageSize                int `yaml:"getJobGroupsPageSize"`
	MaxConcurrentQueries                int `yaml:"maxConcurrentQueries,omitempty"`
	MaxErrorsToCollect                  int `yaml:"maxErrorsToCollect,omitempty"`
}

type QueueConfig struct {
	Name         string         `yaml:"name"`
	Proportion   float64        `yaml:"proportion"`
	JobSetConfig []JobSetConfig `yaml:"jobSetConfig"`
}

type TestConfig struct {
	FeatureToggles  FeatureToggles  `yaml:"featureToggles,omitempty"`
	TestDuration    time.Duration   `yaml:"testDuration"`
	WarmupDuration  time.Duration   `yaml:"warmupDuration"`
	SkipTearDown    bool            `yaml:"skipTearDown,omitempty"`
	DatabaseConfig  DatabaseConfig  `yaml:"databaseConfig"`
	QueueConfig     []QueueConfig   `yaml:"queueConfig"`
	IngestionConfig IngestionConfig `yaml:"ingestionConfig"`
	QueryConfig     QueryConfig     `yaml:"queryConfig"`
	ActionsConfig   ActionsConfig   `yaml:"actionsConfig"`
}