68 lines
1.8 KiB
Go
68 lines
1.8 KiB
Go
package models
|
|
|
|
import (
|
|
v12 "k8s.io/api/core/v1"
|
|
"k8s.io/apimachinery/pkg/api/resource"
|
|
"k8s.io/apimachinery/pkg/apis/meta/v1"
|
|
|
|
"kubesphere.io/kubesphere/pkg/client"
|
|
)
|
|
|
|
type PvcListBySc struct {
|
|
Name string `json:"name"`
|
|
Claims []v12.PersistentVolumeClaim `json:"items"`
|
|
}
|
|
|
|
type ScMetrics struct {
|
|
Name string `json:"name"`
|
|
Metrics StorageMetrics `json:"metrics"`
|
|
}
|
|
|
|
type StorageMetrics struct {
|
|
Capacity string `json:"capacity,omitempty"`
|
|
Usage string `json:"usage,omitempty"`
|
|
}
|
|
|
|
// List persistent volume claims of a specific storage class
|
|
func GetPvcListBySc(storageclass string) (res []v12.PersistentVolumeClaim, err error) {
|
|
// Create Kubernetes client
|
|
cli := client.NewK8sClient()
|
|
// Get all persistent volume claims
|
|
claimList, err := cli.CoreV1().PersistentVolumeClaims("").List(v1.ListOptions{})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
// Select persistent volume claims which
|
|
// storage class name is equal to the specific storage class.
|
|
for _, claim := range claimList.Items {
|
|
if claim.Spec.StorageClassName != nil &&
|
|
*claim.Spec.StorageClassName == storageclass {
|
|
res = append(res, claim)
|
|
} else {
|
|
continue
|
|
}
|
|
}
|
|
return res, nil
|
|
}
|
|
|
|
// Get metrics of a specific storage class
|
|
func GetScMetrics(storageclass string) (res StorageMetrics, err error) {
|
|
// Create Kubernetes client
|
|
cli := client.NewK8sClient()
|
|
// Get persistent volumes
|
|
pvList, err := cli.CoreV1().PersistentVolumes().List(v1.ListOptions{})
|
|
if err != nil {
|
|
return StorageMetrics{}, err
|
|
}
|
|
|
|
var total resource.Quantity
|
|
// Gathering metrics of a specific storage class
|
|
for _, volume := range pvList.Items {
|
|
if volume.Spec.StorageClassName != storageclass {
|
|
continue
|
|
}
|
|
total.Add(volume.Spec.Capacity[v12.ResourceStorage])
|
|
}
|
|
return StorageMetrics{Usage: total.String()}, nil
|
|
}
|