truenas/library/common/templates/lib/pod/_runtimeClassName.tpl

{{/* Returns Runtime Class Name */}}
{{/* Call this template:
{{ include "ix.v1.common.lib.pod.runtimeClassName" (dict "rootCtx" $ "objectData" $objectData) }}
rootCtx: The root context of the chart.
objectData: The object data to be used to render the Pod.
*/}}
{{- define "ix.v1.common.lib.pod.runtimeClassName" -}}
  {{- $rootCtx := .rootCtx -}}
  {{- $objectData := .objectData -}}

  {{- $runtime := "" -}}

  {{/* Initialize from the "defaults" */}}
  {{- with $rootCtx.Values.podOptions.runtimeClassName -}}
    {{- $runtime = tpl . $rootCtx -}}
  {{- end -}}

  {{/* Override from the pod values, if defined */}}
  {{- with $objectData.podSpec.runtimeClassName -}}
    {{- $runtime = tpl . $rootCtx -}}
  {{- end -}}

  {{- if hasKey $rootCtx.Values.global "ixChartContext" -}}
    {{- if $rootCtx.Values.global.ixChartContext.addNvidiaRuntimeClass -}}

      {{- range $rootCtx.Values.scaleGPU -}}
        {{- if .gpu -}} {{/* Make sure it has a value... */}}

          {{- if (kindIs "map" .targetSelector) -}}
            {{- range $podName, $containers := .targetSelector -}}
              {{- if eq $objectData.shortName $podName -}} {{/* If the pod is selected */}}
                {{- $runtime = $rootCtx.Values.global.ixChartContext.nvidiaRuntimeClassName -}}
              {{- end -}}
            {{- end -}}

          {{- else if $objectData.primary -}}

            {{/* If the pod is primary and no targetSelector is given, assign to primary */}}
            {{- $runtime = $rootCtx.Values.global.ixChartContext.nvidiaRuntimeClassName -}}

          {{- end -}}
        {{- end -}}
      {{- end -}}
    {{- end -}}
  {{- end -}}

  {{- $runtime -}}
{{- end -}}