Merge pull request #15 from prometheus-operator/chore/adding-create-s…

…vc-monitor

[CHORE] adding create svc monitor command

cmd/servicemonitor.go

@@ -0,0 +1,138 @@
+// Copyright 2024 The prometheus-operator Authors
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package cmd
+
+import (
+	"context"
+	"errors"
+	"fmt"
+
+	"github.com/prometheus-operator/poctl/internal/k8sutil"
+	"github.com/prometheus-operator/poctl/internal/log"
+	monitoringv1 "github.com/prometheus-operator/prometheus-operator/pkg/client/applyconfiguration/monitoring/v1"
+	monitoringclient "github.com/prometheus-operator/prometheus-operator/pkg/client/versioned"
+	"github.com/spf13/cobra"
+	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
+	applyConfigMetav1 "k8s.io/client-go/applyconfigurations/meta/v1"
+	"k8s.io/client-go/kubernetes"
+	"k8s.io/utils/ptr"
+)
+
+var (
+	serviceName       string
+	namespace         string
+	port              string
+	servicemonitorCmd = &cobra.Command{
+		Use:   "servicemonitor",
+		Short: "Create a service monitor object",
+		Long:  `Create a service monitor object based on user input parameters or taking as source of truth a kubernetes service`,
+		RunE:  runServiceMonitor,
+	}
+)
+
+func runServiceMonitor(_ *cobra.Command, _ []string) error {
+	logger, err := log.NewLogger()
+	if err != nil {
+		fmt.Println(err)
+		return err
+	}
+
+	//TODO(nicolastakashi): Replace it when the PR #6623 is merged.
+	restConfig, err := k8sutil.GetRestConfig(logger, kubeconfig)
+	if err != nil {
+		logger.With("error", err.Error()).Error("error while getting kubeconfig")
+		return err
+	}
+
+	kclient, err := kubernetes.NewForConfig(restConfig)
+	if err != nil {
+		logger.With("error", err.Error()).Error("error while creating k8s client")
+		return err
+	}
+
+	mclient, err := monitoringclient.NewForConfig(restConfig)
+	if err != nil {
+		logger.With("error", err.Error()).Error("error while creating Prometheus Operator client")
+		return err
+	}
+
+	if serviceName == "" {
+		return errors.New("service name is required")
+	}
+
+	err = createFromService(context.Background(), kclient, mclient, namespace, serviceName, port)
+	if err != nil {
+		logger.With("error", err.Error()).Error("error while creating service monitor")
+		return err
+	}
+
+	return nil
+}
+
+func createFromService(
+	ctx context.Context,
+	k8sClient *kubernetes.Clientset,
+	mClient *monitoringclient.Clientset,
+	namespace string,
+	serviceName string,
+	port string) error {
+
+	service, err := k8sClient.CoreV1().Services(namespace).Get(ctx, serviceName, metav1.GetOptions{})
+	if err != nil {
+		return fmt.Errorf("error while getting service %s: %v", serviceName, err)
+	}
+
+	svcMonitor := &monitoringv1.ServiceMonitorApplyConfiguration{
+		TypeMetaApplyConfiguration: applyConfigMetav1.TypeMetaApplyConfiguration{
+			Kind:       ptr.To("ServiceMonitor"),
+			APIVersion: ptr.To("monitoring.coreos.com/v1"),
+		},
+		ObjectMetaApplyConfiguration: &applyConfigMetav1.ObjectMetaApplyConfiguration{
+			Name:      ptr.To(serviceName),
+			Namespace: ptr.To(namespace),
+			Labels:    service.Labels,
+		},
+		Spec: &monitoringv1.ServiceMonitorSpecApplyConfiguration{
+			Selector: &metav1.LabelSelector{
+				MatchLabels: service.Spec.Selector,
+			},
+		},
+	}
+
+	for _, p := range service.Spec.Ports {
+		if port != "" && p.Name != port {
+			continue
+		}
+
+		svcMonitor.Spec.Endpoints = append(svcMonitor.Spec.Endpoints, monitoringv1.EndpointApplyConfiguration{
+			HonorLabels: ptr.To(true),
+			Port:        ptr.To(p.Name),
+		})
+	}
+
+	_, err = mClient.MonitoringV1().ServiceMonitors(namespace).Apply(ctx, svcMonitor, k8sutil.ApplyOption)
+	if err != nil {
+		return fmt.Errorf("error while creating service monitor %s: %v", serviceName, err)
+	}
+
+	return nil
+}
+
+func init() {
+	createCmd.AddCommand(servicemonitorCmd)
+	servicemonitorCmd.Flags().StringVarP(&serviceName, "service", "s", "", "Service name to create the service monitor from")
+	servicemonitorCmd.Flags().StringVarP(&namespace, "namespace", "n", "default", "Namespace of the service")
+	servicemonitorCmd.Flags().StringVarP(&port, "port", "p", "", "Port of the service")
+}

cmd/stack.go

@@ -28,6 +28,7 @@ import (
 	"github.com/spf13/cobra"
 	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
 	"k8s.io/apimachinery/pkg/apis/meta/v1/unstructured"
+
 	"k8s.io/apimachinery/pkg/runtime"
 	"k8s.io/apimachinery/pkg/runtime/schema"
 	"k8s.io/client-go/dynamic"
@@ -39,7 +40,7 @@ var (
 		Use:   "stack",
 		Short: "create a stack of Prometheus Operator resources.",
 		Long:  `create a stack of Prometheus Operator resources.`,
-		Run:   run,
+		Run:   runStack,
 	}
 
 	crds = []string{
@@ -70,7 +71,7 @@ func init() {
 	// stackCmd.Flags().BoolP("toggle", "t", false, "Help message for toggle")
 }
 
-func run(cmd *cobra.Command, _ []string) {
+func runStack(cmd *cobra.Command, _ []string) {
 	logger, err := log.NewLogger()
 	if err != nil {
 		fmt.Println(err)