Update code with new type logic

RaunakShah · RaunakShah · commit da516a53368e · 2022-04-14T00:12:11.000+05:30
diff --git a/pkg/common-controller/snapshot_controller.go b/pkg/common-controller/snapshot_controller.go
@@ -685,17 +685,10 @@ func (ctrl *csiSnapshotCommonController) createSnapshotContent(snapshot *crdv1.V
 	}
 
 	if ctrl.preventVolumeModeConversion {
-		volumeMode := volume.Spec.VolumeMode
-		if volumeMode != nil {
-			snapshotContent.Spec.SourceVolumeMode = new(crdv1.SourceVolumeMode)
-			switch *volumeMode {
-			case v1.PersistentVolumeBlock:
-				*snapshotContent.Spec.SourceVolumeMode = crdv1.SourceVolumeModeBlock
-			case v1.PersistentVolumeFilesystem:
-				*snapshotContent.Spec.SourceVolumeMode = crdv1.SourceVolumeModeFilesystem
-			}
+		if volume.Spec.VolumeMode != nil {
+			snapshotContent.Spec.SourceVolumeMode = volume.Spec.VolumeMode
+			klog.V(5).Infof("snapcontent %s has volume mode %s", snapshotContent.Name, *snapshotContent.Spec.SourceVolumeMode)
 		}
-		klog.V(5).Infof("snapcontent %s has volume mode %s", snapshotContent.Name, *snapshotContent.Spec.SourceVolumeMode)
 	}
 
 	// Set AnnDeletionSecretRefName and AnnDeletionSecretRefNamespace

Original file line number	Diff line number	Diff line change
`@@ -685,17 +685,10 @@ func (ctrl csiSnapshotCommonController) createSnapshotContent(snapshot crdv1.V`
`685`	`685`	`}`
`686`	`686`
`687`	`687`	`if ctrl.preventVolumeModeConversion {`
`688`		`- volumeMode := volume.Spec.VolumeMode`
`689`		`- if volumeMode != nil {`
`690`		`- snapshotContent.Spec.SourceVolumeMode = new(crdv1.SourceVolumeMode)`
`691`		`- switch *volumeMode {`
`692`		`- case v1.PersistentVolumeBlock:`
`693`		`- *snapshotContent.Spec.SourceVolumeMode = crdv1.SourceVolumeModeBlock`
`694`		`- case v1.PersistentVolumeFilesystem:`
`695`		`- *snapshotContent.Spec.SourceVolumeMode = crdv1.SourceVolumeModeFilesystem`
`696`		`- }`
	`688`	`+ if volume.Spec.VolumeMode != nil {`
	`689`	`+ snapshotContent.Spec.SourceVolumeMode = volume.Spec.VolumeMode`
	`690`	`+ klog.V(5).Infof("snapcontent %s has volume mode %s", snapshotContent.Name, *snapshotContent.Spec.SourceVolumeMode)`
`697`	`691`	`}`
`698`		`- klog.V(5).Infof("snapcontent %s has volume mode %s", snapshotContent.Name, *snapshotContent.Spec.SourceVolumeMode)`
`699`	`692`	`}`
`700`	`693`
`701`	`694`	`// Set AnnDeletionSecretRefName and AnnDeletionSecretRefNamespace`