From 2e34bdf9a2a1cfc3428a6ce18d83fb55d8290cb8 Mon Sep 17 00:00:00 2001 From: f10atin9 Date: Thu, 3 Mar 2022 15:55:32 +0800 Subject: [PATCH] Add "snapshot-count" annotation for volumesnapshotClass Signed-off-by: f10atin9 --- .../volumesnapshotclass.go | 23 +++++++++++++++++++ 1 file changed, 23 insertions(+) diff --git a/pkg/models/resources/v1alpha3/volumesnapshotclass/volumesnapshotclass.go b/pkg/models/resources/v1alpha3/volumesnapshotclass/volumesnapshotclass.go index 7e480aac6..e9982a54e 100644 --- a/pkg/models/resources/v1alpha3/volumesnapshotclass/volumesnapshotclass.go +++ b/pkg/models/resources/v1alpha3/volumesnapshotclass/volumesnapshotclass.go @@ -17,8 +17,11 @@ limitations under the License. package volumesnapshotclass import ( + "strconv" "strings" + "k8s.io/apimachinery/pkg/labels" + v1 "github.com/kubernetes-csi/external-snapshotter/client/v4/apis/volumesnapshot/v1" "github.com/kubernetes-csi/external-snapshotter/client/v4/informers/externalversions" "k8s.io/apimachinery/pkg/runtime" @@ -53,6 +56,11 @@ func (v *volumeSnapshotClassGetter) List(namespace string, query *query.Query) ( var result []runtime.Object for _, snapshotClass := range all { + count := v.countVolumeSnapshots(snapshotClass.Name) + if snapshotClass.Annotations == nil { + snapshotClass.Annotations = make(map[string]string) + } + snapshotClass.Annotations["kubesphere.io/snapshot-count"] = strconv.Itoa(count) result = append(result, snapshotClass) } @@ -86,3 +94,18 @@ func (v *volumeSnapshotClassGetter) filter(object runtime.Object, filter query.F return v1alpha3.DefaultObjectMetaFilter(snapshotClass.ObjectMeta, filter) } } + +func (v *volumeSnapshotClassGetter) countVolumeSnapshots(name string) int { + snapshots, err := v.informers.Snapshot().V1().VolumeSnapshots().Lister().VolumeSnapshots("").List(labels.Everything()) + if err != nil { + return 0 + } + var count int + + for _, snapshot := range snapshots { + if snapshot.Spec.VolumeSnapshotClassName != nil && *snapshot.Spec.VolumeSnapshotClassName == name { + count++ + } + } + return count +}