Move volume capability

Signed-off-by: Pierre-Emmanuel Jacquier <[email protected]>
exoscale · Jan 23, 2024 · 5040fd5 · 5040fd5
1 parent afb159c
commit 5040fd5
Showing 1 changed file with 13 additions and 13 deletions.
diff --git a/driver/controller.go b/driver/controller.go
@@ -622,6 +622,19 @@ func (d *controllerService) ControllerExpandVolume(ctx context.Context, req *csi
 		return nil, err
 	}
 
+	nodeExpansionRequired := true
+	volumeCapability := req.GetVolumeCapability()
+	if volumeCapability != nil {
+		err := validateVolumeCapability(volumeCapability)
+		if err != nil {
+			return nil, status.Errorf(codes.InvalidArgument, "volumeCapabilities not supported: %s", err)
+		}
+
+		if _, ok := volumeCapability.GetAccessType().(*csi.VolumeCapability_Block); ok {
+			nodeExpansionRequired = false
+		}
+	}
+
 	newSize, err := getNewVolumeSize(req.GetCapacityRange())
 	if err != nil {
 		return nil, status.Errorf(codes.OutOfRange, "invalid capacity range: %v", err)
@@ -638,19 +651,6 @@ func (d *controllerService) ControllerExpandVolume(ctx context.Context, req *csi
 		return nil, err
 	}
 
-	nodeExpansionRequired := true
-	volumeCapability := req.GetVolumeCapability()
-	if volumeCapability != nil {
-		err := validateVolumeCapability(volumeCapability)
-		if err != nil {
-			return nil, status.Errorf(codes.InvalidArgument, "volumeCapabilities not supported: %s", err)
-		}
-
-		if _, ok := volumeCapability.GetAccessType().(*csi.VolumeCapability_Block); ok {
-			nodeExpansionRequired = false
-		}
-	}
-
 	return &csi.ControllerExpandVolumeResponse{
 		CapacityBytes:         newSize,
 		NodeExpansionRequired: nodeExpansionRequired,