Merge pull request #59 from mage-ai/xiaoyou/provision-volume

[xy] Provision PVC in helm chart

charts/mageai/Chart.yaml

@@ -76,4 +76,4 @@ type: application
 # This is the chart version. This version number should be incremented each time you make changes
 # to the chart and its templates, including the app version.
 # Versions are expected to follow Semantic Versioning (https://semver.org/)
-version: 0.2.5
+version: 0.2.6

charts/mageai/templates/pv.yaml

@@ -0,0 +1,41 @@
+{{- if and .Values.persistence .Values.persistence.enabled -}}
+
+{{- if not .Values.persistence.dynamicProvisioning }}
+apiVersion: v1
+kind: PersistentVolume
+metadata:
+  name: pv-{{ include "mageai.fullname" . }}
+  labels:
+    deployment: {{ include "mageai.fullname" . }}
+spec:
+  capacity:
+    storage: {{ .Values.persistence.size}}
+  volumeMode: Filesystem
+  accessModes:
+    - ReadWriteMany
+  persistentVolumeReclaimPolicy: Retain
+  storageClassName: {{ .Values.persistence.storageClassName}}
+  {{- with .Values.persistence.csi }}
+  csi:
+    {{- toYaml . | nindent 4 }}
+  {{- end }}
+{{- end }}
+
+---
+apiVersion: v1
+kind: PersistentVolumeClaim
+metadata:
+  name: pvc-{{ include "mageai.fullname" . }}
+spec:
+  accessModes:
+    - ReadWriteMany
+  storageClassName: {{ .Values.persistence.storageClassName}}
+  resources:
+    requests:
+      storage: {{ .Values.persistence.size}}
+  {{- if not .Values.persistence.dynamicProvisioning }}
+  selector:
+    matchLabels:
+      deployment: {{ include "mageai.fullname" . }}
+  {{- end}}
+{{- end }}

charts/mageai/values.yaml

@@ -185,6 +185,14 @@ extraVolumes:
     hostPath:
       path: /path/to/mage_project
 
+persistence:
+  enabled: false
+  storageClassName: storage-class-name
+  size: 5Gi
+  # csi:
+  #   driver: efs.csi.aws.com
+  #   volumeHandle: fs-0123456789
+
 # config: Default configuration for mageai as environment variables. These get injected directly in the container.
 config: {}