【修改】k8s 配置

apps/devops/services/k8s/pv/pv_common.go

@@ -0,0 +1,94 @@
+package pv
+
+import (
+	"context"
+	"fmt"
+
+	v1 "k8s.io/api/core/v1"
+	metaV1 "k8s.io/apimachinery/pkg/apis/meta/v1"
+	client "k8s.io/client-go/kubernetes"
+	k8scommon "pandax/apps/devops/services/k8s/common"
+	"pandax/apps/devops/services/k8s/dataselect"
+	"strings"
+)
+
+// GetStorageClassPersistentVolumes gets persistentvolumes that are associated with this storageclass.
+func GetStorageClassPersistentVolumes(client client.Interface, storageClassName string,
+	dsQuery *dataselect.DataSelectQuery) (*PersistentVolumeList, error) {
+
+	storageClass, err := client.StorageV1().StorageClasses().Get(context.TODO(), storageClassName, metaV1.GetOptions{})
+
+	if err != nil {
+		return nil, err
+	}
+
+	channels := &k8scommon.ResourceChannels{
+		PersistentVolumeList: k8scommon.GetPersistentVolumeListChannel(
+			client, 1),
+	}
+
+	persistentVolumeList := <-channels.PersistentVolumeList.List
+
+	err = <-channels.PersistentVolumeList.Error
+	if err != nil {
+		return nil, err
+	}
+
+	storagePersistentVolumes := make([]v1.PersistentVolume, 0)
+	for _, pv := range persistentVolumeList.Items {
+		if strings.Compare(pv.Spec.StorageClassName, storageClass.Name) == 0 {
+			storagePersistentVolumes = append(storagePersistentVolumes, pv)
+		}
+	}
+
+	global.Log.Info(fmt.Sprintf("Found %d persistentvolumes related to %s storageclass", len(storagePersistentVolumes), storageClassName))
+
+	return toPersistentVolumeList(storagePersistentVolumes, dsQuery), nil
+}
+
+// getPersistentVolumeClaim returns Persistent Volume claim using "namespace/claim" format.
+func getPersistentVolumeClaim(pv *v1.PersistentVolume) string {
+	var claim string
+
+	if pv.Spec.ClaimRef != nil {
+		claim = pv.Spec.ClaimRef.Namespace + "/" + pv.Spec.ClaimRef.Name
+	}
+
+	return claim
+}
+
+// PersistentVolumeCell allows to perform complex data section on []api.PersistentVolume.
+type PersistentVolumeCell v1.PersistentVolume
+
+// GetProperty allows to perform complex data section on PersistentVolumeCell.
+func (self PersistentVolumeCell) GetProperty(name dataselect.PropertyName) dataselect.ComparableValue {
+	switch name {
+	case dataselect.NameProperty:
+		return dataselect.StdComparableString(self.ObjectMeta.Name)
+	case dataselect.CreationTimestampProperty:
+		return dataselect.StdComparableTime(self.ObjectMeta.CreationTimestamp.Time)
+	case dataselect.NamespaceProperty:
+		return dataselect.StdComparableString(self.ObjectMeta.Namespace)
+	default:
+		// if name is not supported then just return a constant dummy value, sort will have no effect.
+		return nil
+	}
+}
+
+// toCells converts []api.PersistentVolume to []dataselect.DataCell.
+func toCells(std []v1.PersistentVolume) []dataselect.DataCell {
+	cells := make([]dataselect.DataCell, len(std))
+	for i := range std {
+		cells[i] = PersistentVolumeCell(std[i])
+	}
+	return cells
+}
+
+// fromCells converts cells []dataselect.DataCell to []api.PersistentVolume.
+func fromCells(cells []dataselect.DataCell) []v1.PersistentVolume {
+	std := make([]v1.PersistentVolume, len(cells))
+	for i := range std {
+		std[i] = v1.PersistentVolume(cells[i].(PersistentVolumeCell))
+	}
+	return std
+}