common

type ControllerInterface interface {
	// GetNativeJob returns the native job object, or nil if not found.
	GetNativeJob(context.Context, types.NamespacedName) (interface{}, error)
	// GetAIMasterPodSpec tries to find AIMaster pod spec from the job's replicas spec.
	GetAIMasterPodSpec(interface{}) *corev1.PodTemplateSpec
	// PreprocessJobSpec updates the job object for enabling some feature gates.
	PreprocessJobSpec(context.Context, interface{}) (bool, error)
	// ReconcileNativeJob reconciles the native job object.
	ReconcileNativeJob(context.Context, interface{}, interface{}, string, *common.JobStatus, logr.Logger) error
	// DeleteJob deletes the job object.
	DeleteJob(context.Context, interface{}) error
	// RemoveNativeJobFinalizer removes the finalizer from the native job object.
	RemoveNativeJobFinalizer(context.Context, interface{}) error
	// UpdateStatus updates the Status of the job object.
	UpdateStatus(context.Context, interface{}, *common.JobStatus) error
	// GenLabels generates the labels that the native job operator adds to the pods.
	GenLabels(string) map[string]string
	// GetDefaultContainerName returns the default container name for the specific job CRD.
	GetDefaultContainerName() string
	// GetDefaultContainerPortName returns the default container port name for the specific job CRD.
	GetDefaultContainerPortName() string
	// GetDefaultPort returns the default port for the specific job CRD.
	GetDefaultPort() int32
}

type GangScheduler string

type JobController struct {
	client.Client
	Scheme     *runtime.Scheme
	Log        logr.Logger
	Recorder   record.EventRecorder
	Controller ControllerInterface
	Config     JobControllerConfiguration
}

type JobControllerConfiguration struct {
	// Enable gang scheduling by volcano
	EnableGangScheduling bool
}