From 51fe550a3e22154ee7bbec90f65bdbebe039be38 Mon Sep 17 00:00:00 2001
From: wangxiaoyou1993 <xiaoyou@mage.ai>
Date: Thu, 11 Jul 2024 14:19:10 -0700
Subject: [PATCH 1/2] [xy] Support provisioning PVC in helm chart.

---
 charts/mageai/templates/pv.yaml | 41 +++++++++++++++++++++++++++++++++
 charts/mageai/values.yaml       |  8 +++++++
 2 files changed, 49 insertions(+)
 create mode 100644 charts/mageai/templates/pv.yaml

diff --git a/charts/mageai/templates/pv.yaml b/charts/mageai/templates/pv.yaml
new file mode 100644
index 0000000..9d9581f
--- /dev/null
+++ b/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 }}
diff --git a/charts/mageai/values.yaml b/charts/mageai/values.yaml
index 73826cc..9b460aa 100644
--- a/charts/mageai/values.yaml
+++ b/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: {}
 

From 78a710e9eb63a690288e68d4f2ba0aac8acb7616 Mon Sep 17 00:00:00 2001
From: wangxiaoyou1993 <xiaoyou@mage.ai>
Date: Thu, 11 Jul 2024 14:19:46 -0700
Subject: [PATCH 2/2] [xy] Bump chart version to 0.2.6

---
 charts/mageai/Chart.yaml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/charts/mageai/Chart.yaml b/charts/mageai/Chart.yaml
index 4d1bb4b..33ce8b9 100644
--- a/charts/mageai/Chart.yaml
+++ b/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