{{- if .Values.memcachedIndexWrites.enabled }}
apiVersion: apps/v1
kind: StatefulSet
metadata:
  name: {{ include "loki.memcachedIndexWritesFullname" . }}
  namespace: {{ .Release.Namespace }}
  labels:
    {{- include "loki.memcachedIndexWritesLabels" . | nindent 4 }}
  {{- with .Values.loki.annotations }}
  annotations:
    {{- toYaml . | nindent 4 }}
  {{- end }}
spec:
  replicas: {{ .Values.memcachedIndexWrites.replicas }}
  updateStrategy:
    rollingUpdate:
      partition: 0
  serviceName: {{ include "loki.memcachedIndexWritesFullname" . }}
  revisionHistoryLimit: {{ .Values.loki.revisionHistoryLimit }}
  selector:
    matchLabels:
      {{- include "loki.memcachedIndexWritesSelectorLabels" . | nindent 6 }}
  template:
    metadata:
      {{- if or .Values.loki.podAnnotations .Values.memcachedIndexWrites.podAnnotations }}
      annotations:
        {{- with .Values.loki.podAnnotations }}
        {{- toYaml . | nindent 8 }}
        {{- end }}
        {{- with .Values.memcachedIndexWrites.podAnnotations }}
        {{- toYaml . | nindent 8 }}
        {{- end }}
      {{- end }}
      labels:
        {{- include "loki.memcachedIndexWritesSelectorLabels" . | nindent 8 }}
        {{- with .Values.loki.podLabels }}
        {{- toYaml . | nindent 8 }}
        {{- end }}
        {{- with .Values.memcachedIndexWrites.podLabels }}
        {{- toYaml . | nindent 8 }}
        {{- end }}
    spec:
      serviceAccountName: {{ include "loki.serviceAccountName" . }}
      {{- with .Values.imagePullSecrets }}
      imagePullSecrets:
        {{- toYaml . | nindent 8 }}
      {{- end }}
      {{- with .Values.memcachedIndexWrites.hostAliases }}
      hostAliases:
        {{- toYaml . | nindent 8 }}
      {{- end }}
      {{- include "loki.memcachedIndexWritesPriorityClassName" . | nindent 6 }}
      securityContext:
        {{- toYaml .Values.memcached.podSecurityContext | nindent 8 }}
      terminationGracePeriodSeconds: {{ .Values.memcachedIndexWrites.terminationGracePeriodSeconds }}
      containers:
        - name: memcached
          {{- if .Values.memcachedIndexWrites.persistence.enabled }}
          lifecycle:
            preStop:
              exec:
                command:
                - /bin/sh
                - -ec
                - |
                  /usr/bin/pkill -10 memcached
                  sleep 60s
          {{- end }}
          image: {{ include "loki.memcachedImage" . }}
          imagePullPolicy: {{ .Values.memcached.image.pullPolicy }}
          {{- if or .Values.memcachedIndexWrites.extraArgs .Values.memcachedIndexWrites.persistence.enabled }}
          args:
          {{- if .Values.memcachedIndexWrites.persistence.enabled }}
            - --memory-file=/cache-state/memory_file
          {{- end }}
          {{- with .Values.memcachedIndexWrites.extraArgs }}
            {{- toYaml . | nindent 12 }}
          {{- end }}
          {{- end }}
          ports:
            - name: http
              containerPort: 11211
              protocol: TCP
          {{- with .Values.memcachedIndexWrites.extraEnv }}
          env:
            {{- toYaml . | nindent 12 }}
          {{- end }}
          {{- with .Values.memcachedIndexWrites.extraEnvFrom }}
          envFrom:
            {{- toYaml . | nindent 12 }}
          {{- end }}
          securityContext:
            {{- toYaml .Values.memcached.containerSecurityContext | nindent 12 }}
          readinessProbe:
            {{- toYaml .Values.memcached.readinessProbe | nindent 12 }}
          livenessProbe:
            {{- toYaml .Values.memcached.livenessProbe | nindent 12 }}
          {{- if .Values.memcachedIndexWrites.persistence.enabled }}
          volumeMounts:
            - name: data
              mountPath: /cache-state
          {{- end }}
          resources:
            {{- toYaml .Values.memcachedIndexWrites.resources | nindent 12 }}
        {{- if .Values.memcachedExporter.enabled }}
        - name: exporter
          args:
            - --memcached.address=localhost:11211
            - --web.listen-address=0.0.0.0:9150
          image: {{ include "loki.memcachedExporterImage" . }}
          imagePullPolicy: {{ .Values.memcachedExporter.image.pullPolicy }}
          ports:
            - name: http-metrics
              containerPort: 9150
              protocol: TCP
          securityContext:
            {{- toYaml .Values.memcachedExporter.containerSecurityContext | nindent 12 }}
          resources:
            {{- toYaml .Values.memcachedExporter.resources | nindent 12 }}
        {{- end }}
        {{- if .Values.memcachedIndexWrites.extraContainers }}
        {{- toYaml .Values.memcachedIndexWrites.extraContainers | nindent 8}}
        {{- end }}
      {{- with .Values.memcachedIndexWrites.affinity }}
      affinity:
        {{- tpl . $ | nindent 8 }}
      {{- end }}
      {{- with .Values.memcachedIndexWrites.nodeSelector }}
      nodeSelector:
        {{- toYaml . | nindent 8 }}
      {{- end }}
      {{- with .Values.memcachedIndexWrites.tolerations }}
      tolerations:
        {{- toYaml . | nindent 8 }}
      {{- end }}
  {{- if .Values.memcachedIndexWrites.persistence.enabled }}
  volumeClaimTemplates:
    - metadata:
        name: data
      spec:
        accessModes:
          - ReadWriteOnce
        {{- with .Values.memcachedIndexWrites.persistence.storageClass }}
        storageClassName: {{ if (eq "-" .) }}""{{ else }}{{ . }}{{ end }}
        {{- end }}
        resources:
          requests:
            storage: {{ .Values.memcachedIndexWrites.persistence.size | quote }}
  {{- end }}
{{- end }}