Skip to content

Commit

Permalink
CR fixes
Browse files Browse the repository at this point in the history
Signed-off-by: Arnon Gilboa <agilboa@redhat.com>
  • Loading branch information
arnongilboa committed Apr 27, 2023
1 parent 38c569c commit a202278
Show file tree
Hide file tree
Showing 3 changed files with 25 additions and 26 deletions.
33 changes: 20 additions & 13 deletions pkg/controller/datavolume/controller-base.go
Original file line number Diff line number Diff line change
Expand Up @@ -192,6 +192,16 @@ func getIndexArgs() []indexArgs {
return nil
},
},
{
obj: &corev1.PersistentVolume{},
field: claimStorageClassNameField,
extractValue: func(obj client.Object) []string {
if pv, ok := obj.(*corev1.PersistentVolume); ok && pv.Status.Phase == corev1.VolumeAvailable {
return []string{pv.Spec.StorageClassName}
}
return nil
},
},
}
}

Expand All @@ -206,14 +216,6 @@ func CreateCommonIndexes(mgr manager.Manager) error {
return err
}
}
if err := mgr.GetFieldIndexer().IndexField(context.TODO(), &corev1.PersistentVolume{}, claimStorageClassNameField, func(obj client.Object) []string {
if pv, ok := obj.(*corev1.PersistentVolume); ok {
return []string{pv.Spec.StorageClassName}
}
return nil
}); err != nil {
return err
}
return nil
}

Expand Down Expand Up @@ -322,6 +324,7 @@ func addDataVolumeControllerCommonWatches(mgr manager.Manager, dataVolumeControl
if storage != nil &&
storage.StorageClassName != nil &&
*storage.StorageClassName == pv.Spec.StorageClassName &&
pv.Status.Phase == corev1.VolumeAvailable &&
getDataVolumeOp(mgr.GetLogger(), &dv, mgr.GetClient()) == op {
reqs = append(reqs, reconcile.Request{NamespacedName: types.NamespacedName{Name: dv.Name, Namespace: dv.Namespace}})
}
Expand Down Expand Up @@ -446,11 +449,15 @@ func (r *ReconcilerBase) syncDvPvcState(log logr.Logger, req reconcile.Request,
}
}

syncState.pvcSpec, err = renderPvcSpec(r.client, r.recorder, log, dv)
if err != nil {
r.syncDataVolumeStatusPhaseWithEvent(&syncState, cdiv1.PhaseUnset, nil,
Event{corev1.EventTypeWarning, cc.ErrClaimNotValid, err.Error()})
return syncState, err
if syncState.pvc == nil {
syncState.pvcSpec, err = renderPvcSpec(r.client, r.recorder, log, dv)
if err != nil {
r.syncDataVolumeStatusPhaseWithEvent(&syncState, cdiv1.PhaseUnset, nil,
Event{corev1.EventTypeWarning, cc.ErrClaimNotValid, err.Error()})
return syncState, err
}
} else {
syncState.pvcSpec = syncState.pvc.Spec.DeepCopy()
}

if err := r.handleStaticVolume(&syncState, log); err != nil || syncState.result != nil {
Expand Down
14 changes: 3 additions & 11 deletions pkg/controller/datavolume/util.go
Original file line number Diff line number Diff line change
Expand Up @@ -78,7 +78,7 @@ func pvcFromStorage(client client.Client, recorder record.EventRecorder, log log
}
if storageClass == nil {
// If SC was not found, look for PV with this SC name
pv, err := getPV(client, pvcSpec, dv.Name, dv.Namespace)
pv, err := getAvailablePV(client, pvcSpec, dv.Name, dv.Namespace)
if err != nil {
return nil, err
}
Expand Down Expand Up @@ -160,8 +160,8 @@ func copyStorageAsPvc(log logr.Logger, storage *cdiv1.StorageSpec) *v1.Persisten
return pvcSpec
}

// Gets either the bound PV or an available satisfying PV
func getPV(c client.Client, pvcSpec *v1.PersistentVolumeClaimSpec, pvcName, pvcNamespace string) (*corev1.PersistentVolume, error) {
// Gets an available satisfying PV
func getAvailablePV(c client.Client, pvcSpec *v1.PersistentVolumeClaimSpec, pvcName, pvcNamespace string) (*corev1.PersistentVolume, error) {
if pvcSpec.StorageClassName == nil {
return nil, nil
}
Expand All @@ -171,14 +171,6 @@ func getPV(c client.Client, pvcSpec *v1.PersistentVolumeClaimSpec, pvcName, pvcN
if err := c.List(context.TODO(), pvList, fields); err != nil {
return nil, err
}
for _, pv := range pvList.Items {
if pv.Status.Phase == corev1.VolumeBound &&
pv.Spec.ClaimRef != nil &&
pv.Spec.ClaimRef.Name == pvcName &&
pv.Spec.ClaimRef.Namespace == pvcNamespace {
return &pv, nil
}
}
for _, pv := range pvList.Items {
if pv.Status.Phase == corev1.VolumeAvailable {
pvc := &corev1.PersistentVolumeClaim{Spec: *pvcSpec}
Expand Down
4 changes: 2 additions & 2 deletions tests/datavolume_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -2421,7 +2421,7 @@ var _ = Describe("[vendor:cnv-qe@redhat.com][level:component]DataVolume tests",
err = utils.WaitForPersistentVolumeClaimPhase(f.K8sClient, dataVolume.Namespace, v1.ClaimBound, dataVolume.Name)
Expect(err).ToNot(HaveOccurred())

By("waiting for dv succeeeded")
By("waiting for dv succeeded")
err = utils.WaitForDataVolumePhase(f, dataVolume.Namespace, cdiv1.Succeeded, dataVolume.Name)
Expect(err).ToNot(HaveOccurred())
},
Expand Down Expand Up @@ -2479,7 +2479,7 @@ var _ = Describe("[vendor:cnv-qe@redhat.com][level:component]DataVolume tests",
err = utils.WaitForPersistentVolumeClaimPhase(f.K8sClient, dataVolume.Namespace, v1.ClaimBound, dataVolume.Name)
Expect(err).ToNot(HaveOccurred())

By("waiting for dv succeeeded")
By("waiting for dv succeeded")
err = utils.WaitForDataVolumePhase(f, dataVolume.Namespace, cdiv1.Succeeded, dataVolume.Name)
Expect(err).ToNot(HaveOccurred())
},
Expand Down

0 comments on commit a202278

Please sign in to comment.