[CHORE] adding recording rule group set suffix flag] (#125)

Signed-off-by: Nicolas Takashi <nicolas.takashi@coralogix.com>
coralogix · Jul 31, 2024 · 95feab7 · 95feab7
1 parent ac9d221
commit 95feab7
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 7 deletions.
diff --git a/controllers/alphacontrollers/recordingrulegroupset_controller.go b/controllers/alphacontrollers/recordingrulegroupset_controller.go
@@ -41,8 +41,9 @@ var recordingRuleGroupSetFinalizerName = "recordingrulegroupset.coralogix.com/fi
 // RecordingRuleGroupSetReconciler reconciles a RecordingRuleGroupSet object
 type RecordingRuleGroupSetReconciler struct {
 	client.Client
-	CoralogixClientSet clientset.ClientSetInterface
-	Scheme             *runtime.Scheme
+	CoralogixClientSet          clientset.ClientSetInterface
+	Scheme                      *runtime.Scheme
+	RecordingRuleGroupSetSuffix string
 }
 
 //+kubebuilder:rbac:groups=coralogix.com,resources=recordingrulegroupsets,verbs=get;list;watch;create;update;patch;delete
@@ -97,7 +98,7 @@ func (r *RecordingRuleGroupSetReconciler) create(ctx context.Context, recordingR
 	response, err := r.CoralogixClientSet.
 		RecordingRuleGroups().
 		CreateRecordingRuleGroupSet(ctx, &rrg.CreateRuleGroupSet{
-			Name:   ptr.To(recordingRuleGroupSet.Name),
+			Name:   ptr.To(fmt.Sprintf("%s%s", recordingRuleGroupSet.Name, r.RecordingRuleGroupSetSuffix)),
 			Groups: recordingRuleGroupSet.Spec.ExtractRecordingRuleGroups(),
 		})
 
@@ -143,7 +144,7 @@ func (r *RecordingRuleGroupSetReconciler) update(ctx context.Context, recordingR
 		UpdateRecordingRuleGroupSet(ctx, &rrg.UpdateRuleGroupSet{
 			Id:     remoteRecordingRule.Id,
 			Groups: recordingRuleGroupSet.Spec.ExtractRecordingRuleGroups(),
-			Name:   ptr.To(recordingRuleGroupSet.Name),
+			Name:   ptr.To(fmt.Sprintf("%s%s", recordingRuleGroupSet.Name, r.RecordingRuleGroupSetSuffix)),
 		}); err != nil {
 		return fmt.Errorf("failed to update recording rule groupSet: %w", err)
 	}

diff --git a/main.go b/main.go
@@ -87,6 +87,9 @@ func main() {
 	var prometheusRuleController bool
 	flag.BoolVar(&prometheusRuleController, "prometheus-rule-controller", true, "Determine if the prometheus rule controller should be started. Default is true.")
 
+	var recordingRuleGroupSetSuffix string
+	flag.StringVar(&recordingRuleGroupSetSuffix, "recording-rule-group-set-suffix", "", "Suffix to be added to the RecordingRuleGroupSet")
+
 	opts := zap.Options{}
 	opts.BindFlags(flag.CommandLine)
 	flag.Parse()
@@ -158,9 +161,10 @@ func main() {
 		}
 	}
 	if err = (&alphacontrollers.RecordingRuleGroupSetReconciler{
-		CoralogixClientSet: clientset.NewClientSet(targetUrl, apiKey),
-		Client:             mgr.GetClient(),
-		Scheme:             mgr.GetScheme(),
+		CoralogixClientSet:          clientset.NewClientSet(targetUrl, apiKey),
+		Client:                      mgr.GetClient(),
+		Scheme:                      mgr.GetScheme(),
+		RecordingRuleGroupSetSuffix: recordingRuleGroupSetSuffix,
 	}).SetupWithManager(mgr); err != nil {
 		setupLog.Error(err, "unable to create controller", "controller", "RecordingRuleGroupSet")
 		os.Exit(1)