golang调用k8s API操作servicemonitor

go 获取k8s servicemonitor

package test
import (
	"context"
	"fmt"
	monitoringv1 "github.com/prometheus-operator/prometheus-operator/pkg/apis/monitoring/v1"
	"k8s.io/apimachinery/pkg/runtime"
	"k8s.io/client-go/kubernetes/scheme"

	"k8s.io/client-go/rest"
	"k8s.io/client-go/tools/clientcmd"
	"os"
	"testing"
)

func TestAddEndpoints(t *testing.T) {
	_ = monitoringv1.AddToScheme(scheme.Scheme)
	// 加载 Kubeconfig
	kubeconfig := os.Getenv("KUBECONFIG")
	if kubeconfig == "" {
		kubeconfig = "E:/kubeconfig" // 替换为你的 kubeconfig 路径
	}
	config, err := clientcmd.BuildConfigFromFlags("", kubeconfig)
	if err != nil {
		panic(err.Error())
	}
	
	config.GroupVersion = &monitoringv1.SchemeGroupVersion
	config.APIPath = "/apis"
	config.NegotiatedSerializer = scheme.Codecs

	restClient, err := rest.RESTClientFor(config)
	if err != nil {
		fmt.Println(err)
	}

	// 获取现有的 ServiceMonitor
	svcMonitorClient := restClient.
		Get().
		Resource("servicemonitors").
		Namespace("cloudos-dbaas-nosql").    // 替换为你的命名空间
		Name("redis-monitor-servicemonitor") // 替换为你的 ServiceMonitor 名称

	svcMonitor := &monitoringv1.ServiceMonitor{}
	err = svcMonitorClient.Do(context.Background()).Into(svcMonitor)
	if err != nil {
		panic(err.Error())
	}

	fmt.Println(svcMonitor)
}

go 向servicemonitor中添加endpoints

package test

import (
	"context"
	"fmt"
	monitoringv1 "github.com/prometheus-operator/prometheus-operator/pkg/apis/monitoring/v1"
	"k8s.io/client-go/kubernetes/scheme"

	"k8s.io/client-go/rest"
	"k8s.io/client-go/tools/clientcmd"
	"os"
	"testing"
)


func TestAddEndpoints(t *testing.T) {
	_ = monitoringv1.AddToScheme(scheme.Scheme)

	// 加载 Kubeconfig
	kubeconfig := os.Getenv("KUBECONFIG")
	if kubeconfig == "" {
		kubeconfig = "E:/kubeconfig" // 替换为你的 kubeconfig 路径
	}
	config, err := clientcmd.BuildConfigFromFlags("", kubeconfig)
	if err != nil {
		panic(err.Error())
	}
	config.GroupVersion = &monitoringv1.SchemeGroupVersion
	config.APIPath = "/apis"
	config.NegotiatedSerializer = scheme.Codecs

	restClient, err := rest.RESTClientFor(config)
	if err != nil {
		fmt.Println(err)
	}
	// 获取现有的 ServiceMonitor
	svcMonitorClient := restClient.
		Get().
		Resource("servicemonitors").
		Namespace("cloudos-dbaas-nosql").    // 替换为你的命名空间
		Name("redis-monitor-servicemonitor") // 替换为你的 ServiceMonitor 名称

	svcMonitor := &monitoringv1.ServiceMonitor{}
	err = svcMonitorClient.Do(context.Background()).Into(svcMonitor)
	if err != nil {
		panic(err.Error())
	}

	// 添加新的 endpoints
	newEndpoint := monitoringv1.Endpoint{
		Port:     "http",     // 替换为你的服务端口名
		Path:     "/metrics", // 替换为你的指标路径
		Interval: "30s",
		Scheme:   "HTTP",
	}

	// 添加新 endpoints
	svcMonitor.Spec.Endpoints = append(svcMonitor.Spec.Endpoints, newEndpoint)
	// 更新 ServiceMonitor
	err = restClient.
		Put().
		Resource("servicemonitors").
		Namespace(svcMonitor.Namespace). // 替换为你的命名空间
		Name(svcMonitor.Name).
		Body(svcMonitor).
		Do(context.Background()).Into(svcMonitor)
	if err != nil {
		panic(err.Error())
	}

	fmt.Printf("Updated ServiceMonitor %s with new endpoints\n", err)
}