Add multi cluster support

Icinga · Dec 18, 2024 · 5310012 · 5310012
1 parent 28cab7f
commit 5310012
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 7 deletions.
diff --git a/cmd/icinga-kubernetes/main.go b/cmd/icinga-kubernetes/main.go
@@ -294,7 +294,7 @@ func main() {
 		})
 	}
 
-	err = internal.SyncPrometheusConfig(ctx, db2, &cfg.Prometheus)
+	err = internal.SyncPrometheusConfig(ctx, db2, &cfg.Prometheus, clusterInstance.Uuid)
 	if err != nil {
 		klog.Error(errors.Wrap(err, "cannot sync prometheus config"))
 	}

diff --git a/internal/prometheus.go b/internal/prometheus.go
@@ -16,29 +16,30 @@ import (
 	"strings"
 )
 
-func SyncPrometheusConfig(ctx context.Context, db *database.DB, config *metrics.PrometheusConfig) error {
+func SyncPrometheusConfig(ctx context.Context, db *database.DB, config *metrics.PrometheusConfig, clusterUuid types.UUID) error {
 	_true := types.Bool{Bool: true, Valid: true}
 
 	if config.Url != "" {
 		toDb := []schemav1.Config{
-			{Key: schemav1.ConfigKeyPrometheusUrl, Value: config.Url, Locked: _true},
+			{ClusterUuid: clusterUuid, Key: schemav1.ConfigKeyPrometheusUrl, Value: config.Url, Locked: _true},
 		}
 
 		if config.Username != "" {
 			toDb = append(
 				toDb,
-				schemav1.Config{Key: schemav1.ConfigKeyPrometheusUsername, Value: config.Username, Locked: _true},
-				schemav1.Config{Key: schemav1.ConfigKeyPrometheusPassword, Value: config.Password, Locked: _true},
+				schemav1.Config{ClusterUuid: clusterUuid, Key: schemav1.ConfigKeyPrometheusUsername, Value: config.Username, Locked: _true},
+				schemav1.Config{ClusterUuid: clusterUuid, Key: schemav1.ConfigKeyPrometheusPassword, Value: config.Password, Locked: _true},
 			)
 		}
 
 		err := db.ExecTx(ctx, func(ctx context.Context, tx *sqlx.Tx) error {
 			if _, err := tx.ExecContext(
 				ctx,
 				fmt.Sprintf(
-					`DELETE FROM "%s" WHERE "key" LIKE ? AND "locked" = ?`,
+					`DELETE FROM "%s" WHERE "cluster_uuid" = ? AND "key" LIKE ? AND "locked" = ?`,
 					database.TableName(&schemav1.Config{}),
 				),
+				clusterUuid,
 				`prometheus.%`,
 				_true,
 			); err != nil {
@@ -60,9 +61,10 @@ func SyncPrometheusConfig(ctx context.Context, db *database.DB, config *metrics.
 			if _, err := tx.ExecContext(
 				ctx,
 				fmt.Sprintf(
-					`DELETE FROM "%s" WHERE "key" LIKE ? AND "locked" = ?`,
+					`DELETE FROM "%s" WHERE "cluster_uuid" = ? AND "key" LIKE ? AND "locked" = ?`,
 					database.TableName(&schemav1.Config{}),
 				),
+				clusterUuid,
 				`prometheus.%`,
 				_true,
 			); err != nil {