Allow to enable the PVC (#76)

Author: CsongSY

pkg/client/k8s/pvc.go

@@ -28,6 +28,14 @@ func (c *Client) DeletePVC(pvc *corev1.PersistentVolumeClaim) error {
 	return nil
 }
 
+func (c *Client) ExpandPVC(pvc *corev1.PersistentVolumeClaim) error {
+	if err := c.client.Update(ctx, pvc); err != nil {
+		return err
+	}
+	c.logger.V(1).Info("expand pvc size successfully", "pvc", pvc.Name)
+	return nil
+}
+
 func (c *Client) ListPVC(namespace string, labels map[string]string) (*corev1.PersistentVolumeClaimList, error) {
 	var pvcList corev1.PersistentVolumeClaimList
 	if err := c.client.List(ctx, &pvcList, k8sApiClient.InNamespace(namespace), k8sApiClient.MatchingLabels(labels)); err != nil {

pkg/controllers/cluster/handler.go

@@ -85,6 +85,10 @@ func (h *KVRocksClusterHandler) Handle() (error, bool) {
 	if err != nil || h.requeue {
 		return err, false
 	}
+	err = h.expandPersistentVolumeClaim()
+	if err != nil || h.requeue {
+		return err, false
+	}
 	return nil, true
 }
 

pkg/controllers/cluster/kubernetes.go

@@ -2,6 +2,7 @@ package cluster
 
 import (
 	"fmt"
+	v1 "k8s.io/api/core/v1"
 	"sort"
 	"strconv"
 	"strings"
@@ -317,3 +318,20 @@ func (h *KVRocksClusterHandler) cleanPersistentVolumeClaim() error {
 	}
 	return nil
 }
+
+func (h *KVRocksClusterHandler) expandPersistentVolumeClaim() error {
+	pvcList, err := h.k8s.ListStatefulSetPVC(h.key)
+	if err != nil {
+		return err
+	}
+	for _, pvc := range pvcList.Items {
+		pvcSize := pvc.Spec.Resources.Requests[v1.ResourceStorage]
+		if pvcSize.Cmp(h.instance.Spec.Storage.Size) < 0 {
+			pvc.Spec.Resources.Requests[v1.ResourceStorage] = h.instance.Spec.Storage.Size
+			if err := h.k8s.ExpandPVC(&pvc); err != nil {
+				return err
+			}
+		}
+	}
+	return nil
+}

pkg/controllers/standard/handler.go

@@ -56,6 +56,10 @@ func (h *KVRocksStandardHandler) Handle() (error, bool) {
 	if err != nil || h.requeue {
 		return err, false
 	}
+	err = h.expandPersistentVolumeClaim()
+	if err != nil || h.requeue {
+		return err, false
+	}
 	return nil, true
 }
 

pkg/controllers/standard/kubernetes.go

@@ -1,6 +1,7 @@
 package standard
 
 import (
+	v1 "k8s.io/api/core/v1"
 	"sort"
 
 	"k8s.io/apimachinery/pkg/api/errors"
@@ -141,3 +142,20 @@ func (h *KVRocksStandardHandler) cleanPersistentVolumeClaim() error {
 	}
 	return nil
 }
+
+func (h *KVRocksStandardHandler) expandPersistentVolumeClaim() error {
+	pvcList, err := h.k8s.ListStatefulSetPVC(h.key)
+	if err != nil {
+		return err
+	}
+	for _, pvc := range pvcList.Items {
+		pvcSize := pvc.Spec.Resources.Requests[v1.ResourceStorage]
+		if pvcSize.Cmp(h.instance.Spec.Storage.Size) < 0 {
+			pvc.Spec.Resources.Requests[v1.ResourceStorage] = h.instance.Spec.Storage.Size
+			if err := h.k8s.ExpandPVC(&pvc); err != nil {
+				return err
+			}
+		}
+	}
+	return nil
+}