mirror of
https://github.com/kubevirt/containerized-data-importer.git
synced 2025-06-03 06:30:22 +00:00

* Set htpp(s)_proxy to lower case env variable CURL used by nbdkit doesn't read upper case http(s)_proxy environment variables, and thus was not using the proxy. Changed the variable to be lower case. Added a significant number of tests to test many more variations of using a proxy. Also added https + auth endpoint to the file-host container, so we can test https + auth with the proxy. Added https endpoint to proxy, so we can test an https proxy. Cleaned up some of the error handling in the import controller for the proxy, in particular if a trustedCAProxy is defined. Fixed some of the cluster wide proxy configuration so it works properly inside an openshift cluster. Signed-off-by: Alexander Wels <awels@redhat.com> * Add https proxy support to registry import. Added extra functional tests to test all registry import combinations Signed-off-by: Alexander Wels <awels@redhat.com> * Fixed some tests to work better in Open Shift. Signed-off-by: Alexander Wels <awels@redhat.com>
640 lines
22 KiB
Go
640 lines
22 KiB
Go
package controller
|
|
|
|
import (
|
|
"context"
|
|
"reflect"
|
|
"regexp"
|
|
|
|
"github.com/go-logr/logr"
|
|
ocpconfigv1 "github.com/openshift/api/config/v1"
|
|
routev1 "github.com/openshift/api/route/v1"
|
|
v1 "k8s.io/api/core/v1"
|
|
networkingv1 "k8s.io/api/networking/v1"
|
|
storagev1 "k8s.io/api/storage/v1"
|
|
"k8s.io/apimachinery/pkg/api/errors"
|
|
"k8s.io/apimachinery/pkg/api/meta"
|
|
"k8s.io/apimachinery/pkg/api/resource"
|
|
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
|
"k8s.io/apimachinery/pkg/runtime"
|
|
"k8s.io/apimachinery/pkg/types"
|
|
"sigs.k8s.io/controller-runtime/pkg/cache"
|
|
"sigs.k8s.io/controller-runtime/pkg/client"
|
|
"sigs.k8s.io/controller-runtime/pkg/controller"
|
|
"sigs.k8s.io/controller-runtime/pkg/event"
|
|
"sigs.k8s.io/controller-runtime/pkg/handler"
|
|
"sigs.k8s.io/controller-runtime/pkg/manager"
|
|
"sigs.k8s.io/controller-runtime/pkg/predicate"
|
|
"sigs.k8s.io/controller-runtime/pkg/reconcile"
|
|
"sigs.k8s.io/controller-runtime/pkg/source"
|
|
|
|
cdiv1 "kubevirt.io/containerized-data-importer-api/pkg/apis/core/v1beta1"
|
|
"kubevirt.io/containerized-data-importer/pkg/common"
|
|
"kubevirt.io/containerized-data-importer/pkg/operator"
|
|
"kubevirt.io/containerized-data-importer/pkg/util"
|
|
)
|
|
|
|
// AnnConfigAuthority is the annotation specifying a resource as the CDIConfig authority
|
|
const (
|
|
AnnConfigAuthority = "cdi.kubevirt.io/configAuthority"
|
|
defaultCPULimit = "750m"
|
|
defaultMemLimit = "600M"
|
|
defaultCPURequest = "100m"
|
|
defaultMemRequest = "60M"
|
|
)
|
|
|
|
// CDIConfigReconciler members
|
|
type CDIConfigReconciler struct {
|
|
client client.Client
|
|
// use this for getting any resources not in the install namespace or cluster scope
|
|
uncachedClient client.Client
|
|
scheme *runtime.Scheme
|
|
log logr.Logger
|
|
uploadProxyServiceName string
|
|
configName string
|
|
cdiNamespace string
|
|
installerLabels map[string]string
|
|
}
|
|
|
|
func isErrCacheNotStarted(err error) bool {
|
|
if err == nil {
|
|
return false
|
|
}
|
|
_, ok := err.(*cache.ErrCacheNotStarted)
|
|
return ok
|
|
}
|
|
|
|
// Reconcile the reconcile loop for the CDIConfig object.
|
|
func (r *CDIConfigReconciler) Reconcile(_ context.Context, req reconcile.Request) (reconcile.Result, error) {
|
|
log := r.log.WithValues("CDIConfig", req.NamespacedName)
|
|
log.Info("reconciling CDIConfig")
|
|
|
|
config, err := r.createCDIConfig()
|
|
if err != nil {
|
|
log.Error(err, "Unable to create CDIConfig")
|
|
return reconcile.Result{}, err
|
|
}
|
|
// Keep a copy of the original for comparison later.
|
|
currentConfigCopy := config.DeepCopyObject()
|
|
|
|
config.Status.Preallocation = config.Spec.Preallocation != nil && *config.Spec.Preallocation
|
|
|
|
// ignore whatever is in config spec and set to operator view
|
|
if err := r.setOperatorParams(config); err != nil {
|
|
return reconcile.Result{}, err
|
|
}
|
|
|
|
if err := r.reconcileUploadProxyURL(config); err != nil {
|
|
return reconcile.Result{}, err
|
|
}
|
|
|
|
if err := r.reconcileStorageClass(config); err != nil {
|
|
return reconcile.Result{}, err
|
|
}
|
|
|
|
if err := r.reconcileDefaultPodResourceRequirements(config); err != nil {
|
|
return reconcile.Result{}, err
|
|
}
|
|
|
|
if err := r.reconcileFilesystemOverhead(config); err != nil {
|
|
return reconcile.Result{}, err
|
|
}
|
|
|
|
if err := r.reconcileImportProxy(config); err != nil {
|
|
return reconcile.Result{}, err
|
|
}
|
|
|
|
if !reflect.DeepEqual(currentConfigCopy, config) {
|
|
// Updates have happened, update CDIConfig.
|
|
log.Info("Updating CDIConfig", "CDIConfig.Name", config.Name, "config", config)
|
|
if err := r.client.Update(context.TODO(), config); err != nil {
|
|
return reconcile.Result{}, err
|
|
}
|
|
}
|
|
|
|
return reconcile.Result{}, nil
|
|
}
|
|
|
|
func (r *CDIConfigReconciler) setOperatorParams(config *cdiv1.CDIConfig) error {
|
|
util.SetRecommendedLabels(config, r.installerLabels, "cdi-controller")
|
|
|
|
cdiCR, err := GetActiveCDI(r.client)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if cdiCR == nil {
|
|
return nil
|
|
}
|
|
|
|
if _, ok := cdiCR.Annotations[AnnConfigAuthority]; !ok {
|
|
return nil
|
|
}
|
|
|
|
if cdiCR.Spec.Config == nil {
|
|
config.Spec = cdiv1.CDIConfigSpec{}
|
|
} else {
|
|
config.Spec = *cdiCR.Spec.Config
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (r *CDIConfigReconciler) reconcileUploadProxyURL(config *cdiv1.CDIConfig) error {
|
|
log := r.log.WithName("CDIconfig").WithName("UploadProxyReconcile")
|
|
config.Status.UploadProxyURL = config.Spec.UploadProxyURLOverride
|
|
// No override, try Ingress
|
|
if config.Status.UploadProxyURL == nil {
|
|
if err := r.reconcileIngress(config); err != nil {
|
|
log.Error(err, "Unable to reconcile Ingress")
|
|
return err
|
|
}
|
|
}
|
|
// No override or Ingress, try Route
|
|
if config.Status.UploadProxyURL == nil {
|
|
if err := r.reconcileRoute(config); err != nil {
|
|
log.Error(err, "Unable to reconcile Routes")
|
|
return err
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (r *CDIConfigReconciler) reconcileIngress(config *cdiv1.CDIConfig) error {
|
|
log := r.log.WithName("CDIconfig").WithName("IngressReconcile")
|
|
ingressList := &networkingv1.IngressList{}
|
|
if err := r.client.List(context.TODO(), ingressList, &client.ListOptions{}); IgnoreIsNoMatchError(err) != nil {
|
|
return err
|
|
}
|
|
for _, ingress := range ingressList.Items {
|
|
ingressURL := getURLFromIngress(&ingress, r.uploadProxyServiceName)
|
|
if ingressURL != "" {
|
|
log.Info("Setting upload proxy url", "IngressURL", ingressURL)
|
|
config.Status.UploadProxyURL = &ingressURL
|
|
return nil
|
|
}
|
|
}
|
|
log.Info("No ingress found, setting to blank", "IngressURL", "")
|
|
config.Status.UploadProxyURL = nil
|
|
return nil
|
|
}
|
|
|
|
func (r *CDIConfigReconciler) reconcileRoute(config *cdiv1.CDIConfig) error {
|
|
log := r.log.WithName("CDIconfig").WithName("RouteReconcile")
|
|
routeList := &routev1.RouteList{}
|
|
if err := r.client.List(context.TODO(), routeList, &client.ListOptions{}); IgnoreIsNoMatchError(err) != nil {
|
|
return err
|
|
}
|
|
for _, route := range routeList.Items {
|
|
routeURL := getURLFromRoute(&route, r.uploadProxyServiceName)
|
|
if routeURL != "" {
|
|
log.Info("Setting upload proxy url", "RouteURL", routeURL)
|
|
config.Status.UploadProxyURL = &routeURL
|
|
return nil
|
|
}
|
|
}
|
|
log.Info("No route found, setting to blank", "RouteURL", "")
|
|
config.Status.UploadProxyURL = nil
|
|
return nil
|
|
}
|
|
|
|
func (r *CDIConfigReconciler) reconcileStorageClass(config *cdiv1.CDIConfig) error {
|
|
log := r.log.WithName("CDIconfig").WithName("StorageClassReconcile")
|
|
storageClassList := &storagev1.StorageClassList{}
|
|
if err := r.client.List(context.TODO(), storageClassList, &client.ListOptions{}); err != nil {
|
|
return err
|
|
}
|
|
|
|
// Check config for scratch space class
|
|
if config.Spec.ScratchSpaceStorageClass != nil {
|
|
for _, storageClass := range storageClassList.Items {
|
|
if storageClass.Name == *config.Spec.ScratchSpaceStorageClass {
|
|
log.Info("Setting scratch space to override", "storageClass.Name", storageClass.Name)
|
|
config.Status.ScratchSpaceStorageClass = storageClass.Name
|
|
return nil
|
|
}
|
|
}
|
|
}
|
|
// Check for default storage class.
|
|
for _, storageClass := range storageClassList.Items {
|
|
if defaultClassValue, ok := storageClass.Annotations[AnnDefaultStorageClass]; ok {
|
|
if defaultClassValue == "true" {
|
|
log.Info("Setting scratch space to default", "storageClass.Name", storageClass.Name)
|
|
config.Status.ScratchSpaceStorageClass = storageClass.Name
|
|
return nil
|
|
}
|
|
}
|
|
}
|
|
log.Info("No default storage class found, setting scratch space to blank")
|
|
// No storage class found, blank it out.
|
|
config.Status.ScratchSpaceStorageClass = ""
|
|
return nil
|
|
}
|
|
|
|
func (r *CDIConfigReconciler) reconcileDefaultPodResourceRequirements(config *cdiv1.CDIConfig) error {
|
|
cpuLimit, _ := resource.ParseQuantity(defaultCPULimit)
|
|
memLimit, _ := resource.ParseQuantity(defaultMemLimit)
|
|
cpuRequest, _ := resource.ParseQuantity(defaultCPURequest)
|
|
memRequest, _ := resource.ParseQuantity(defaultMemRequest)
|
|
config.Status.DefaultPodResourceRequirements = &v1.ResourceRequirements{
|
|
Limits: map[v1.ResourceName]resource.Quantity{
|
|
v1.ResourceCPU: cpuLimit,
|
|
v1.ResourceMemory: memLimit,
|
|
},
|
|
Requests: map[v1.ResourceName]resource.Quantity{
|
|
v1.ResourceCPU: cpuRequest,
|
|
v1.ResourceMemory: memRequest,
|
|
},
|
|
}
|
|
|
|
if config.Spec.PodResourceRequirements != nil {
|
|
if config.Spec.PodResourceRequirements.Limits != nil {
|
|
if cpu, exist := config.Spec.PodResourceRequirements.Limits[v1.ResourceCPU]; exist {
|
|
config.Status.DefaultPodResourceRequirements.Limits[v1.ResourceCPU] = cpu
|
|
}
|
|
|
|
if memory, exist := config.Spec.PodResourceRequirements.Limits[v1.ResourceMemory]; exist {
|
|
config.Status.DefaultPodResourceRequirements.Limits[v1.ResourceMemory] = memory
|
|
}
|
|
}
|
|
|
|
if config.Spec.PodResourceRequirements.Requests != nil {
|
|
if cpu, exist := config.Spec.PodResourceRequirements.Requests[v1.ResourceCPU]; exist {
|
|
config.Status.DefaultPodResourceRequirements.Requests[v1.ResourceCPU] = cpu
|
|
}
|
|
|
|
if memory, exist := config.Spec.PodResourceRequirements.Requests[v1.ResourceMemory]; exist {
|
|
config.Status.DefaultPodResourceRequirements.Requests[v1.ResourceMemory] = memory
|
|
}
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (r *CDIConfigReconciler) reconcileFilesystemOverhead(config *cdiv1.CDIConfig) error {
|
|
var globalOverhead cdiv1.Percent = common.DefaultGlobalOverhead
|
|
var perStorageConfig = make(map[string]cdiv1.Percent)
|
|
|
|
log := r.log.WithName("CDIconfig").WithName("FilesystemOverhead")
|
|
|
|
// Avoid nil maps and segfaults for the initial case, where filesystemOverhead
|
|
// is nil for both the spec and the status.
|
|
if config.Status.FilesystemOverhead == nil {
|
|
log.Info("No filesystem overhead found in status, initializing to defaults")
|
|
config.Status.FilesystemOverhead = &cdiv1.FilesystemOverhead{
|
|
Global: globalOverhead,
|
|
StorageClass: make(map[string]cdiv1.Percent),
|
|
}
|
|
}
|
|
|
|
if config.Spec.FilesystemOverhead != nil {
|
|
if valid, _ := validOverhead(config.Spec.FilesystemOverhead.Global); valid {
|
|
globalOverhead = config.Spec.FilesystemOverhead.Global
|
|
}
|
|
if config.Spec.FilesystemOverhead.StorageClass != nil {
|
|
perStorageConfig = config.Spec.FilesystemOverhead.StorageClass
|
|
}
|
|
}
|
|
|
|
// Set status global overhead
|
|
config.Status.FilesystemOverhead.Global = globalOverhead
|
|
|
|
// Set status per-storageClass overhead
|
|
storageClassList := &storagev1.StorageClassList{}
|
|
if err := r.client.List(context.TODO(), storageClassList, &client.ListOptions{}); err != nil {
|
|
return err
|
|
}
|
|
config.Status.FilesystemOverhead.StorageClass = make(map[string]cdiv1.Percent)
|
|
for _, storageClass := range storageClassList.Items {
|
|
storageClassName := storageClass.GetName()
|
|
storageClassNameOverhead, found := perStorageConfig[storageClassName]
|
|
|
|
if found {
|
|
valid, err := validOverhead(storageClassNameOverhead)
|
|
if !valid {
|
|
return err
|
|
}
|
|
config.Status.FilesystemOverhead.StorageClass[storageClassName] = storageClassNameOverhead
|
|
} else {
|
|
config.Status.FilesystemOverhead.StorageClass[storageClassName] = globalOverhead
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func validOverhead(overhead cdiv1.Percent) (bool, error) {
|
|
return regexp.MatchString(`^(0(?:\.\d{1,3})?|1)$`, string(overhead))
|
|
}
|
|
|
|
// createCDIConfig creates a new instance of the CDIConfig object if it doesn't exist already, and returns the existing one if found.
|
|
// It also sets the operator to be the owner of the CDIConfig object.
|
|
func (r *CDIConfigReconciler) createCDIConfig() (*cdiv1.CDIConfig, error) {
|
|
config := &cdiv1.CDIConfig{}
|
|
if err := r.uncachedClient.Get(context.TODO(), types.NamespacedName{Name: r.configName}, config); err != nil {
|
|
if errors.IsNotFound(err) {
|
|
config = MakeEmptyCDIConfigSpec(r.configName)
|
|
if err := operator.SetOwnerRuntime(r.uncachedClient, config); err != nil {
|
|
return nil, err
|
|
}
|
|
util.SetRecommendedLabels(config, r.installerLabels, "cdi-controller")
|
|
if err := r.client.Create(context.TODO(), config); err != nil {
|
|
if errors.IsAlreadyExists(err) {
|
|
config := &cdiv1.CDIConfig{}
|
|
if err := r.uncachedClient.Get(context.TODO(), types.NamespacedName{Name: r.configName}, config); err == nil {
|
|
return config, nil
|
|
}
|
|
return nil, err
|
|
}
|
|
return nil, err
|
|
}
|
|
} else {
|
|
return nil, err
|
|
}
|
|
}
|
|
return config, nil
|
|
}
|
|
|
|
func (r *CDIConfigReconciler) reconcileImportProxy(config *cdiv1.CDIConfig) error {
|
|
config.Status.ImportProxy = config.Spec.ImportProxy
|
|
|
|
// Avoid nil pointers and segfaults for the initial case, where ImportProxy is nil for both the spec and the status.
|
|
if config.Status.ImportProxy == nil {
|
|
config.Status.ImportProxy = &cdiv1.ImportProxy{
|
|
HTTPProxy: new(string),
|
|
HTTPSProxy: new(string),
|
|
NoProxy: new(string),
|
|
TrustedCAProxy: new(string),
|
|
}
|
|
|
|
// Try Openshift cluster wide proxy only if the CDIConfig default config is empty
|
|
clusterWideProxy, err := GetClusterWideProxy(r.client)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
config.Status.ImportProxy.HTTPProxy = &clusterWideProxy.Status.HTTPProxy
|
|
config.Status.ImportProxy.HTTPSProxy = &clusterWideProxy.Status.HTTPSProxy
|
|
config.Status.ImportProxy.NoProxy = &clusterWideProxy.Status.NoProxy
|
|
if clusterWideProxy.Spec.TrustedCA.Name != "" {
|
|
config.Status.ImportProxy.TrustedCAProxy = &clusterWideProxy.Spec.TrustedCA.Name
|
|
err = r.reconcileImportProxyCAConfigMap(config, clusterWideProxy)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// Create/Update a configmap with the CA certificates in the controllor context with the cluster-wide proxy CA certificates to be used by the importer pod
|
|
func (r *CDIConfigReconciler) reconcileImportProxyCAConfigMap(config *cdiv1.CDIConfig, proxy *ocpconfigv1.Proxy) error {
|
|
client := r.uncachedClient
|
|
cmName := proxy.Spec.TrustedCA.Name
|
|
if cmName == "" {
|
|
// Using the default cluster-wide proxy CA certificates configmap name
|
|
cmName = ClusterWideProxyConfigMapName
|
|
}
|
|
clusterWideProxyConfigMap := &v1.ConfigMap{}
|
|
if err := client.Get(context.TODO(), types.NamespacedName{Name: cmName, Namespace: ClusterWideProxyConfigMapNameSpace}, clusterWideProxyConfigMap); err == nil {
|
|
// Copy the cluster-wide proxy CA certificates to the importer pod proxy CA certificates configMap
|
|
if certBytes, ok := clusterWideProxyConfigMap.Data[ClusterWideProxyConfigMapKey]; ok {
|
|
configMap := &v1.ConfigMap{}
|
|
if err := client.Get(context.TODO(), types.NamespacedName{Name: common.ImportProxyConfigMapName, Namespace: r.cdiNamespace}, configMap); errors.IsNotFound(err) {
|
|
proxyConfigMap := r.createProxyConfigMap(certBytes)
|
|
util.SetRecommendedLabels(proxyConfigMap, r.installerLabels, "cdi-controller")
|
|
if err := client.Create(context.TODO(), proxyConfigMap); err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
if configMap != nil {
|
|
configMap.Data[common.ImportProxyConfigMapKey] = certBytes
|
|
util.SetRecommendedLabels(configMap, r.installerLabels, "cdi-controller")
|
|
if err := client.Update(context.TODO(), configMap); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (r *CDIConfigReconciler) createProxyConfigMap(certBytes string) *v1.ConfigMap {
|
|
return &v1.ConfigMap{
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
Name: common.ImportProxyConfigMapName,
|
|
Namespace: r.cdiNamespace},
|
|
Data: map[string]string{common.ImportProxyConfigMapKey: certBytes},
|
|
}
|
|
}
|
|
|
|
// Init initializes a CDIConfig object.
|
|
func (r *CDIConfigReconciler) Init() error {
|
|
_, err := r.createCDIConfig()
|
|
return err
|
|
}
|
|
|
|
// NewConfigController creates a new instance of the config controller.
|
|
func NewConfigController(mgr manager.Manager, log logr.Logger, uploadProxyServiceName, configName string, installerLabels map[string]string) (controller.Controller, error) {
|
|
uncachedClient, err := client.New(mgr.GetConfig(), client.Options{
|
|
Scheme: mgr.GetScheme(),
|
|
Mapper: mgr.GetRESTMapper(),
|
|
})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
reconciler := &CDIConfigReconciler{
|
|
client: mgr.GetClient(),
|
|
uncachedClient: uncachedClient,
|
|
scheme: mgr.GetScheme(),
|
|
log: log.WithName("config-controller"),
|
|
uploadProxyServiceName: uploadProxyServiceName,
|
|
configName: configName,
|
|
cdiNamespace: util.GetNamespace(),
|
|
installerLabels: installerLabels,
|
|
}
|
|
|
|
configController, err := controller.New("config-controller", mgr, controller.Options{
|
|
Reconciler: reconciler,
|
|
})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if err := addConfigControllerWatches(mgr, configController, reconciler.cdiNamespace, configName, uploadProxyServiceName, log); err != nil {
|
|
return nil, err
|
|
}
|
|
if err := reconciler.Init(); err != nil {
|
|
log.Error(err, "Unable to initalize CDIConfig")
|
|
}
|
|
log.Info("Initialized CDI Config object")
|
|
return configController, nil
|
|
}
|
|
|
|
// addConfigControllerWatches sets up the watches used by the config controller.
|
|
func addConfigControllerWatches(mgr manager.Manager, configController controller.Controller, cdiNamespace, configName, uploadProxyServiceName string, log logr.Logger) error {
|
|
// Add schemes.
|
|
if err := cdiv1.AddToScheme(mgr.GetScheme()); err != nil {
|
|
return err
|
|
}
|
|
if err := storagev1.AddToScheme(mgr.GetScheme()); err != nil {
|
|
return err
|
|
}
|
|
if err := networkingv1.AddToScheme(mgr.GetScheme()); err != nil {
|
|
return err
|
|
}
|
|
if err := routev1.Install(mgr.GetScheme()); err != nil {
|
|
return err
|
|
}
|
|
if err := ocpconfigv1.Install(mgr.GetScheme()); err != nil {
|
|
return err
|
|
}
|
|
|
|
// Setup watches
|
|
if err := watchCDIConfig(configController, configName); err != nil {
|
|
return err
|
|
}
|
|
if err := watchStorageClass(configController, configName); err != nil {
|
|
return err
|
|
}
|
|
if err := watchIngress(configController, cdiNamespace, configName, uploadProxyServiceName); err != nil {
|
|
return err
|
|
}
|
|
if err := watchRoutes(mgr, configController, cdiNamespace, configName, uploadProxyServiceName); err != nil {
|
|
return err
|
|
}
|
|
if err := watchClusterProxy(mgr, configController, configName); err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func watchCDIConfig(configController controller.Controller, configName string) error {
|
|
if err := configController.Watch(&source.Kind{Type: &cdiv1.CDIConfig{}}, &handler.EnqueueRequestForObject{}); err != nil {
|
|
return err
|
|
}
|
|
return configController.Watch(&source.Kind{Type: &cdiv1.CDI{}}, handler.EnqueueRequestsFromMapFunc(
|
|
func(client.Object) []reconcile.Request {
|
|
return []reconcile.Request{{
|
|
NamespacedName: types.NamespacedName{Name: configName},
|
|
}}
|
|
},
|
|
))
|
|
}
|
|
|
|
func watchStorageClass(configController controller.Controller, configName string) error {
|
|
return configController.Watch(&source.Kind{Type: &storagev1.StorageClass{}}, handler.EnqueueRequestsFromMapFunc(
|
|
func(client.Object) []reconcile.Request {
|
|
return []reconcile.Request{{
|
|
NamespacedName: types.NamespacedName{Name: configName},
|
|
}}
|
|
},
|
|
))
|
|
}
|
|
|
|
func watchIngress(configController controller.Controller, cdiNamespace, configName, uploadProxyServiceName string) error {
|
|
err := configController.Watch(&source.Kind{Type: &networkingv1.Ingress{}}, handler.EnqueueRequestsFromMapFunc(
|
|
func(client.Object) []reconcile.Request {
|
|
return []reconcile.Request{{
|
|
NamespacedName: types.NamespacedName{Name: configName},
|
|
}}
|
|
}),
|
|
predicate.Funcs{
|
|
CreateFunc: func(e event.CreateEvent) bool {
|
|
return "" != getURLFromIngress(e.Object.(*networkingv1.Ingress), uploadProxyServiceName) &&
|
|
e.Object.(*networkingv1.Ingress).GetNamespace() == cdiNamespace
|
|
},
|
|
UpdateFunc: func(e event.UpdateEvent) bool {
|
|
return "" != getURLFromIngress(e.ObjectNew.(*networkingv1.Ingress), uploadProxyServiceName) &&
|
|
e.ObjectNew.(*networkingv1.Ingress).GetNamespace() == cdiNamespace
|
|
},
|
|
DeleteFunc: func(e event.DeleteEvent) bool {
|
|
return "" != getURLFromIngress(e.Object.(*networkingv1.Ingress), uploadProxyServiceName) &&
|
|
e.Object.(*networkingv1.Ingress).GetNamespace() == cdiNamespace
|
|
},
|
|
})
|
|
return err
|
|
}
|
|
|
|
// we only watch the route obj if they exist, i.e., if it is an OpenShift cluster
|
|
func watchRoutes(mgr manager.Manager, configController controller.Controller, cdiNamespace, configName, uploadProxyServiceName string) error {
|
|
err := mgr.GetClient().List(context.TODO(), &routev1.RouteList{})
|
|
if !meta.IsNoMatchError(err) {
|
|
if err == nil || isErrCacheNotStarted(err) {
|
|
err := configController.Watch(&source.Kind{Type: &routev1.Route{}}, handler.EnqueueRequestsFromMapFunc(
|
|
func(client.Object) []reconcile.Request {
|
|
return []reconcile.Request{{
|
|
NamespacedName: types.NamespacedName{Name: configName},
|
|
}}
|
|
}),
|
|
predicate.Funcs{
|
|
CreateFunc: func(e event.CreateEvent) bool {
|
|
return "" != getURLFromRoute(e.Object.(*routev1.Route), uploadProxyServiceName) &&
|
|
e.Object.(*routev1.Route).GetNamespace() == cdiNamespace
|
|
},
|
|
UpdateFunc: func(e event.UpdateEvent) bool {
|
|
return "" != getURLFromRoute(e.ObjectNew.(*routev1.Route), uploadProxyServiceName) &&
|
|
e.ObjectNew.(*routev1.Route).GetNamespace() == cdiNamespace
|
|
},
|
|
DeleteFunc: func(e event.DeleteEvent) bool {
|
|
return "" != getURLFromRoute(e.Object.(*routev1.Route), uploadProxyServiceName) &&
|
|
e.Object.(*routev1.Route).GetNamespace() == cdiNamespace
|
|
},
|
|
})
|
|
return err
|
|
}
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// we only watch the cluster-wide proxy obj if they exist, i.e., if it is an OpenShift cluster
|
|
func watchClusterProxy(mgr manager.Manager, configController controller.Controller, configName string) error {
|
|
err := mgr.GetClient().List(context.TODO(), &ocpconfigv1.ProxyList{})
|
|
if !meta.IsNoMatchError(err) {
|
|
if err == nil || isErrCacheNotStarted(err) {
|
|
return configController.Watch(&source.Kind{Type: &ocpconfigv1.Proxy{}}, handler.EnqueueRequestsFromMapFunc(
|
|
func(client.Object) []reconcile.Request {
|
|
return []reconcile.Request{{
|
|
NamespacedName: types.NamespacedName{Name: configName},
|
|
}}
|
|
},
|
|
))
|
|
}
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func getURLFromIngress(ing *networkingv1.Ingress, uploadProxyServiceName string) string {
|
|
if ing.Spec.DefaultBackend != nil && ing.Spec.DefaultBackend.Service != nil {
|
|
if ing.Spec.DefaultBackend.Service.Name != uploadProxyServiceName {
|
|
return ""
|
|
}
|
|
return ing.Spec.Rules[0].Host
|
|
}
|
|
for _, rule := range ing.Spec.Rules {
|
|
if rule.HTTP == nil {
|
|
continue
|
|
}
|
|
for _, path := range rule.HTTP.Paths {
|
|
if path.Backend.Service != nil && path.Backend.Service.Name == uploadProxyServiceName {
|
|
if rule.Host != "" {
|
|
return rule.Host
|
|
}
|
|
}
|
|
}
|
|
}
|
|
return ""
|
|
}
|
|
|
|
func getURLFromRoute(route *routev1.Route, uploadProxyServiceName string) string {
|
|
if route.Spec.To.Name == uploadProxyServiceName {
|
|
if len(route.Status.Ingress) > 0 {
|
|
return route.Status.Ingress[0].Host
|
|
}
|
|
}
|
|
return ""
|
|
|
|
}
|