Calculate total number of endpoints for service

In this commit the Reconcile function of the endpointslice controller
is implemented and calculates the total number of endpoints for
a service.

Signed-off-by: Elis Lulja <[email protected]>

Author: asimpleidea

controllers/endpointslice_controller.go

@@ -17,7 +17,7 @@
 package controllers
 
 import (
-	"context"
+	"fmt"
 	"sync"
 
 	"k8s.io/api/discovery/v1beta1"
@@ -52,10 +52,32 @@ type EndpointSliceReconciler struct {
 
 // Reconcile keeps track counts in the endpointslice length
 func (r *EndpointSliceReconciler) Reconcile(req ctrl.Request) (ctrl.Result, error) {
-	_ = context.Background()
-	_ = r.Log.WithName("EndpointSliceReconciler").WithValues("endpointslice", req.NamespacedName)
+	l := r.Log.WithName("EndpointSliceReconciler").WithValues("endpointslice", req.NamespacedName)
+	namespacedName := req.NamespacedName.String()
+	data := func(nname string) *epsData {
+		r.lock.Lock()
+		defer r.lock.Unlock()
+		defer delete(r.epsDataActions, nname)
+
+		return r.epsDataActions[nname]
+	}(namespacedName)
+	if data == nil {
+		l.Error(fmt.Errorf("no data exists"), "could not get endpointslice data for this endpointslice")
+		return ctrl.Result{}, nil
+	}
+
+	srvName := ktypes.NamespacedName{Namespace: req.Namespace, Name: data.srv}.String()
+	if _, exists := r.srvCounts[srvName]; !exists {
+		r.srvCounts[srvName] = map[string]int{}
+	}
+	r.srvCounts[srvName][req.Name] = data.count
+
+	totalCount := 0
+	for _, c := range r.srvCounts[srvName] {
+		totalCount += c
+	}
 
-	// TODO: implement me
+	// TODO: do something with the total count
 
 	return ctrl.Result{}, nil
 }

controllers/endpointslice_controller_test.go

@@ -26,6 +26,7 @@ import (
 	"k8s.io/api/discovery/v1beta1"
 	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
 	"k8s.io/apimachinery/pkg/runtime"
+	ktypes "k8s.io/apimachinery/pkg/types"
 	ctrl "sigs.k8s.io/controller-runtime"
 	"sigs.k8s.io/controller-runtime/pkg/client"
 	fakecli "sigs.k8s.io/controller-runtime/pkg/client/fake"
@@ -213,3 +214,66 @@ func TestEpSliceDeletePredicate(t *testing.T) {
 		}
 	}
 }
+
+func TestReconcile(t *testing.T) {
+	a := assert.New(t)
+	cases := []struct {
+		dataActions  map[string]*epsData
+		srvCounts    map[string]map[string]int
+		expRes       ctrl.Result
+		expErr       error
+		expSrvCounts map[string]map[string]int
+	}{
+		{},
+		{
+			dataActions: map[string]*epsData{
+				"ns-name/name": {
+					count: 3,
+					srv:   "srv",
+				},
+			},
+			srvCounts: map[string]map[string]int{
+				"ns-name/srv": {
+					"another-name": 5,
+				},
+			},
+			expSrvCounts: map[string]map[string]int{
+				"ns-name/srv": {
+					"another-name": 5,
+					"name":         3,
+				},
+			},
+		},
+		{
+			dataActions: map[string]*epsData{
+				"ns-name/name": {
+					count: 3,
+					srv:   "srv",
+				},
+			},
+			srvCounts: map[string]map[string]int{},
+			expSrvCounts: map[string]map[string]int{
+				"ns-name/srv": {
+					"name": 3,
+				},
+			},
+		},
+	}
+	failed := func(i int) {
+		a.FailNow(fmt.Sprintf("case %d failed", i))
+	}
+	for i, currCase := range cases {
+		eps := &EndpointSliceReconciler{
+			BaseReconciler: &BaseReconciler{
+				Log: ctrl.Log.WithName("controllers"),
+			},
+			epsDataActions: currCase.dataActions,
+			srvCounts:      currCase.srvCounts,
+		}
+
+		res, err := eps.Reconcile(ctrl.Request{NamespacedName: ktypes.NamespacedName{Namespace: "ns-name", Name: "name"}})
+		if !a.Equal(currCase.expRes, res) || !a.Equal(currCase.expErr, err) || !a.Equal(currCase.expSrvCounts, eps.srvCounts) {
+			failed(i)
+		}
+	}
+}