deployment.yaml

{{- if .Values.agent.enabled -}}
apiVersion: apps/v1
kind: Deployment
metadata:
  name: {{ include "prefect-server.nameField" (merge (dict "component" "agent") .) }}
  labels: 
    {{- include "prefect-server.commonLabels" . | nindent 4 }}
    app.kubernetes.io/component: agent
  annotations:
    {{- merge .Values.agent.annotations .Values.annotations | toYaml | nindent 4 }}
spec:
  replicas: {{ .Values.agent.replicas }}
  {{- with .Values.agent.strategy }}
  strategy:
    {{- toYaml . | nindent 4 }}
  {{- end }}
  selector:
    matchLabels:
      {{- include "prefect-server.matchLabels" . | nindent 6 }}
      app.kubernetes.io/component: agent
  template:
    metadata:
      labels:
        {{- include "prefect-server.matchLabels" . | nindent 8 }}
        app.kubernetes.io/component: agent
      {{- with .Values.agent.podAnnotations }}
      annotations: {{- toYaml . | nindent 8 }}
      {{- end }}
    spec:
    {{- if .Values.agent.image.pullSecrets }}
      imagePullSecrets:
        {{- toYaml .Values.agent.image.pullSecrets | nindent 8 }}
    {{- else if .Values.imagePullSecrets }}
      imagePullSecrets:
        {{- toYaml .Values.imagePullSecrets | nindent 8 }}
    {{- end }}
      serviceAccountName: {{ include "prefect-server.serviceAccountName" . }}
      {{- with .Values.agent.podSecurityContext }}
      securityContext:
        {{- toYaml . | nindent 8 }}
      {{- end }}
      containers:
      - name: agent
        {{- with .Values.agent.podSecurityContext }}
        securityContext:
          {{- toYaml . | nindent 8 }}
        {{- end }}
        image: "{{ .Values.agent.image.name }}:{{ .Values.agent.image.tag | default .Values.prefectVersionTag }}"
        imagePullPolicy: {{ .Values.agent.image.pullPolicy  }}
        command:
          - bash
          - "-c"
          {{- if .Values.agent.jobTemplateFilePath }}
          - "prefect agent kubernetes start --job-template {{ .Values.agent.jobTemplateFilePath }}"
          {{- else }}
          - "prefect agent kubernetes start"
          {{- end }}
        env:
          - name: PREFECT__CLOUD__API
            value: {{ include "prefect-server.apollo-api-url" . }}
          - name: NAMESPACE
            value: {{ .Release.Namespace }}
        {{- if .Values.agent.job.imagePullSecrets }}
          - name: IMAGE_PULL_SECRETS
            value: {{ .Values.agent.job.imagePullSecrets | join "," | quote }}
        {{- end }}
          - name: PREFECT__CLOUD__AGENT__LABELS
            value: {{ .Values.agent.prefectLabels | toJson | quote }}
        {{- if .Values.agent.job.resources.requests.memory }}
          - name: JOB_MEM_REQUEST
            value: {{ .Values.agent.job.resources.requests.memory | quote }}
        {{- end }}
        {{- if .Values.agent.job.resources.limits.memory }}
          - name: JOB_MEM_LIMIT
            value: {{ .Values.agent.job.resources.limits.memory | quote }}
        {{- end }}
        {{- if .Values.agent.job.resources.requests.cpu }}
          - name: JOB_CPU_REQUEST
            value: {{ .Values.agent.job.resources.requests.cpu | quote }}
        {{- end }}
        {{- if .Values.agent.job.resources.limits.cpu }}
          - name: JOB_CPU_LIMIT
            value: {{ .Values.agent.job.resources.limits.cpu | quote }}
        {{- end }}
        {{- if .Values.agent.job.imagePullPolicy }}
          - name: IMAGE_PULL_POLICY
            value: {{ .Values.agent.job.imagePullPolicy | quote }}
        {{- end }}
          - name: SERVICE_ACCOUNT_NAME
            value: {{ include "prefect-server.serviceAccountName" . }}
          - name: PREFECT__BACKEND
            value: server
          - name: PREFECT__CLOUD__AGENT__AGENT_ADDRESS
            value: http://0.0.0.0:8080
          {{- with .Values.agent.env }}
            {{- toYaml . | nindent 10 }}
          {{- end }}
    {{- if .Values.agent.jobTemplateFileConfigMap }}
        volumeMounts:
          - mountPath: {{ .Values.agent.jobTemplateFilePath }}
            name: job-template
            subPath: {{ base .Values.agent.jobTemplateFilePath }}
    {{- end }}
        livenessProbe:
          failureThreshold: 2
          httpGet:
            path: /api/health
            port: 8080
          initialDelaySeconds: 40
          periodSeconds: 40
        {{- with .Values.agent.resources }}
        resources:
          {{- toYaml . | nindent 12 }}
        {{- end }}
      {{- with .Values.agent.nodeSelector }}
      nodeSelector:
        {{- toYaml . | nindent 8 }}
      {{- end }}
      {{- if .Values.agent.jobTemplateFileConfigMap }}
      volumes:
      - configMap:
          name: {{ .Values.agent.jobTemplateFileConfigMap }}
        name: job-template
      {{- end }}
    {{- with .Values.agent.affinity }}
      affinity:
        {{- toYaml . | nindent 8 }}
    {{- end }}
    {{- with .Values.agent.tolerations }}
      tolerations:
        {{- toYaml . | nindent 8 }}
    {{- end }}
{{- end }}