v1_controller.go

/*
Copyright 2022.

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 controllers

import (
	"context"
	"time"

	"github.com/f5devcentral/bigip-kubernetes-gateway/internal/k8s"
	"github.com/f5devcentral/bigip-kubernetes-gateway/internal/pkg"
	"github.com/f5devcentral/f5-bigip-rest-go/utils"
	ctrl "sigs.k8s.io/controller-runtime"
	"sigs.k8s.io/controller-runtime/pkg/client"

	v1 "k8s.io/api/core/v1"
)

type EndpointsReconciler struct {
	ObjectType client.Object
	Client     client.Client
	// LogLevel   string
}

type ServiceReconciler struct {
	ObjectType client.Object
	Client     client.Client
	// LogLevel   string
}

type NodeReconciler struct {
	ObjectType client.Object
	Client     client.Client
	// LogLevel   string
}

type NamespaceReconciler struct {
	ObjectType client.Object
	Client     client.Client
	// LogLevel   string
}

func (r *NamespaceReconciler) Reconcile(ctx context.Context, req ctrl.Request) (ctrl.Result, error) {
	if !pkg.ActiveSIGs.SyncedAtStart {
		<-time.After(100 * time.Millisecond)
		return ctrl.Result{Requeue: true}, nil
	}

	var obj v1.Namespace

	lctx := pkg.NewContext()
	slog := utils.LogFromContext(lctx)
	slog.Debugf("Namespace event: " + req.Name)

	// TODO: update resource mappings since namespace labels are changed.
	if err := r.Client.Get(ctx, req.NamespacedName, &obj); err != nil {
		if client.IgnoreNotFound(err) == nil {
			pkg.ActiveSIGs.UnsetNamespace(req.Name)
			return ctrl.Result{}, nil
		} else {
			return ctrl.Result{}, err
		}
	} else {
		ns := pkg.ActiveSIGs.GetNamespace(req.Name)
		if ns != nil && !utils.DeepEqual(ns.Labels, obj.Labels) {
			cls := pkg.ActiveSIGs.NSImpactedGatewayClasses(&obj)
			pkg.ActiveSIGs.SetNamespace(obj.DeepCopy())
			err := pkg.DeployForEvent(lctx, cls)
			if err != nil {
				return ctrl.Result{}, err
			}
		}

		pkg.ActiveSIGs.SetNamespace(obj.DeepCopy())
		return ctrl.Result{}, nil
	}
}

func (r *NamespaceReconciler) GetResObject() client.Object {
	return r.ObjectType
}

func (r *EndpointsReconciler) Reconcile(ctx context.Context, req ctrl.Request) (ctrl.Result, error) {
	lctx := pkg.NewContext()
	var obj v1.Endpoints
	// // too many logs.
	// slog.Debugf("endpoint event: " + req.NamespacedName.String())
	ns := req.Namespace
	if err := r.Client.Get(ctx, req.NamespacedName, &obj); err != nil {
		if client.IgnoreNotFound(err) == nil {
			pkg.ActiveSIGs.UnsetEndpoints(req.NamespacedName.String())
			return ctrl.Result{}, pkg.HandleBackends(lctx, ns)
		} else {
			return ctrl.Result{}, err
		}
	} else {
		pkg.ActiveSIGs.SetEndpoints(obj.DeepCopy())
		return ctrl.Result{}, pkg.HandleBackends(lctx, ns)
	}
}

func (r *EndpointsReconciler) GetResObject() client.Object {
	return r.ObjectType
}

func (r *ServiceReconciler) Reconcile(ctx context.Context, req ctrl.Request) (ctrl.Result, error) {
	var obj v1.Service
	lctx := pkg.NewContext()
	slog := utils.LogFromContext(lctx)
	slog.Debugf("Service event: " + req.NamespacedName.String())
	if err := r.Client.Get(ctx, req.NamespacedName, &obj); err != nil {
		if client.IgnoreNotFound(err) == nil {
			pkg.ActiveSIGs.UnsetService(req.NamespacedName.String())
			return ctrl.Result{}, pkg.HandleBackends(lctx, req.Namespace)
		} else {
			return ctrl.Result{}, err
		}
	} else {
		pkg.ActiveSIGs.SetService(obj.DeepCopy())
		return ctrl.Result{}, pkg.HandleBackends(lctx, req.Namespace)
	}
}

func (r *ServiceReconciler) GetResObject() client.Object {
	return r.ObjectType
}

func (r *NodeReconciler) Reconcile(ctx context.Context, req ctrl.Request) (ctrl.Result, error) {
	// slog := utils.NewLog().WithRequestID(uuid.New().String()).WithLevel(r.LogLevel)
	// lctx := context.WithValue(ctx, utils.CtxKey_Logger, slog)
	if !pkg.ActiveSIGs.SyncedAtStart {
		<-time.After(100 * time.Millisecond)
		return ctrl.Result{Requeue: true}, nil
	}

	var obj v1.Node
	if err := r.Client.Get(ctx, req.NamespacedName, &obj); err != nil {
		if client.IgnoreNotFound(err) == nil {
			k8s.NodeCache.Unset(req.Name)
		} else {
			return ctrl.Result{}, err
		}
	} else {
		k8s.NodeCache.Set(obj.DeepCopy())
	}
	return ctrl.Result{}, nil
}

func (r *NodeReconciler) GetResObject() client.Object {
	return r.ObjectType
}