Address comments

Signed-off-by: QuentinBisson <[email protected]>
giantswarm · Apr 10, 2024 · bc1f838 · bc1f838
1 parent be81015
commit bc1f838
Show file tree

Hide file tree

Showing 4 changed files with 23 additions and 13 deletions.
diff --git a/internal/controller/cluster_monitoring_controller.go b/internal/controller/cluster_monitoring_controller.go
@@ -118,7 +118,7 @@ func (r *ClusterMonitoringReconciler) reconcile(ctx context.Context, cluster *cl
 	}
 
 	// Create or update PrometheusAgent remote write configuration.
-	err := r.PrometheusAgentService.ReconcileRemoteWriteConfig(ctx, cluster)
+	err := r.PrometheusAgentService.ReconcileRemoteWriteConfiguration(ctx, cluster)
 	if err != nil {
 		logger.Error(err, "failed to create or update prometheus agent remote write config")
 		return ctrl.Result{RequeueAfter: 5 * time.Minute}, errors.WithStack(err)
@@ -139,7 +139,7 @@ func (r *ClusterMonitoringReconciler) reconcileDelete(ctx context.Context, clust
 			}
 		}
 
-		err := r.PrometheusAgentService.DeleteRemoteWriteConfig(ctx, cluster)
+		err := r.PrometheusAgentService.DeleteRemoteWriteConfiguration(ctx, cluster)
 		if err != nil {
 			logger.Error(err, "failed to delete prometheus agent remote write config")
 			return ctrl.Result{RequeueAfter: 5 * time.Minute}, errors.WithStack(err)

diff --git a/pkg/common/types.go b/pkg/common/types.go
@@ -47,6 +47,13 @@ type ManagementCluster struct {
 	Region string
 }
 
+func GetClusterType(cluster *clusterv1.Cluster, mc ManagementCluster) string {
+	if cluster.Name == mc.Name {
+		return "management_cluster"
+	}
+	return "workload_cluster"
+}
+
 func GetClusterProvider(cluster *clusterv1.Cluster) (string, error) {
 	switch cluster.Spec.InfrastructureRef.Kind {
 	case AWSClusterKind:

diff --git a/pkg/monitoring/prometheusagent/config.go b/pkg/monitoring/prometheusagent/config.go
@@ -33,14 +33,9 @@ func (pas PrometheusAgentService) buildRemoteWriteConfig(ctx context.Context,
 		return nil, errors.WithStack(err)
 	}
 
-	clusterType := "workload_cluster"
-	if val, ok := cluster.Labels["cluster.x-k8s.io/cluster-name"]; ok && val == pas.ManagementCluster.Name {
-		clusterType = "management_cluster"
-	}
-
 	externalLabels := map[string]string{
 		"cluster_id":       cluster.Name,
-		"cluster_type":     clusterType,
+		"cluster_type":     common.GetClusterType(cluster, pas.ManagementCluster),
 		"customer":         pas.ManagementCluster.Customer,
 		"installation":     pas.ManagementCluster.Name,
 		"organization":     organization,
@@ -69,6 +64,12 @@ func (pas PrometheusAgentService) buildRemoteWriteConfig(ctx context.Context,
 		return nil, errors.WithStack(err)
 	}
 
+	if currentShards < shards {
+		logger.Info("scaling up shards", "old", currentShards, "new", shards)
+	} else if currentShards > shards {
+		logger.Info("scaling down shards", "old", currentShards, "new", shards)
+	}
+
 	return &corev1.ConfigMap{
 		ObjectMeta: metav1.ObjectMeta{
 			Name:      getPrometheusAgentRemoteWriteConfigName(cluster),

diff --git a/pkg/monitoring/prometheusagent/service.go b/pkg/monitoring/prometheusagent/service.go
@@ -27,14 +27,14 @@ type PrometheusAgentService struct {
 	PrometheusVersion string
 }
 
-// ReconcileRemoteWriteConfig ensures that the prometheus remote write config is present in the cluster.
-func (pas *PrometheusAgentService) ReconcileRemoteWriteConfig(
+// ReconcileRemoteWriteConfiguration ensures that the prometheus remote write config is present in the cluster.
+func (pas *PrometheusAgentService) ReconcileRemoteWriteConfiguration(
 	ctx context.Context, cluster *clusterv1.Cluster) error {
 
 	logger := log.FromContext(ctx).WithValues("cluster", cluster.Name)
 	logger.Info("ensuring prometheus agent remote write configmap and secret")
 
-	err := pas.createOrUpdateConfig(ctx, cluster, logger)
+	err := pas.createOrUpdateConfigMap(ctx, cluster, logger)
 	if err != nil {
 		logger.Error(err, "failed to create or update prometheus agent remote write configmap")
 		return errors.WithStack(err)
@@ -51,7 +51,7 @@ func (pas *PrometheusAgentService) ReconcileRemoteWriteConfig(
 	return nil
 }
 
-func (pas PrometheusAgentService) createOrUpdateConfig(ctx context.Context,
+func (pas PrometheusAgentService) createOrUpdateConfigMap(ctx context.Context,
 	cluster *clusterv1.Cluster, logger logr.Logger) error {
 
 	objectKey := client.ObjectKey{
@@ -150,7 +150,9 @@ func (pas PrometheusAgentService) createOrUpdateSecret(ctx context.Context,
 	return nil
 }
 
-func (pas *PrometheusAgentService) DeleteRemoteWriteConfig(ctx context.Context, cluster *clusterv1.Cluster) error {
+func (pas *PrometheusAgentService) DeleteRemoteWriteConfiguration(
+	ctx context.Context, cluster *clusterv1.Cluster) error {
+
 	logger := log.FromContext(ctx).WithValues("cluster", cluster.Name)
 	logger.Info("deleting prometheus agent remote write configmap and secret")