[Feature] Agency Lock bugfix (#1335)

Author: ajanikow

CHANGELOG.md

@@ -6,6 +6,7 @@
 - (Feature) Metrics Counter
 - (Feature) Requests Bytes Counter
 - (Feature) Agency Poll System
+- (Bugfix) (CE) Agency Lock bugfix
 
 ## [1.2.29](https://github.com/arangodb/kube-arangodb/tree/1.2.29) (2023-06-08)
 - (Maintenance) Add govulncheck to pipeline, update golangci-linter

pkg/debug_package/cli/cli.go

@@ -24,7 +24,7 @@ import "github.com/spf13/cobra"
 
 func Register(cmd *cobra.Command) {
 	f := cmd.Flags()
-	f.StringVar(&input.Namespace, "namespace", "default", "Kubernetes namespace")
+	f.StringVarP(&input.Namespace, "namespace", "n", "default", "Kubernetes namespace")
 	f.BoolVar(&input.HideSensitiveData, "hide-sensitive-data", true, "Hide sensitive data")
 	f.BoolVar(&input.PodLogs, "pod-logs", true, "Collect pod logs")
 }

pkg/debug_package/generators/kubernetes/agency.go

@@ -51,7 +51,7 @@ func agencyDump(logger zerolog.Logger, files chan<- shared.File) error {
 		return errors.Newf("Client is not initialised")
 	}
 
-	deployments, err := ListDeployments(k)
+	deployments, err := listArangoDeployments(k.Arango())()
 	if err != nil {
 		return err
 	}
@@ -102,7 +102,7 @@ func discoverExecFunc() (ArangoOperatorExecFunc, error) {
 			return nil, errors.Newf("Client is not initialised")
 		}
 
-		pods, err := ListPods(k)
+		pods, err := listPods(k.Kubernetes())()
 		if err != nil {
 			return nil, err
 		}

pkg/debug_package/generators/kubernetes/deployments.go renamed to pkg/debug_package/generators/kubernetes/arango_deployments.go

@@ -30,6 +30,7 @@ import (
 	api "github.com/arangodb/kube-arangodb/pkg/apis/deployment/v1"
 	"github.com/arangodb/kube-arangodb/pkg/debug_package/cli"
 	"github.com/arangodb/kube-arangodb/pkg/debug_package/shared"
+	arangoClient "github.com/arangodb/kube-arangodb/pkg/generated/clientset/versioned"
 	"github.com/arangodb/kube-arangodb/pkg/util/errors"
 	"github.com/arangodb/kube-arangodb/pkg/util/kclient"
 )
@@ -38,20 +39,32 @@ func Deployments() shared.Factory {
 	return shared.NewFactory("deployments", true, deployments)
 }
 
+func listArangoDeployments(client arangoClient.Interface) func() ([]*api.ArangoDeployment, error) {
+	return func() ([]*api.ArangoDeployment, error) {
+		return ListObjects[*api.ArangoDeploymentList, *api.ArangoDeployment](context.Background(), client.DatabaseV1().ArangoDeployments(cli.GetInput().Namespace), func(result *api.ArangoDeploymentList) []*api.ArangoDeployment {
+			q := make([]*api.ArangoDeployment, len(result.Items))
+
+			for id, e := range result.Items {
+				q[id] = e.DeepCopy()
+			}
+
+			return q
+		})
+	}
+}
+
 func deployments(logger zerolog.Logger, files chan<- shared.File) error {
 	k, ok := kclient.GetDefaultFactory().Client()
 	if !ok {
 		return errors.Newf("Client is not initialised")
 	}
 
-	deployments, err := ListDeployments(k)
+	deploymentList, err := listArangoDeployments(k.Arango())()
 	if err != nil {
 		return err
 	}
 
-	deploymentList := deployments.AsList()
-
-	errDeployments := make([]error, len(deployments))
+	errDeployments := make([]error, len(deploymentList))
 
 	for id := range deploymentList {
 		errDeployments[id] = deployment(k, deploymentList[id], files)

pkg/debug_package/generators/kubernetes/events.go

@@ -21,8 +21,13 @@
 package kubernetes
 
 import (
+	"context"
+
 	"github.com/rs/zerolog"
+	core "k8s.io/api/core/v1"
+	"k8s.io/client-go/kubernetes"
 
+	"github.com/arangodb/kube-arangodb/pkg/debug_package/cli"
 	"github.com/arangodb/kube-arangodb/pkg/debug_package/shared"
 	"github.com/arangodb/kube-arangodb/pkg/util/errors"
 	"github.com/arangodb/kube-arangodb/pkg/util/kclient"
@@ -32,26 +37,27 @@ func Events() shared.Factory {
 	return shared.NewFactory("kubernetes-events", true, events)
 }
 
+func listEvents(client kubernetes.Interface) func() ([]*core.Event, error) {
+	return func() ([]*core.Event, error) {
+		return ListObjects[*core.EventList, *core.Event](context.Background(), client.CoreV1().Events(cli.GetInput().Namespace), func(result *core.EventList) []*core.Event {
+			q := make([]*core.Event, len(result.Items))
+
+			for id, e := range result.Items {
+				q[id] = e.DeepCopy()
+			}
+
+			return q
+		})
+	}
+}
+
 func events(logger zerolog.Logger, files chan<- shared.File) error {
 	k, ok := kclient.GetDefaultFactory().Client()
 	if !ok {
 		return errors.Newf("Client is not initialised")
 	}
 
-	events, err := ListEvents(k)
-	if err != nil {
-		return err
-	}
-
-	files <- shared.NewYAMLFile("kubernetes/events.yaml", func() ([]interface{}, error) {
-		q := make([]interface{}, 0, len(events))
-
-		for _, e := range events {
-			q = append(q, e)
-		}
-
-		return q, nil
-	})
+	files <- shared.NewYAMLFile("kubernetes/events.yaml", listEvents(k.Kubernetes()))
 
 	return nil
 }

pkg/debug_package/generators/kubernetes/lister.go

@@ -0,0 +1,71 @@
+//
+// DISCLAIMER
+//
+// Copyright 2023 ArangoDB GmbH, Cologne, Germany
+//
+// 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.
+//
+// Copyright holder is ArangoDB GmbH, Cologne, Germany
+//
+
+package kubernetes
+
+import (
+	"context"
+
+	meta "k8s.io/apimachinery/pkg/apis/meta/v1"
+	"k8s.io/apimachinery/pkg/types"
+
+	"github.com/arangodb/kube-arangodb/pkg/util/k8sutil"
+)
+
+type ObjectList[T meta.Object] map[types.UID]T
+
+func (d ObjectList[T]) AsList() []T {
+	list := make([]T, 0, len(d))
+	for _, p := range d {
+		list = append(list, p)
+	}
+
+	return list
+}
+
+func MapObjects[L k8sutil.ListContinue, T meta.Object](ctx context.Context, k k8sutil.ListAPI[L], extract func(result L) []T) (ObjectList[T], error) {
+	objects := ObjectList[T]{}
+
+	if err := k8sutil.APIList[L](ctx, k, meta.ListOptions{}, func(result L, err error) error {
+		if err != nil {
+			return err
+		}
+		for _, obj := range extract(result) {
+			obj.SetManagedFields(nil)
+
+			objects[obj.GetUID()] = obj
+		}
+
+		return nil
+	}); err != nil {
+		return nil, err
+	}
+
+	return objects, nil
+}
+
+func ListObjects[L k8sutil.ListContinue, T meta.Object](ctx context.Context, k k8sutil.ListAPI[L], extract func(result L) []T) ([]T, error) {
+	objects, err := MapObjects[L, T](ctx, k, extract)
+	if err != nil {
+		return nil, err
+	}
+
+	return objects.AsList(), nil
+}