fix: detect used by bug (#284)

Author: Code2Life

internal/constants/env.go

@@ -119,7 +119,7 @@ const (
 	HypervisorListenAddrEnv         = "API_LISTEN_ADDR"
 	HypervisorMetricsFormatEnv      = "TF_HYPERVISOR_METRICS_FORMAT"
 	HypervisorMetricsExtraLabelsEnv = "TF_HYPERVISOR_METRICS_EXTRA_LABELS"
-	HypervisorDetectUsedGPUEnv      = "DETECT_IN_USED_GPU"
+	HypervisorDetectUsedGPUEnv      = "DETECT_IN_USED_GPUS"
 
 	HypervisorDefaultPortNumber int32  = 8000
 	HypervisorPortName          string = "http"

internal/controller/gpu_controller.go

@@ -88,6 +88,15 @@ func (r *GPUReconciler) Reconcile(ctx context.Context, req ctrl.Request) (ctrl.R
 		return ctrl.Result{}, fmt.Errorf("node %s is not assigned to any pool", gpunode.Name)
 	}
 
+	if gpu.Status.UsedBy == "" {
+		patch := client.MergeFrom(gpu.DeepCopy())
+		gpu.Status.UsedBy = tfv1.UsedByTensorFusion
+		if err := r.Status().Patch(ctx, gpu, patch); err != nil {
+			return ctrl.Result{}, fmt.Errorf("patch gpu %s: %w", gpu.Name, err)
+		}
+		return ctrl.Result{}, nil
+	}
+
 	// No need to calculate patch since GPU's owner pool not changed
 	if gpu.Labels != nil && gpu.Labels[constants.GpuPoolKey] == poolName {
 		return ctrl.Result{}, nil
@@ -98,13 +107,10 @@ func (r *GPUReconciler) Reconcile(ctx context.Context, req ctrl.Request) (ctrl.R
 		gpu.Labels = make(map[string]string)
 	}
 	gpu.Labels[constants.GpuPoolKey] = poolName
-	if gpu.Status.UsedBy == "" {
-		gpu.Status.UsedBy = tfv1.UsedByTensorFusion
-	}
-
 	if err := r.Patch(ctx, gpu, patch); err != nil {
 		return ctrl.Result{}, fmt.Errorf("patch gpu %s: %w", gpu.Name, err)
 	}
+
 	return ctrl.Result{}, nil
 }