From a3e442b342e86adbc9cb5d0abb080dbf946c9f82 Mon Sep 17 00:00:00 2001
From: coderth <coderth@outlook.com>
Date: Tue, 20 Feb 2024 17:05:00 +0800
Subject: [PATCH] feat: Add servicemonitor switch for metrics acquisition

Signed-off-by: coderth <coderth@outlook.com>
---
 .../device-plugin/servicemonitor.yaml         | 21 +++++++++++++++++++
 .../templates/scheduler/servicemonitor.yaml   | 21 +++++++++++++++++++
 charts/hami/values.yaml                       |  4 ++++
 3 files changed, 46 insertions(+)
 create mode 100644 charts/hami/templates/device-plugin/servicemonitor.yaml
 create mode 100644 charts/hami/templates/scheduler/servicemonitor.yaml

diff --git a/charts/hami/templates/device-plugin/servicemonitor.yaml b/charts/hami/templates/device-plugin/servicemonitor.yaml
new file mode 100644
index 000000000..c56d591a9
--- /dev/null
+++ b/charts/hami/templates/device-plugin/servicemonitor.yaml
@@ -0,0 +1,21 @@
+{{- if .Values.devicePlugin.serviceMonitor.enable }}
+apiVersion: monitoring.coreos.com/v1
+kind: ServiceMonitor
+metadata:
+  name: vgpu-monitor
+  namespace: {{ .Release.Namespace | quote }}
+spec:
+  endpoints:
+  - bearerTokenSecret:
+      key: ""
+    interval: 15s
+    path: /metrics
+    port: monitorport
+  jobLabel: app
+  namespaceSelector:
+    matchNames:
+    - {{ .Release.Namespace | quote }}
+  selector:
+    matchLabels:
+      app.kubernetes.io/component: 4pd-scheduler
+{{- end }}
\ No newline at end of file
diff --git a/charts/hami/templates/scheduler/servicemonitor.yaml b/charts/hami/templates/scheduler/servicemonitor.yaml
new file mode 100644
index 000000000..772dcf292
--- /dev/null
+++ b/charts/hami/templates/scheduler/servicemonitor.yaml
@@ -0,0 +1,21 @@
+{{- if .Values.scheduler.serviceMonitor.enable }}
+apiVersion: monitoring.coreos.com/v1
+kind: ServiceMonitor
+metadata:
+  name: vgpu-scheduler
+  namespace: {{ .Release.Namespace | quote }}
+spec:
+  endpoints:
+  - bearerTokenSecret:
+      key: ""
+    interval: 15s
+    path: /metrics
+    port: monitor
+  jobLabel: app
+  namespaceSelector:
+    matchNames:
+    - {{ .Release.Namespace | quote }}
+  selector:
+    matchLabels:
+      app.kubernetes.io/component: 4pd-scheduler
+{{- end }}
\ No newline at end of file
diff --git a/charts/hami/values.yaml b/charts/hami/values.yaml
index aae46d1ed..d5fdb8c44 100644
--- a/charts/hami/values.yaml
+++ b/charts/hami/values.yaml
@@ -56,6 +56,8 @@ scheduler:
     extraArgs:
       - --debug
       - -v=4
+  serviceMonitor:
+    enable: false
   podAnnotations: {}
   #nodeSelector: 
   #  gpu: "on"
@@ -118,4 +120,6 @@ devicePlugin:
   hygonnodeSelector:
     dcu: "on"
   tolerations: []
+  serviceMonitor:
+    enable: false