pkg/monitortests/testframework/watchnamespaces/namespaces.go - main - githubmirror/origin

pkg/monitortests/testframework/watchnamespaces/namespaces.go
85 строк3 KB
Kevin Hannon
exclude must gather from platform force delete
23 сен 2024, 16:54
23 сен 2024, 16:54bd50a5a
package watchnamespaces

import (
	"context"

	"github.com/openshift/origin/pkg/monitor/monitorapi"
	"github.com/openshift/origin/pkg/monitortestlibrary"
	"github.com/openshift/origin/pkg/monitortestlibrary/platformidentification"
	corev1 "k8s.io/api/core/v1"
	"k8s.io/apimachinery/pkg/fields"
	"k8s.io/apimachinery/pkg/util/sets"
	"k8s.io/client-go/kubernetes"
	"k8s.io/client-go/tools/cache"
)

// allObservedPlatformNamespaces contains a list of namespaces observed
var allObservedPlatformNamespaces = sets.Set[string]{}

func startNamespaceMonitoring(ctx context.Context, m monitorapi.RecorderWriter, client kubernetes.Interface) {
	namespaceChangeFns := []func(namespace, oldNamespace *corev1.Namespace) []monitorapi.Interval{
		// this is first so namespace created shows up first when queried
		func(namespace, oldNamespace *corev1.Namespace) []monitorapi.Interval {
			// we only care about creates
			if oldNamespace != nil {
				return nil
			}
			if platformidentification.IsPlatformNamespace(namespace.Name) {
				allObservedPlatformNamespaces.Insert(namespace.Name)
			}
			return nil
		},
	}

	listWatch := cache.NewListWatchFromClient(client.CoreV1().RESTClient(), "namespaces", "", fields.Everything())
	customStore := monitortestlibrary.NewMonitoringStore(
		"namespaces",
		toCreateFns(namespaceChangeFns),
		toUpdateFns(namespaceChangeFns),
		toDeleteFns(namespaceChangeFns),
		m,
		m,
	)
	reflector := cache.NewReflector(listWatch, &corev1.Namespace{}, customStore, 0)
	go reflector.Run(ctx.Done())
}

func toCreateFns(namespaceUpdateFns []func(namespace, oldNamespace *corev1.Namespace) []monitorapi.Interval) []monitortestlibrary.ObjCreateFunc {
	ret := []monitortestlibrary.ObjCreateFunc{}

	for i := range namespaceUpdateFns {
		fn := namespaceUpdateFns[i]
		ret = append(ret, func(obj interface{}) []monitorapi.Interval {
			return fn(obj.(*corev1.Namespace), nil)
		})
	}

	return ret
}

func toDeleteFns(namespaceUpdateFns []func(namespace, oldNamespace *corev1.Namespace) []monitorapi.Interval) []monitortestlibrary.ObjDeleteFunc {
	ret := []monitortestlibrary.ObjDeleteFunc{}

	for i := range namespaceUpdateFns {
		fn := namespaceUpdateFns[i]
		ret = append(ret, func(obj interface{}) []monitorapi.Interval {
			return fn(nil, obj.(*corev1.Namespace))
		})
	}
	return ret
}
func toUpdateFns(namespaceUpdateFns []func(namespace, oldNamespace *corev1.Namespace) []monitorapi.Interval) []monitortestlibrary.ObjUpdateFunc {
	ret := []monitortestlibrary.ObjUpdateFunc{}

	for i := range namespaceUpdateFns {
		fn := namespaceUpdateFns[i]
		ret = append(ret, func(obj, oldObj interface{}) []monitorapi.Interval {
			if oldObj == nil {
				return fn(obj.(*corev1.Namespace), nil)
			}
			return fn(obj.(*corev1.Namespace), oldObj.(*corev1.Namespace))
		})
	}

	return ret
}