Support partitions in VolumeAttributes

Author: davidz627

pkg/common/constants.go

@@ -23,4 +23,7 @@ const (
 
 	// Keys for Topology. This key will be shared amonst drivers from GCP
 	TopologyKeyZone = "com.google.topology/zone"
+
+	// VolumeAttributes for Partition
+	VolumeAttributePartition = "partition"
 )

pkg/gce-pd-csi-driver/node.go

@@ -174,9 +174,12 @@ func (ns *GCENodeServer) NodeStageVolume(ctx context.Context, req *csi.NodeStage
 	}
 
 	// Part 1: Get device path of attached device
-	// TODO(#83): Get real partitions
 	partition := ""
 
+	if part, ok := req.GetVolumeAttributes()[common.VolumeAttributePartition]; ok {
+		partition = part
+	}
+
 	deviceName, err := common.GetDeviceName(volumeKey)
 	if err != nil {
 		status.Error(codes.Internal, fmt.Sprintf("error getting device name: %v", err))