conf

type PluginOption struct {
	// The name of Plugin
	Name string `yaml:"name" json:"name"`
	// JobOrderDisabled defines whether jobOrderFn is disabled
	JobOrderDisabled bool `yaml:"disableJobOrder" json:"disableJobOrder"`
	// TaskOrderDisabled defines whether taskOrderFn is disabled
	TaskOrderDisabled bool `yaml:"disableTaskOrder" json:"disableTaskOrder"`
	// PreemptableDisabled defines whether preemptableFn is disabled
	PreemptableDisabled bool `yaml:"disablePreemptable" json:"disablePreemptable"`
	// ReclaimableDisabled defines whether reclaimableFn is disabled
	ReclaimableDisabled bool `yaml:"disableReclaimable" json:"disableReclaimable"`
	// QueueOrderDisabled defines whether queueOrderFn is disabled
	QueueOrderDisabled bool `yaml:"disableQueueOrder" json:"disableQueueOrder"`
	// PredicateDisabled defines whether predicateFn is disabled
	PredicateDisabled bool `yaml:"disablePredicate" json:"disablePredicate"`
	// NodeOrderDisabled defines whether NodeOrderFn is disabled
	NodeOrderDisabled bool `yaml:"disableNodeOrder" json:"disableNodeOrder"`
	// Arguments defines the different arguments that can be given to different plugins
	Arguments map[string]string `yaml:"arguments" json:"arguments"`
}

type SchedulerConfiguration struct {
	// Actions defines the actions list of scheduler in order
	Actions string `yaml:"actions" json:"actions"`
	// Tiers defines plugins in different tiers
	Tiers []Tier `yaml:"tiers,omitempty" json:"tiers,omitempty"`
	// QueueDepthPerAction max number of jobs to try for action per queue
	QueueDepthPerAction map[string]int `yaml:"queueDepthPerAction,omitempty" json:"queueDepthPerAction,omitempty"`
}

type SchedulerParams struct {
	SchedulerName                     string                    `json:"schedulerName,omitempty"`
	RestrictSchedulingNodes           bool                      `json:"restrictSchedulingNodes,omitempty"`
	PartitionParams                   *SchedulingNodePoolParams `json:"partitionParams,omitempty"`
	IsInferencePreemptible            bool                      `json:"isInferencePreemptible,omitempty"`
	MaxNumberConsolidationPreemptees  int                       `json:"maxNumberConsolidationPreemptees,omitempty"`
	ScheduleCSIStorage                bool                      `json:"scheduleCSIStorage,omitempty"`
	UseSchedulingSignatures           bool                      `json:"useSchedulingSignatures,omitempty"`
	FullHierarchyFairness             bool                      `json:"fullHierarchyFairness,omitempty"`
	NodeLevelScheduler                bool                      `json:"nodeLevelScheduler,omitempty"`
	AllowConsolidatingReclaim         bool                      `json:"allowConsolidatingReclaim,omitempty"`
	NumOfStatusRecordingWorkers       int                       `json:"numOfStatusRecordingWorkers,omitempty"`
	GlobalDefaultStalenessGracePeriod time.Duration             `json:"globalDefaultStalenessGracePeriod,omitempty"`
	SchedulePeriod                    time.Duration             `json:"schedulePeriod,omitempty"`
	DetailedFitErrors                 bool                      `json:"detailedFitErrors,omitempty"`
}

type SchedulingNodePoolParams struct {
	NodePoolLabelKey   string
	NodePoolLabelValue string
}

type Tier struct {
	Plugins []PluginOption `yaml:"plugins" json:"plugins"`
}