Ensure the use secondary vast cluster feature flag is used

Author: Pushkar Acharya

internal/crusoe/nfs_flag.go

@@ -14,6 +14,7 @@ import (
 const (
 	nfsFlagRouteTemplate                     = "%s/projects/%s/storage/nfs/is-using-nfs"
 	vastUseSecondaryClusterFlagRouteTemplate = "%s/projects/%s/storage/nfs/vast-use-secondary-cluster"
+	DEBUG                                    = 8
 )
 
 var (
@@ -51,7 +52,7 @@ func getFlag(ctx context.Context, crusoeHTTPClient *http.Client, flagRoute strin
 
 	klog.Infof("Flag API response - Status: %d, Content-Type: %s, Body length: %d bytes",
 		resp.StatusCode, resp.Header.Get("Content-Type"), len(bodyBytes))
-	klog.Infof("Flag API raw response body: %q", string(bodyBytes))
+	klog.V(DEBUG).Infof("Flag API raw response body: %q", string(bodyBytes))
 
 	// Check HTTP status code before unmarshaling
 	if resp.StatusCode != http.StatusOK {

internal/node/fs/fs_node.go

@@ -82,7 +82,7 @@ func (d *Node) NodePublishVolume(ctx context.Context, request *csi.NodePublishVo
 	nfsHost := d.NFSHost
 	nfsRemotePorts := d.NFSRemotePorts
 	klog.Infof("Host instance location: %q, checking against icatLocation: %q", d.HostInstance.Location, icatLocation)
-	if d.HostInstance.Location == icatLocation {
+	if d.useDNSForMount(ctx) {
 		klog.Infof("Using DNS-based NFS host for ICAT location: %s", crusoeCloudDNSNFSHost)
 		nfsHost = crusoeCloudDNSNFSHost
 		nfsRemotePorts = dnsRemotePorts
@@ -102,6 +102,18 @@ func (d *Node) NodePublishVolume(ctx context.Context, request *csi.NodePublishVo
 	return &csi.NodePublishVolumeResponse{}, nil
 }
 
+func (d *Node) useDNSForMount(ctx context.Context) bool {
+	useSecondaryVast, err := crusoe.GetVastUseSecondaryClusterFlag(
+		ctx, d.CrusoeHTTPClient, d.CrusoeAPIEndpoint, d.HostInstance.ProjectId)
+	if err != nil {
+		klog.Errorf("failed to fetch VastUseSecondaryCluster flag: %s", err.Error())
+
+		return false
+	}
+
+	return useSecondaryVast && d.HostInstance.Location == icatLocation
+}
+
 func (d *Node) NodeUnpublishVolume(_ context.Context, request *csi.NodeUnpublishVolumeRequest) (
 	*csi.NodeUnpublishVolumeResponse,
 	error,