From 71fa32cd6d6299ece051456ed7c0ad70b56464d3 Mon Sep 17 00:00:00 2001
From: hichem-belhocine <belhocine.hic@gmail.com>
Date: Tue, 18 Jun 2024 11:50:23 +0200
Subject: [PATCH] feat: enable metrics for csi-provisioner

---
 .../templates/controller-deployment.yaml               | 10 ++++++++++
 charts/aws-efs-csi-driver/values.yaml                  |  5 +++++
 2 files changed, 15 insertions(+)

diff --git a/charts/aws-efs-csi-driver/templates/controller-deployment.yaml b/charts/aws-efs-csi-driver/templates/controller-deployment.yaml
index 94e321ca1..709a6cd9e 100644
--- a/charts/aws-efs-csi-driver/templates/controller-deployment.yaml
+++ b/charts/aws-efs-csi-driver/templates/controller-deployment.yaml
@@ -134,12 +134,22 @@ spec:
             {{- if hasKey .Values.controller "leaderElectionLeaseDuration" }}
             - --leader-election-lease-duration={{ .Values.controller.leaderElectionLeaseDuration }}
             {{- end }}
+            {{- if .Values.controller.metrics.enabled }}
+            - --metrics-address=:{{ .Values.controller.metrics.port }}
+            - --metrics-path={{ .Values.controller.metrics.path }}
+            {{- end }}
           env:
             - name: ADDRESS
               value: /var/lib/csi/sockets/pluginproxy/csi.sock
           volumeMounts:
             - name: socket-dir
               mountPath: /var/lib/csi/sockets/pluginproxy/
+          {{- if .Values.controller.metrics.enabled }}
+          ports:
+            - name: metrics
+              containerPort: {{ .Values.controller.metrics.port }}
+              protocol: TCP
+          {{- end }}
           {{- with .Values.sidecars.csiProvisioner.resources }}
           resources: {{ toYaml . | nindent 12 }}
           {{- end }}
diff --git a/charts/aws-efs-csi-driver/values.yaml b/charts/aws-efs-csi-driver/values.yaml
index f6e9013b6..cdca8c9d0 100644
--- a/charts/aws-efs-csi-driver/values.yaml
+++ b/charts/aws-efs-csi-driver/values.yaml
@@ -113,6 +113,11 @@ controller:
     privileged: true
   leaderElectionRenewDeadline: 10s
   leaderElectionLeaseDuration: 15s
+  # Enable metrics for csi-provisioner
+  metrics:
+    enabled: false
+    port: 8080
+    path: /metrics
 
 
 ## Node daemonset variables