Storage plugin support to check if volume on datastore requires access for migration

engine/api/src/main/java/org/apache/cloudstack/engine/subsystem/api/storage/PrimaryDataStoreDriver.java

@@ -42,6 +42,8 @@ enum QualityOfServiceState { MIGRATION, NO_MIGRATION }
 
     void revokeAccess(DataObject dataObject, Host host, DataStore dataStore);
 
+    boolean requiresAccessForMigration(DataObject dataObject);
+
     /**
      * intended for managed storage (cloud.storage_pool.managed = true)
      * if not managed, return volume.getSize()

engine/api/src/main/java/org/apache/cloudstack/engine/subsystem/api/storage/VolumeService.java

@@ -54,6 +54,8 @@ public VolumeInfo getVolume() {
 
     void revokeAccess(DataObject dataObject, Host host, DataStore dataStore);
 
+    boolean requiresAccessForMigration(DataObject dataObject, DataStore dataStore);
+
     /**
      * Creates the volume based on the given criteria
      *

engine/orchestration/src/main/java/org/apache/cloudstack/engine/orchestration/VolumeOrchestrator.java

@@ -1228,8 +1228,8 @@ public void release(long vmId, long hostId) {
                 DataStore dataStore = dataStoreMgr.getDataStore(volumeForVm.getPoolId(), DataStoreRole.Primary);
                 PrimaryDataStore primaryDataStore = (PrimaryDataStore)dataStore;
 
-                // This might impact other managed storages, grant access for PowerFlex storage pool only
-                if (primaryDataStore.isManaged() && primaryDataStore.getPoolType() == Storage.StoragePoolType.PowerFlex) {
+                // This might impact other managed storages, enable requires access for migration in relevant datastore driver (currently enabled for PowerFlex storage pool only)
+                if (primaryDataStore.isManaged() && volService.requiresAccessForMigration(volumeInfo, dataStore)) {
                     volService.revokeAccess(volumeInfo, host, dataStore);
                 }
             }
@@ -1507,8 +1507,8 @@ public void prepareForMigration(VirtualMachineProfile vm, DeployDestination dest
             disk.setDetails(getDetails(volumeInfo, dataStore));
 
             PrimaryDataStore primaryDataStore = (PrimaryDataStore)dataStore;
-            // This might impact other managed storages, grant access for PowerFlex storage pool only
-            if (primaryDataStore.isManaged() && primaryDataStore.getPoolType() == Storage.StoragePoolType.PowerFlex) {
+            // This might impact other managed storages, enable requires access for migration in relevant datastore driver (currently enabled for PowerFlex storage pool only)
+            if (primaryDataStore.isManaged() && volService.requiresAccessForMigration(volumeInfo, dataStore)) {
                 volService.grantAccess(volFactory.getVolume(vol.getId()), dest.getHost(), dataStore);
             }
 

engine/storage/volume/src/main/java/org/apache/cloudstack/storage/volume/VolumeServiceImpl.java

@@ -266,6 +266,19 @@ public void revokeAccess(DataObject dataObject, Host host, DataStore dataStore)
         }
     }
 
+    @Override
+    public boolean requiresAccessForMigration(DataObject dataObject, DataStore dataStore) {
+        DataStoreDriver dataStoreDriver = dataStore != null ? dataStore.getDriver() : null;
+        if (dataStoreDriver == null) {
+            return false;
+        }
+
+        if (dataStoreDriver instanceof PrimaryDataStoreDriver) {
+            return ((PrimaryDataStoreDriver)dataStoreDriver).requiresAccessForMigration(dataObject);
+        }
+        return false;
+    }
+
     @Override
     public AsyncCallFuture<VolumeApiResult> createVolumeAsync(VolumeInfo volume, DataStore dataStore) {
         AsyncCallFuture<VolumeApiResult> future = new AsyncCallFuture<VolumeApiResult>();

plugins/storage/volume/datera/src/main/java/org/apache/cloudstack/storage/datastore/driver/DateraPrimaryDataStoreDriver.java

@@ -394,6 +394,11 @@ public void revokeAccess(DataObject dataObject, Host host, DataStore dataStore)
         }
     }
 
+    @Override
+    public boolean requiresAccessForMigration(DataObject dataObject) {
+        return false;
+    }
+
     /**
      * Returns the size of template on this primary storage. If we already have a
      * template on this storage, we return 0

plugins/storage/volume/default/src/main/java/org/apache/cloudstack/storage/datastore/driver/CloudStackPrimaryDataStoreDriverImpl.java

@@ -170,6 +170,11 @@ public boolean grantAccess(DataObject dataObject, Host host, DataStore dataStore
     public void revokeAccess(DataObject dataObject, Host host, DataStore dataStore) {
     }
 
+    @Override
+    public boolean requiresAccessForMigration(DataObject dataObject) {
+        return false;
+    }
+
     @Override
     public long getUsedBytes(StoragePool storagePool) {
         return 0;

plugins/storage/volume/linstor/src/main/java/org/apache/cloudstack/storage/datastore/driver/LinstorPrimaryDataStoreDriverImpl.java

@@ -148,6 +148,11 @@ public void revokeAccess(DataObject dataObject, Host host, DataStore dataStore)
     {
     }
 
+    @Override
+    public boolean requiresAccessForMigration(DataObject dataObject) {
+        return false;
+    }
+
     @Override
     public long getUsedBytes(StoragePool storagePool)
     {

plugins/storage/volume/nexenta/src/main/java/org/apache/cloudstack/storage/datastore/driver/NexentaPrimaryDataStoreDriver.java

@@ -68,6 +68,11 @@ public void revokeAccess(DataObject dataObject, Host host, DataStore dataStore)
         //To change body of implemented methods use File | Settings | File Templates.
     }
 
+    @Override
+    public boolean requiresAccessForMigration(DataObject dataObject) {
+        return false;
+    }
+
     @Override
     public long getUsedBytes(StoragePool storagePool) {
         return 0;

plugins/storage/volume/sample/src/main/java/org/apache/cloudstack/storage/datastore/driver/SamplePrimaryDataStoreDriverImpl.java

@@ -88,6 +88,11 @@ public ChapInfo getChapInfo(DataObject dataObject) {
     @Override
     public void revokeAccess(DataObject dataObject, Host host, DataStore dataStore) {}
 
+    @Override
+    public boolean requiresAccessForMigration(DataObject dataObject) {
+        return false;
+    }
+
     @Override
     public long getUsedBytes(StoragePool storagePool) {
         return 0;

plugins/storage/volume/scaleio/src/main/java/org/apache/cloudstack/storage/datastore/driver/ScaleIOPrimaryDataStoreDriver.java

@@ -302,6 +302,11 @@ public String getConnectedSdc(long poolId, long hostId) {
         return null;
     }
 
+    @Override
+    public boolean requiresAccessForMigration(DataObject dataObject) {
+        return true;
+    }
+
     @Override
     public long getUsedBytes(StoragePool storagePool) {
         long usedSpaceBytes = 0;

plugins/storage/volume/solidfire/src/main/java/org/apache/cloudstack/storage/datastore/driver/SolidFirePrimaryDataStoreDriver.java

@@ -384,6 +384,11 @@ private Iops getIops(Long minIops, Long maxIops, long storagePoolId) {
         return new Iops(minIops, maxIops, getDefaultBurstIops(storagePoolId, maxIops));
     }
 
+    @Override
+    public boolean requiresAccessForMigration(DataObject dataObject) {
+        return false;
+    }
+
     @Override
     public long getUsedBytes(StoragePool storagePool) {
         return getUsedBytes(storagePool, Long.MIN_VALUE);

plugins/storage/volume/storpool/src/main/java/org/apache/cloudstack/storage/datastore/driver/StorPoolPrimaryDataStoreDriver.java

@@ -177,6 +177,11 @@ public boolean grantAccess(DataObject data, Host host, DataStore dataStore) {
     public void revokeAccess(DataObject data, Host host, DataStore dataStore) {
     }
 
+    @Override
+    public boolean requiresAccessForMigration(DataObject dataObject) {
+        return false;
+    }
+
     private void updateStoragePool(final long poolId, final long deltaUsedBytes) {
         StoragePoolVO storagePool = primaryStoreDao.findById(poolId);
         final long capacity = storagePool.getCapacityBytes();