use node UID for VG creation instead of name

Author: Sneha-at

cmd/gce-pd-csi-driver/main.go

@@ -27,6 +27,7 @@ import (
 	"strings"
 	"time"
 
+	v1 "k8s.io/api/core/v1"
 	"k8s.io/klog/v2"
 	"k8s.io/utils/strings/slices"
 	"sigs.k8s.io/gcp-compute-persistent-disk-csi-driver/pkg/common"
@@ -245,7 +246,11 @@ func handle() {
 		if err != nil {
 			klog.Fatalf("Failed to set up metadata service: %v", err.Error())
 		}
-		isDataCacheEnabledNodePool, err := isDataCacheEnabledNodePool(ctx, *nodeName)
+		node, err := driver.FetchNodeWithRetry(ctx, *nodeName)
+		if err != nil {
+			klog.Fatalf("Failed to get node info from API server: %v", err.Error())
+		}
+		isDataCacheEnabledNodePool, err := isDataCacheEnabledNodePool(ctx, node)
 		if err != nil {
 			klog.Fatalf("Failed to get node info from API server: %v", err.Error())
 		}
@@ -264,7 +269,7 @@ func handle() {
 				klog.Errorf("Data Cache enabled, but --node-name not passed")
 			}
 			if nsArgs.DataCacheEnabledNodePool {
-				if err := setupDataCache(ctx, *nodeName, nodeServer.MetadataService.GetName()); err != nil {
+				if err := setupDataCache(ctx, node); err != nil {
 					klog.Errorf("Data Cache setup failed: %v", err)
 				}
 				go driver.StartWatcher(*nodeName)
@@ -351,12 +356,13 @@ func urlFlag(target **url.URL, name string, usage string) {
 	})
 }
 
-func isDataCacheEnabledNodePool(ctx context.Context, nodeName string) (bool, error) {
+func isDataCacheEnabledNodePool(ctx context.Context, node *v1.Node) (bool, error) {
 	if !*enableDataCacheFlag {
 		return false, nil
 	}
+	nodeName := node.Name
 	if len(nodeName) > 0 && nodeName != common.TestNode { // disregard logic below when E2E testing.
-		dataCacheLSSDCount, err := driver.GetDataCacheCountFromNodeLabel(ctx, nodeName)
+		dataCacheLSSDCount, err := driver.GetDataCacheCountFromNodeLabel(ctx, node)
 		return dataCacheLSSDCount != 0, err
 	}
 	return true, nil
@@ -394,7 +400,8 @@ func fetchLssdsForRaiding(lssdCount int) ([]string, error) {
 	return availableLssds[:lssdCount], nil
 }
 
-func setupDataCache(ctx context.Context, nodeName string, nodeId string) error {
+func setupDataCache(ctx context.Context, node *v1.Node) error {
+	nodeName := node.Name
 	isAlreadyRaided, err := driver.IsRaided()
 	if err != nil {
 		klog.V(4).Infof("Errored while scanning for available LocalSSDs err:%v; continuing Raiding", err)
@@ -404,9 +411,10 @@ func setupDataCache(ctx context.Context, nodeName string, nodeId string) error {
 	}
 
 	lssdCount := common.LocalSSDCountForDataCache
+	nodeUid := nodeName
 	if nodeName != common.TestNode {
-		var err error
-		lssdCount, err = driver.GetDataCacheCountFromNodeLabel(ctx, nodeName)
+		nodeUid = string(node.ObjectMeta.UID)
+		lssdCount, err = driver.GetDataCacheCountFromNodeLabel(ctx, node)
 		if err != nil {
 			return err
 		}
@@ -425,7 +433,7 @@ func setupDataCache(ctx context.Context, nodeName string, nodeId string) error {
 	}
 
 	// Initializing data cache node (VG checks w/ raided lssd)
-	if err := driver.InitializeDataCacheNode(nodeId); err != nil {
+	if err := driver.InitializeDataCacheNode(nodeUid); err != nil {
 		return err
 	}
 

pkg/gce-pd-csi-driver/cache.go

@@ -248,19 +248,7 @@ func ValidateDataCacheConfig(dataCacheMode string, dataCacheSize string, ctx con
 	return fmt.Errorf("Data Cache is not enabled for PVC (data-cache-size: %v, data-cache-mode: %v). Please set both parameters in StorageClass to enable caching", dataCacheSize, dataCacheMode)
 }
 
-func GetDataCacheCountFromNodeLabel(ctx context.Context, nodeName string) (int, error) {
-	cfg, err := rest.InClusterConfig()
-	if err != nil {
-		return 0, err
-	}
-	kubeClient, err := kubernetes.NewForConfig(cfg)
-	if err != nil {
-		return 0, err
-	}
-	node, err := getNodeWithRetry(ctx, kubeClient, nodeName)
-	if err != nil {
-		return 0, err
-	}
+func GetDataCacheCountFromNodeLabel(ctx context.Context, node *v1.Node) (int, error) {
 	if val, found := node.GetLabels()[fmt.Sprintf(common.NodeLabelPrefix, common.DataCacheLssdCountLabel)]; found {
 		dataCacheCount, err := strconv.Atoi(val)
 		if err != nil {
@@ -272,14 +260,22 @@ func GetDataCacheCountFromNodeLabel(ctx context.Context, nodeName string) (int,
 	return 0, nil
 }
 
-func getNodeWithRetry(ctx context.Context, kubeClient *kubernetes.Clientset, nodeName string) (*v1.Node, error) {
+func FetchNodeWithRetry(ctx context.Context, nodeName string) (*v1.Node, error) {
 	var nodeObj *v1.Node
+	cfg, err := rest.InClusterConfig()
+	if err != nil {
+		return nil, err
+	}
+	kubeClient, err := kubernetes.NewForConfig(cfg)
+	if err != nil {
+		return nil, err
+	}
 	backoff := wait.Backoff{
 		Duration: 1 * time.Second,
 		Factor:   2.0,
 		Steps:    5,
 	}
-	err := wait.ExponentialBackoffWithContext(ctx, backoff, func(_ context.Context) (bool, error) {
+	err = wait.ExponentialBackoffWithContext(ctx, backoff, func(_ context.Context) (bool, error) {
 		node, err := kubeClient.CoreV1().Nodes().Get(ctx, nodeName, metav1.GetOptions{})
 		if err != nil {
 			klog.Warningf("Error getting node %s: %v, retrying...\n", nodeName, err)