Store drain status in SriovNetworkNodeState

api/v1/sriovnetworknodestate_types.go

@@ -85,6 +85,7 @@ type VirtualFunction struct {
 // SriovNetworkNodeStateStatus defines the observed state of SriovNetworkNodeState
 type SriovNetworkNodeStateStatus struct {
 	Interfaces    InterfaceExts `json:"interfaces,omitempty"`
+	DrainStatus   string        `json:"drainStatus,omitempty"`
 	SyncStatus    string        `json:"syncStatus,omitempty"`
 	LastSyncError string        `json:"lastSyncError,omitempty"`
 }

config/crd/bases/sriovnetwork.openshift.io_sriovnetworknodestates.yaml

@@ -88,6 +88,8 @@ spec:
             description: SriovNetworkNodeStateStatus defines the observed state of
               SriovNetworkNodeState
             properties:
+              drainStatus:
+                type: string
               interfaces:
                 items:
                   properties:

controllers/migration_controller.go

@@ -0,0 +1,69 @@
+package controllers
+
+import (
+	"context"
+	"fmt"
+	v1 "github.com/k8snetworkplumbingwg/sriov-network-operator/api/v1"
+	constants "github.com/k8snetworkplumbingwg/sriov-network-operator/pkg/consts"
+	corev1 "k8s.io/api/core/v1"
+	"k8s.io/apimachinery/pkg/runtime"
+	"k8s.io/apimachinery/pkg/types"
+	ctrl "sigs.k8s.io/controller-runtime"
+	"sigs.k8s.io/controller-runtime/pkg/builder"
+	"sigs.k8s.io/controller-runtime/pkg/client"
+	"sigs.k8s.io/controller-runtime/pkg/log"
+	"sigs.k8s.io/controller-runtime/pkg/reconcile"
+)
+
+type MigrationReconciler struct {
+	client.Client
+	Scheme *runtime.Scheme
+}
+
+//+kubebuilder:rbac:groups="",resources=nodes,verbs=get;list;watch;update;patch
+
+// Reconcile is part of the main kubernetes reconciliation loop which aims to
+// move the current state of the cluster closer to the desired state.
+// For more details, check Reconcile and its Result here:
+// - https://pkg.go.dev/sigs.k8s.io/[email protected]/pkg/reconcile
+func (mr *MigrationReconciler) Reconcile(ctx context.Context, req ctrl.Request) (ctrl.Result, error) {
+	req.Namespace = namespace
+	reqLogger := log.FromContext(ctx).WithValues("drain", req.NamespacedName)
+	reqLogger.Info("Reconciling annotation migration")
+
+	node := &corev1.Node{}
+	err := mr.Get(ctx, types.NamespacedName{
+		Name: req.Name}, node)
+	if err != nil {
+		reqLogger.Error(err, "Error occurred on GET SriovOperatorConfig request from API server.")
+		return reconcile.Result{}, err
+	}
+
+	if anno, ok := node.Annotations[constants.NodeDrainAnnotation]; ok {
+		nodeState := &v1.SriovNetworkNodeState{}
+		err := mr.Get(ctx, types.NamespacedName{
+			Name: node.Name, Namespace: namespace}, nodeState)
+		if err != nil {
+			reqLogger.Error(err, "Error occurred on GET SriovNetworkNodeState request from API server.")
+			return reconcile.Result{}, err
+		}
+		patch := []byte(fmt.Sprintf(`{"status":{"drainStatus":"%s"}}`, anno))
+		err = mr.Client.Patch(context.TODO(), nodeState, client.RawPatch(types.StrategicMergePatchType, patch))
+		if err != nil {
+			reqLogger.Error(err, "Error occurred on SriovNetworkNodeState update.")
+			return reconcile.Result{}, err
+		}
+
+	}
+
+	return reconcile.Result{}, nil
+}
+
+// SetupWithManager sets up the controller with the Manager.
+func (mr *MigrationReconciler) SetupWithManager(mgr ctrl.Manager) error {
+	nodePredicates := builder.WithPredicates(DrainAnnotationPredicate{})
+
+	return ctrl.NewControllerManagedBy(mgr).
+		For(&corev1.Node{}, nodePredicates).
+		Complete(mr)
+}

deployment/sriov-network-operator/crds/sriovnetwork.openshift.io_sriovnetworknodestates.yaml

@@ -88,6 +88,8 @@ spec:
             description: SriovNetworkNodeStateStatus defines the observed state of
               SriovNetworkNodeState
             properties:
+              drainStatus:
+                type: string
               interfaces:
                 items:
                   properties:

main.go

@@ -198,6 +198,13 @@ func main() {
 		setupLog.Error(err, "unable to create controller", "controller", "DrainReconciler")
 		os.Exit(1)
 	}
+	if err = (&controllers.MigrationReconciler{
+		Client: mgr.GetClient(),
+		Scheme: mgr.GetScheme(),
+	}).SetupWithManager(mgr); err != nil {
+		setupLog.Error(err, "unable to create controller", "controller", "MigrationReconciler")
+		os.Exit(1)
+	}
 	// +kubebuilder:scaffold:builder
 
 	// Create a default SriovNetworkNodePolicy

pkg/utils/cluster.go

@@ -119,7 +119,7 @@ func openshiftControlPlaneTopologyStatus(c client.Client) (configv1.TopologyMode
 }
 
 func NodeHasAnnotation(node corev1.Node, annoKey string, value string) bool {
-	// Check if node already contains annotation
+	// Check if node already contains annotation with specified value
 	if anno, ok := node.Annotations[annoKey]; ok && (anno == value) {
 		return true
 	}