Implement support for storage class parameter - fsType

Fixes: #62
kubernetes-sigs · Oct 16, 2018 · a0cdd1b · a0cdd1b
1 parent 290b6d0
commit a0cdd1b
Show file tree

Hide file tree

Showing 4 changed files with 35 additions and 5 deletions.
diff --git a/pkg/cloud/cloud.go b/pkg/cloud/cloud.go
@@ -87,6 +87,7 @@ type Disk struct {
 	VolumeID         string
 	CapacityGiB      int64
 	AvailabilityZone string
+	FsType           string
 }
 
 // DiskOptions represents parameters to create an EBS volume

diff --git a/pkg/driver/controller.go b/pkg/driver/controller.go
@@ -84,6 +84,8 @@ func (d *Driver) CreateVolume(ctx context.Context, req *csi.CreateVolumeRequest)
 	if err != nil {
 		return nil, status.Errorf(codes.Internal, "Could not create volume %q: %v", volName, err)
 	}
+	fsType := req.GetParameters()["fsType"]
+	disk.FsType = fsType
 	return newCreateVolumeResponse(disk), nil
 }
 
@@ -279,6 +281,9 @@ func newCreateVolumeResponse(disk *cloud.Disk) *csi.CreateVolumeResponse {
 		Volume: &csi.Volume{
 			Id:            disk.VolumeID,
 			CapacityBytes: util.GiBToBytes(disk.CapacityGiB),
+			Attributes: map[string]string{
+				"fsType": disk.FsType,
+			},
 			AccessibleTopology: []*csi.Topology{
 				&csi.Topology{
 					Segments: map[string]string{topologyKey: disk.AvailabilityZone},

diff --git a/pkg/driver/controller_test.go b/pkg/driver/controller_test.go
@@ -59,7 +59,7 @@ func TestCreateVolume(t *testing.T) {
 			expVol: &csi.Volume{
 				CapacityBytes: stdVolSize,
 				Id:            "vol-test",
-				Attributes:    nil,
+				Attributes:    map[string]string{"fsType": ""},
 			},
 		},
 		{
@@ -89,7 +89,7 @@ func TestCreateVolume(t *testing.T) {
 			expVol: &csi.Volume{
 				CapacityBytes: stdVolSize,
 				Id:            "vol-test",
-				Attributes:    nil,
+				Attributes:    map[string]string{"fsType": ""},
 			},
 		},
 		{
@@ -118,7 +118,7 @@ func TestCreateVolume(t *testing.T) {
 			expVol: &csi.Volume{
 				CapacityBytes: cloud.DefaultVolumeSize,
 				Id:            "vol-test",
-				Attributes:    nil,
+				Attributes:    map[string]string{"fsType": ""},
 			},
 		},
 		{
@@ -132,7 +132,21 @@ func TestCreateVolume(t *testing.T) {
 			expVol: &csi.Volume{
 				CapacityBytes: 2147483648, // 1 GiB + 1 byte = 2 GiB
 				Id:            "vol-test",
-				Attributes:    nil,
+				Attributes:    map[string]string{"fsType": ""},
+			},
+		},
+		{
+			name: "success with fstype parameter",
+			req: &csi.CreateVolumeRequest{
+				Name:               "vol-test",
+				CapacityRange:      stdCapRange,
+				VolumeCapabilities: stdVolCap,
+				Parameters:         map[string]string{"fsType": "ext4"},
+			},
+			expVol: &csi.Volume{
+				CapacityBytes: stdVolSize,
+				Id:            "vol-test",
+				Attributes:    map[string]string{"fsType": "ext4"},
 			},
 		},
 	}

diff --git a/pkg/driver/node.go b/pkg/driver/node.go
@@ -72,10 +72,20 @@ func (d *Driver) NodeStageVolume(ctx context.Context, req *csi.NodeStageVolumeRe
 		msg := fmt.Sprintf("target %q is not a valid mount point", target)
 		return nil, status.Error(codes.InvalidArgument, msg)
 	}
+	// Get fs type that the volume will be formatted with
+	var (
+		fsType string
+		exists bool
+	)
+	attributes := req.GetVolumeAttributes()
+	fsType, exists = attributes["fsType"]
+	if !exists || fsType == "" {
+		fsType = "ext4"
+	}
 
 	// FormatAndMount will format only if needed
 	glog.V(5).Infof("NodeStageVolume: formatting %s and mounting at %s", source, target)
-	err = d.mounter.FormatAndMount(source, target, "ext4", nil)
+	err = d.mounter.FormatAndMount(source, target, fsType, nil)
 	if err != nil {
 		msg := fmt.Sprintf("could not format %q and mount it at %q", source, target)
 		return nil, status.Error(codes.Internal, msg)