✨ Make CoW Support dynamic

Signed-off-by: Muhammed Hussain Karimi <[email protected]>

Author: mhkarimi1383

rawfile/bd2fs.py

@@ -300,7 +300,7 @@ def CreateSnapshot(self, request: csi_pb2.CreateSnapshotRequest, context):
             copy_on_write = (
                 copy_on_write_param
                 if copy_on_write_param is not None
-                else consts.COW_SUPPORTED
+                else consts.COW_SUPPORT_MAP.get(volume_meta["storage_pool"], False)
             )
 
         def _get_current_snapshot():

rawfile/consts.py

@@ -14,3 +14,4 @@
 CSI_K8S_PVC_NAME_KEY = "csi.storage.k8s.io/pvc/name"
 GA_ID = "Ry1UWkdQNDY2MThX"
 GA_KEY = "OTFKR2RUZzlRd0duN1ktdnZ1TTR6QQ=="
+COW_SUPPORT_MAP = {}

rawfile/rawfile.py

@@ -17,6 +17,7 @@
 import consts
 from analytics.ga4 import run_ping, shutdown_event_worker, run_event_worker
 from orchestrator.k8s import node_ip_mapping
+from utils.rawfile import is_cow_supported
 from utils.volume_manager import manager as volume_manager
 from setproctitle import setproctitle
 import os
@@ -47,7 +48,10 @@ def node_driver_preflight_checks(task_manager: task_manager.TaskManager):
     volume_manager.migrate_metadata_dir()
     volume_manager.migrate_all_volume_schemas()
     task_manager.migrate_tasks_file_path()
-    # consts.COW_SUPPORTED = is_cow_supported(data_dir)
+    consts.COW_SUPPORT_MAP = {
+        name: is_cow_supported(pool.path)
+        for name, pool in config.csi_driver.storage_pools.items()
+    }
 
 
 def csi_driver(driver_config: CSIDriverCmd):

rawfile/utils/volume_manager.py

@@ -92,7 +92,11 @@ def create_volume(
                         temporary=True,
                         copy_on_write=source_meta.get("copy_on_write", None)
                         if source_meta.get("copy_on_write", None) is not None
-                        else (consts.COW_SUPPORTED or False),
+                        else (
+                            consts.COW_SUPPORT_MAP.get(
+                                source_meta["storage_pool"], False
+                            )
+                        ),
                         freeze_fs=source_meta.get("freezefs", False),
                     )
             with VolLock(volume_id):