Add support for mounting PVCs

windycom · Aug 13, 2021 · 120ce73 · 120ce73
1 parent a2a1c46
commit 120ce73
Show file tree

Hide file tree

Showing 3 changed files with 22 additions and 1 deletion.
diff --git a/deploy/charts/ray/Chart.yaml b/deploy/charts/ray/Chart.yaml
@@ -4,7 +4,7 @@ description: A Helm chart for deployments of Ray on Kubernetes.
 type: application
 
 # Chart version.
-version: 0.1.0
+version: 0.1.0-1
 
 # Ray version.
 appVersion: "latest"
diff --git a/deploy/charts/ray/templates/raycluster.yaml b/deploy/charts/ray/templates/raycluster.yaml
@@ -39,6 +39,11 @@ spec:
           - name: dshm
             emptyDir:
               medium: Memory
+          {{- range $volume := .persistentVolumes }}
+          - name: {{ $volume.claimName }}
+            persistentVolumeClaim:
+              claimName: {{ $volume.claimName }}
+          {{- end }}
           containers:
           - name: ray-node
             imagePullPolicy: Always
@@ -59,6 +64,18 @@ spec:
             volumeMounts:
             - mountPath: /dev/shm
               name: dshm
+            {{- range $volume := .persistentVolumes }}
+            {{- range $mount := $volume.mounts }}
+            - name: {{ $volume.claimName }}
+              mountPath: {{ $mount.mountPath }}
+              {{- if $mount.subPath }}
+              subPath: {{ $mount.subPath }}
+              {{- end }}
+              {{- if $mount.readOnly }}
+              readOnly: {{ $mount.readOnly }}
+              {{- end }}
+            {{- end }}
+            {{- end }}
             resources:
               requests:
                 cpu: {{ .CPU }}

diff --git a/deploy/charts/ray/values.yaml b/deploy/charts/ray/values.yaml
@@ -31,6 +31,8 @@ podTypes:
         rayResources: {}
         # Optionally, set a node selector for this podType: https://kubernetes.io/docs/concepts/scheduling-eviction/assign-pod-node/#nodeselector
         nodeSelector: {}
+        # PersistentVolumeClaim mounts
+        persistentVolumes: []
     # The key for each podType is a user-defined string.
     rayWorkerType:
         # minWorkers is the minimum number of Ray workers of this pod type to keep running.
@@ -53,6 +55,8 @@ podTypes:
         rayResources: {}
         # Optionally, set a node selector for this Pod type. See https://kubernetes.io/docs/concepts/scheduling-eviction/assign-pod-node/#nodeselector
         nodeSelector: {}
+        # PersistentVolumeClaim mounts
+        persistentVolumes: []
 
 
 # Operator settings: