feat(controller): snapshot restoration

enix · Apr 1, 2021 · 5a62d21 · 5a62d21
1 parent be99b28
commit 5a62d21
Showing 1 changed file with 17 additions and 3 deletions.
diff --git a/pkg/controller/provisioner.go b/pkg/controller/provisioner.go
@@ -60,9 +60,23 @@ func (controller *Controller) CreateVolume(ctx context.Context, req *csi.CreateV
 	}
 
 	if !volumeExists {
-		_, _, err = controller.dothillClient.CreateVolume(volumeID, sizeStr, parameters[common.PoolConfigKey])
-		if err != nil {
-			return nil, err
+		var sourceID string
+
+		if volume := req.VolumeContentSource.GetVolume(); volume != nil {
+			sourceID = volume.VolumeId
+		}
+
+		if snapshot := req.VolumeContentSource.GetSnapshot(); sourceID == "" && snapshot != nil {
+			sourceID = snapshot.SnapshotId
+		}
+
+		if sourceID != "" {
+			controller.dothillClient.VolumeCopy(sourceID, volumeID, parameters[common.PoolConfigKey])
+		} else {
+			_, _, err = controller.dothillClient.CreateVolume(volumeID, sizeStr, parameters[common.PoolConfigKey])
+			if err != nil {
+				return nil, err
+			}
 		}
 	}