Add pvc to weave

wandb · Aug 27, 2023 · 2613abd · 2613abd
1 parent 7b8aaaa
commit 2613abd
Show file tree

Hide file tree

Showing 11 changed files with 43 additions and 14 deletions.
diff --git a/charts/operator-wandb/Chart.yaml b/charts/operator-wandb/Chart.yaml
@@ -2,7 +2,7 @@ apiVersion: v2
 name: operator-wandb
 description: A Helm chart for deploying W&B to Kubernetes
 type: application
-version: 0.2.0
+version: 0.3.0
 appVersion: 1.0.0
 icon: https://wandb.ai/logo.svg
 maintainers:

diff --git a/charts/operator-wandb/charts/mysql/templates/NOTES.txt b/charts/operator-wandb/charts/mysql/templates/NOTES.txt
diff --git a/charts/operator-wandb/charts/mysql/templates/pvc.yaml b/charts/operator-wandb/charts/mysql/templates/pvc.yaml
@@ -10,7 +10,7 @@ metadata:
 spec:
   accessModes:
     - ReadWriteOnce
-  {{ include "wandb.storageClass" . | nindent 4 }}
+  {{ include "wandb.storageClass" .Values.persistence | nindent 4 }}
   resources:
     requests:
       storage: {{ .Values.persistence.size | quote }}
diff --git a/charts/operator-wandb/charts/mysql/templates/statefulset.yaml b/charts/operator-wandb/charts/mysql/templates/statefulset.yaml
@@ -80,6 +80,7 @@ spec:
               subPath: my.cnf
           resources:
             {{- toYaml .Values.resources | nindent 12 }}
+
       volumes:
         - name: initdb
           configMap:

diff --git a/charts/operator-wandb/charts/weave/templates/NOTES.txt b/charts/operator-wandb/charts/weave/templates/NOTES.txt
diff --git a/charts/operator-wandb/charts/weave/templates/deployment.yaml b/charts/operator-wandb/charts/weave/templates/deployment.yaml
@@ -79,4 +79,12 @@ spec:
 
           resources:
             {{- toYaml .Values.resources | nindent 12 }}
+
+          volumeMounts:
+            - name: cache
+              mountPath: /vol/weave/cache
+      volumes:
+        - name: cache
+          persistentVolumeClaim:
+            claimName: {{ template "weave.fullname" . }}
 {{- end }}
diff --git a/charts/operator-wandb/charts/weave/templates/pvc.yaml b/charts/operator-wandb/charts/weave/templates/pvc.yaml
@@ -0,0 +1,15 @@
+kind: PersistentVolumeClaim
+apiVersion: v1
+metadata:
+  name: {{ template "weave.fullname" . }}
+  namespace: {{ $.Release.Namespace }}
+  labels:
+    {{- include "wandb.commonLabels" . | nindent 4 }}
+    {{- include "weave.commonLabels" . | nindent 4 }}
+    {{- include "weave.labels" . | nindent 4 }}
+spec:
+  accessModes: [ReadWriteOnce]
+  {{ include "wandb.storageClass" . | nindent 4 }}
+  resources:
+    requests:
+      storage: {{ .Values.persistence.size | quote }}
diff --git a/charts/operator-wandb/charts/weave/values.yaml b/charts/operator-wandb/charts/weave/values.yaml
@@ -43,3 +43,7 @@ resources:
   limits:
     cpu: 8
     memory: 16Gi
+
+persistence:
+  # storageClassName: ""
+  size: 20Gi
diff --git a/charts/operator-wandb/templates/_storage.tpl b/charts/operator-wandb/templates/_storage.tpl
@@ -1,6 +1,6 @@
 {{- define "wandb.storageClass" -}}
-{{- if .Values.storageClass -}}
-{{-   if (ne .Values.storageClass "") -}}
+{{- if .Values.persistence.storageClass -}}
+{{-   if (ne .Values.persistence.storageClass "") -}}
 {{-     printf "storageClassName: %s" .Values.storageClass -}}
 {{-   end -}}
 {{- else if .Values.global.storageClass -}}

diff --git a/charts/operator-wandb/templates/ingress.yaml b/charts/operator-wandb/templates/ingress.yaml
@@ -4,12 +4,12 @@ metadata:
   name: {{ .Release.Name }}
   labels:
     {{- include "wandb.commonLabels" . | nindent 4 }}
-    {{- if .Values.ingress.labels -}}
-    {{-   toYaml .Values.ingress.labels | nindent 4 }}
+    {{- if .Values.global.ingress.labels -}}
+    {{-   toYaml .Values.global.ingress.labels | nindent 4 }}
     {{- end }}
   annotations:
-    {{- if .Values.ingress.annotations -}}
-    {{-   toYaml .Values.ingress.annotations | nindent 4 }}
+    {{- if .Values.global.ingress.annotations -}}
+    {{-   toYaml .Values.global.ingress.annotations | nindent 4 }}
     {{- end }}
 spec:
   rules:
@@ -24,7 +24,7 @@ spec:
               number: 8082
   defaultBackend:
     service:
-      {{- if eq .Values.ingress.defaultBackend "console" }}
+      {{- if eq .Values.global.ingress.defaultBackend "console" }}
       name: {{ .Release.Name }}-console
       port:
         number: 8082

diff --git a/charts/operator-wandb/values.yaml b/charts/operator-wandb/values.yaml
@@ -14,6 +14,7 @@ global:
   host: "http://localhost:8080"
 
   storageClass: ""
+
   common:
     labels: {}
     annotations: {}
@@ -60,6 +61,11 @@ global:
     parameters: {}
     caCert: ""
 
+  ingress:
+    defaultBackend: "app"
+    annotations: {}
+    labels: {}
+
 parquet:
   install: true
   image:
@@ -93,10 +99,5 @@ mysql:
 redis:
   install: true
 
-ingress:
-  defaultBackend: "app"
-  annotations: {}
-  labels: {}
-
 nameOverride: ""
 fullnameOverride: ""