fix: strings.CutPrefix pool label

Author: 0x5457

internal/controller/gpupool_controller.go

@@ -284,13 +284,10 @@ func (r *GPUPoolReconciler) SetupWithManager(mgr ctrl.Manager) error {
 
 			node := obj.(*tfv1.GPUNode)
 			for labelKey := range node.Labels {
-				if strings.HasPrefix(labelKey, constants.GPUNodePoolIdentifierLabelPrefix) {
-					tmp := strings.Split(labelKey, "/")
-					if len(tmp) != 3 {
-						continue
-					}
+				poolName, ok := strings.CutPrefix(labelKey, constants.GPUNodePoolIdentifierLabelPrefix)
+				if ok {
 					requests = append(requests, reconcile.Request{
-						NamespacedName: types.NamespacedName{Name: tmp[2]},
+						NamespacedName: types.NamespacedName{Name: poolName},
 					})
 				}
 			}

internal/controller/gpupool_controller_test.go

@@ -71,7 +71,6 @@ var _ = Describe("GPUPool Controller", func() {
 					Name: nodeNamespacedName.Name,
 					Labels: map[string]string{
 						fmt.Sprintf(constants.GPUNodePoolIdentifierLabelFormat, resourceName): "true",
-						"mock-label": "true",
 					},
 				},
 				Spec: tfv1.GPUNodeSpec{