add queuing (#7)

Author: denkensk

go.mod

@@ -8,7 +8,6 @@ require (
 	k8s.io/apimachinery v0.21.2
 	k8s.io/client-go v0.21.2
 	k8s.io/code-generator v0.21.2
-	k8s.io/klog v1.0.0
 	k8s.io/klog/v2 v2.8.0
 	k8s.io/kube-openapi v0.0.0-20200410145947-61e04a5be9a6
 	k8s.io/sample-controller v0.21.2

pkg/contorller/contoller.go

@@ -43,6 +43,7 @@ const (
 	// QuNameSuffix is the suffix of the queue unit name when create a new one.
 	// In this way, different types of jobs with the same name will create different queue unit name.
 	QuNameSuffix = "-pytorch-qu"
+	Queuing      = "Queuing"
 )
 
 type PyTorchExtensionController struct {
@@ -358,6 +359,19 @@ func (pc *PyTorchExtensionController) AddPyTorchJob(obj interface{}) {
 	if err != nil {
 		klog.Errorf("Can't create queueunit for pytorchjob %v/%v,err is:%v", pytorchJob.Namespace, pytorchJob.Name, err)
 	}
+
+	if pytorchJob.Status.Conditions == nil {
+		pytorchJob.Status.Conditions = make([]commonv1.JobCondition, 0)
+		pytorchJob.Status.Conditions = append(pytorchJob.Status.Conditions, commonv1.JobCondition{
+			Type:           Queuing,
+			LastUpdateTime: metav1.Now(),
+		})
+		_, err := pc.pytorchJobClient.KubeflowV1().PyTorchJobs(pytorchJob.Namespace).UpdateStatus(context.TODO(), pytorchJob, metav1.UpdateOptions{})
+		if err != nil {
+			klog.Errorf("update pytorchJob failed Queuing %v/%v %v", pytorchJob.Namespace, pytorchJob.Name, err.Error())
+		}
+		klog.Infof("update pytorchJob %v/%v status Queuing successfully", pytorchJob.Namespace, pytorchJob.Name)
+	}
 }
 
 func (pc *PyTorchExtensionController) UpdatePyTorchJob(_, newObj interface{}) {