feat: add ServiceMonitor crd

Signed-off-by: joshua <i@joshua.su>
CQUPTMirror · Aug 26, 2024 · f4c441b · f4c441b
1 parent 619c423
commit f4c441b
Show file tree

Hide file tree

Showing 3 changed files with 23 additions and 3 deletions.
diff --git a/cmd/main.go b/cmd/main.go
@@ -19,6 +19,8 @@ package main
 
 import (
 	"flag"
+	monitoringv1 "github.com/prometheus-operator/prometheus-operator/pkg/apis/monitoring/v1"
+	"log"
 	"os"
 	"reflect"
 	"strings"
@@ -94,6 +96,11 @@ func main() {
 
 	config := getConfig()
 
+	// Register the ServiceMonitor API
+	if err := monitoringv1.AddToScheme(mgr.GetScheme()); err != nil {
+		log.Fatalf("unable to register ServiceMonitor API: %v", err)
+	}
+
 	if err = (&controller.JobReconciler{
 		Client: mgr.GetClient(),
 		Scheme: mgr.GetScheme(),

diff --git a/config/rbac/role.yaml b/config/rbac/role.yaml
@@ -32,6 +32,7 @@ rules:
   - ""
   resources:
   - persistentvolumeclaims
+  - configmaps
   verbs:
   - create
   - delete
@@ -206,3 +207,15 @@ rules:
   - patch
   - update
   - watch
+- apiGroups:
+    - monitoring.coreos.com
+  resources:
+    - servicemonitors
+  verbs:
+    - create
+    - delete
+    - get
+    - list
+    - patch
+    - update
+    - watch
diff --git a/internal/controller/utils.go b/internal/controller/utils.go
@@ -4,9 +4,9 @@ import "github.com/CQUPTMirror/kubesync/api/v1beta1"
 
 func getCommonLabels(job *v1beta1.Job) map[string]string {
 	labels := map[string]string{
-		"kubernetes.io/app":        job.Name,
-		"kubernetes.io/component":  "mirror",
-		"kubernetes.io/managed-by": "kubesync",
+		"app.kubernetes.io/app":        job.Name,
+		"app.kubernetes.io/component":  "mirror",
+		"app.kubernetes.io/managed-by": "kubesync",
 	}
 
 	return labels