From 18778f44a640ee33d6a01d592db902a73ce0b87a Mon Sep 17 00:00:00 2001
From: Thomas Petit <thomas.marc.petit@gmail.com>
Date: Mon, 3 Jun 2024 20:08:35 +0200
Subject: [PATCH] Add runtimeClassName

---
 charts/plex-media-server/README.md                  | 1 +
 charts/plex-media-server/templates/statefulset.yaml | 3 +++
 charts/plex-media-server/values.yaml                | 3 +++
 3 files changed, 7 insertions(+)

diff --git a/charts/plex-media-server/README.md b/charts/plex-media-server/README.md
index 8dc84eb..6a0f779 100644
--- a/charts/plex-media-server/README.md
+++ b/charts/plex-media-server/README.md
@@ -113,6 +113,7 @@ The following table lists the configurable parameters of the Pms-chart chart and
 | `initContainer.image.sha` | Optional SHA digest to specify a specific image rather than a specific tag | `""` |
 | `initContainer.image.pullPolicy` |  | `"IfNotPresent"` |
 | `initContainer.script` | An optional script that will be run by the init container, it can be used on the first run to stop pms from starting when importing a pre-exiting database | `""` |
+| `runtimeClassName` | Specify your own runtime class name eg use gpu | `""` |
 | `rclone.enabled` | If rclone should be used to mount volumes | `false` |
 | `rclone.image.registry` | The registry that should be used to pull the image from | `"index.docker.io"` |
 | `rclone.image.repository` | The docker repo that will be used for the rclone container | `"rclone/rclone"` |
diff --git a/charts/plex-media-server/templates/statefulset.yaml b/charts/plex-media-server/templates/statefulset.yaml
index 6ce15f2..131a80c 100644
--- a/charts/plex-media-server/templates/statefulset.yaml
+++ b/charts/plex-media-server/templates/statefulset.yaml
@@ -21,6 +21,9 @@ spec:
       annotations:
         {{- toYaml .Values.statefulSet.podAnnotations | nindent 8 }}
     spec:
+      {{- if .Values.runtimeClassName }}
+      runtimeClassName: {{ .Values.runtimeClassName | quote }}
+      {{- end }}
       serviceAccountName: {{ include "pms-chart.serviceAccountName" . }}
       tolerations:
         {{- toYaml .Values.tolerations | nindent 8 }}
diff --git a/charts/plex-media-server/values.yaml b/charts/plex-media-server/values.yaml
index 4c04783..bdb82d6 100644
--- a/charts/plex-media-server/values.yaml
+++ b/charts/plex-media-server/values.yaml
@@ -83,6 +83,9 @@ initContainer:
   #
   #   echo "Done."
 
+# specify your own runtime class name eg use gpu
+runtimeClassName: "" 
+
 # the settings specific to rclone
 rclone:
   # if the rclone sidecar should be created