Add support for pvcs to apiserver (#1118)

Signed-off-by: Paul S. Schweigert <[email protected]>

apiserver/pkg/util/cluster.go

@@ -494,7 +494,18 @@ func buildVols(apiVolumes []*api.Volume) []v1.Volume {
 			}
 			vols = append(vols, vol)
 		}
-		// TODO(Jeffwan@): handle PVC in the future
+		if rayVol.VolumeType == api.Volume_PERSISTENT_VOLUME_CLAIM {
+			vol := v1.Volume{
+				Name: rayVol.Name,
+				VolumeSource: v1.VolumeSource{
+					PersistentVolumeClaim: &v1.PersistentVolumeClaimVolumeSource{
+						ClaimName: rayVol.Name,
+						ReadOnly:  rayVol.ReadOnly,
+					},
+				},
+			}
+			vols = append(vols, vol)
+		}
 	}
 
 	return vols

apiserver/pkg/util/cluster_test.go

@@ -28,6 +28,13 @@ var testFileVolume = &api.Volume{
 	ReadOnly:             true,
 }
 
+var testPVCVolume = &api.Volume{
+	Name:       "test-pvc",
+	VolumeType: api.Volume_PERSISTENT_VOLUME_CLAIM,
+	MountPath:  "/pvc/dir",
+	ReadOnly:   true,
+}
+
 // Spec for testing
 var headGroup = api.HeadGroupSpec{
 	ComputeTemplate: "foo",
@@ -113,6 +120,16 @@ func TestBuildVolumes(t *testing.T) {
 			},
 		},
 	}
+
+	targetPVCVolume := v1.Volume{
+		Name: testPVCVolume.Name,
+		VolumeSource: v1.VolumeSource{
+			PersistentVolumeClaim: &v1.PersistentVolumeClaimVolumeSource{
+				ClaimName: testPVCVolume.Name,
+				ReadOnly:  testPVCVolume.ReadOnly,
+			},
+		},
+	}
 	tests := []struct {
 		name      string
 		apiVolume []*api.Volume
@@ -125,6 +142,11 @@ func TestBuildVolumes(t *testing.T) {
 			},
 			[]v1.Volume{targetVolume, targetFileVolume},
 		},
+		{
+			"pvc test",
+			[]*api.Volume{testPVCVolume},
+			[]v1.Volume{targetPVCVolume},
+		},
 	}
 	for _, tt := range tests {
 		t.Run(tt.name, func(t *testing.T) {
@@ -149,6 +171,11 @@ func TestBuildVolumeMounts(t *testing.T) {
 		MountPath:        testFileVolume.MountPath,
 		MountPropagation: &hostToContainer,
 	}
+	targetPVCVolumeMount := v1.VolumeMount{
+		Name:      testPVCVolume.Name,
+		ReadOnly:  testPVCVolume.ReadOnly,
+		MountPath: testPVCVolume.MountPath,
+	}
 	tests := []struct {
 		name      string
 		apiVolume []*api.Volume
@@ -165,6 +192,11 @@ func TestBuildVolumeMounts(t *testing.T) {
 				targetFileVolumeMount,
 			},
 		},
+		{
+			"pvc test",
+			[]*api.Volume{testPVCVolume},
+			[]v1.VolumeMount{targetPVCVolumeMount},
+		},
 	}
 	for _, tt := range tests {
 		t.Run(tt.name, func(t *testing.T) {