v1alpha1

type CleanPodPolicy string

type DGLJob struct {
	metav1.TypeMeta   `json:",inline"`
	metav1.ObjectMeta `json:"metadata,omitempty"`

	Spec   DGLJobSpec   `json:"spec,omitempty"`
	Status DGLJobStatus `json:"status,omitempty"`
}

type DGLJobList struct {
	metav1.TypeMeta `json:",inline"`
	metav1.ListMeta `json:"metadata,omitempty"`
	Items           []DGLJob `json:"items"`
}

type DGLJobSpec struct {
	SlotsPerWorker *int `json:"slotsPerWorker,omitempty"`

	// PartitionMode defines the policy how to partition. Defaults to DGL-API.
	// +kubebuilder:default:=DGL-API
	PartitionMode *PartitionMode `json:"partitionMode,omitempty"`

	// CleanPodPolicy defines the policy that whether to kill pods after the job completes. Defaults to None.
	// +kubebuilder:default:=Running
	CleanPodPolicy *CleanPodPolicy `json:"cleanPodPolicy,omitempty"`

	// DGLReplicaSpecs describes the spec of launcher or worker base on pod template
	DGLReplicaSpecs map[ReplicaType]*ReplicaSpec `json:"dglReplicaSpecs"`
}

type DGLJobStatus struct {
	Phase JobPhase `json:"phase,omitempty"`

	// ReplicaStatuses is map of ReplicaType and ReplicaStatus,
	// specifies the status of each replica.
	ReplicaStatuses map[ReplicaType]*ReplicaStatus `json:"replicaStatuses"`

	// Represents time when the job was acknowledged by the job controller.
	// It is not guaranteed to be set in happens-before order across separate operations.
	// It is represented in RFC3339 form and is in UTC.
	StartTime *metav1.Time `json:"startTime,omitempty"`

	// Represents time when the job was completed. It is not guaranteed to
	// be set in happens-before order across separate operations.
	// It is represented in RFC3339 form and is in UTC.
	CompletionTime *metav1.Time `json:"completionTime,omitempty"`

	// Represents last time when the job was reconciled. It is not guaranteed to
	// be set in happens-before order across separate operations.
	// It is represented in RFC3339 form and is in UTC.
	LastReconcileTime *metav1.Time `json:"lastReconcileTime,omitempty"`
}

type JobPhase string

type PartitionMode string

type ReplicaSpec struct {
	// Replicas replica
	Replicas *int `json:"replicas,omitempty"`

	// Template is the object that describes the pod that
	// will be created for this replica
	Template corev1.PodTemplateSpec `json:"template"`
}

type ReplicaStatus struct {
	// The number of actively running pods.
	Running int `json:"running,omitempty"`
	// The number of pending pods.
	Pending int `json:"pending,omitempty"`
	// The number of starting pods.
	Starting int `json:"starting,omitempty"`
	// The number of pods which reached phase Succeeded.
	Succeeded int `json:"succeeded,omitempty"`
	// The number of pods which reached phase Failed.
	Failed int `json:"failed,omitempty"`
	// Replica ready string
	Ready string `json:"ready,omitempty"`
}

type ReplicaType string