[Feature] [Integration] SchedulerV2 (#1744)

Author: ajanikow

CHANGELOG.md

@@ -6,6 +6,7 @@
 - (Feature) Helm Client Extension
 - (Feature) (Integration) SchedulerV2 Definition
 - (Maintenance) Proto Lint
+- (Feature) (Integration) SchedulerV2
 
 ## [1.2.43](https://github.com/arangodb/kube-arangodb/tree/1.2.43) (2024-10-14)
 - (Feature) ArangoRoute CRD

integrations/scheduler/v2/configuration.go

@@ -0,0 +1,59 @@
+//
+// DISCLAIMER
+//
+// Copyright 2024 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 v2
+
+import "github.com/pkg/errors"
+
+type Mod func(c Configuration) Configuration
+
+func NewConfiguration() Configuration {
+	return Configuration{
+		Namespace: "default",
+	}
+}
+
+type Configuration struct {
+	Namespace string
+
+	Deployment string
+}
+
+func (c Configuration) Validate() error {
+	if c.Deployment == "" {
+		return errors.Errorf("Invalid empty name of deployment")
+	}
+
+	if c.Namespace == "" {
+		return errors.Errorf("Invalid empty name of namespace")
+	}
+
+	return nil
+}
+
+func (c Configuration) With(mods ...Mod) Configuration {
+	n := c
+
+	for _, mod := range mods {
+		n = mod(n)
+	}
+
+	return n
+}

integrations/scheduler/v2/consts.go

@@ -0,0 +1,25 @@
+//
+// DISCLAIMER
+//
+// Copyright 2024 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 v2
+
+const (
+	LabelArangoDBDeploymentName = "deployment.arangodb.com/name"
+)

integrations/scheduler/v2/helm.go

@@ -0,0 +1,222 @@
+//
+// DISCLAIMER
+//
+// Copyright 2024 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 v2
+
+import (
+	"context"
+
+	"google.golang.org/grpc/codes"
+	"google.golang.org/grpc/status"
+	"helm.sh/helm/v3/pkg/action"
+	"k8s.io/apimachinery/pkg/labels"
+	"k8s.io/apimachinery/pkg/selection"
+
+	pbSchedulerV2 "github.com/arangodb/kube-arangodb/integrations/scheduler/v2/definition"
+	pbSharedV1 "github.com/arangodb/kube-arangodb/integrations/shared/v1/definition"
+	"github.com/arangodb/kube-arangodb/pkg/util"
+)
+
+func (i *implementation) Alive(ctx context.Context, in *pbSharedV1.Empty) (*pbSharedV1.Empty, error) {
+	if err := i.client.Alive(ctx); err != nil {
+		logger.Err(err).Warn("Helm is not alive")
+		return nil, status.Errorf(codes.Unavailable, "Service is not alive")
+	}
+
+	return &pbSharedV1.Empty{}, nil
+}
+
+func (i *implementation) List(ctx context.Context, in *pbSchedulerV2.SchedulerV2ListRequest) (*pbSchedulerV2.SchedulerV2ListResponse, error) {
+	var mods []util.Mod[action.List]
+
+	mods = append(mods, in.GetOptions().Options()...)
+	mods = append(mods, func(action *action.List) {
+		var s = labels.NewSelector()
+		if action.Selector != "" {
+			if n, err := labels.Parse(action.Selector); err == nil {
+				s = n
+			}
+		}
+
+		if r, err := labels.NewRequirement(LabelArangoDBDeploymentName, selection.DoubleEquals, []string{i.cfg.Deployment}); err != nil {
+			logger.Err(err).Warn("Unable to render selector")
+		} else if r != nil {
+			s = s.Add(*r)
+		}
+
+		action.Selector = s.String()
+	})
+
+	resp, err := i.client.List(ctx, mods...)
+	if err != nil {
+		logger.Err(err).Warn("Unable to run action: List")
+		return nil, status.Errorf(codes.Internal, "Unable to run action: List: %s", err.Error())
+	}
+
+	releases := make(map[string]*pbSchedulerV2.SchedulerV2Release, len(resp))
+
+	for _, r := range resp {
+		releases[r.Name] = newChartReleaseFromHelmRelease(&r)
+	}
+
+	return &pbSchedulerV2.SchedulerV2ListResponse{
+		Releases: releases,
+	}, nil
+}
+
+func (i *implementation) Status(ctx context.Context, in *pbSchedulerV2.SchedulerV2StatusRequest) (*pbSchedulerV2.SchedulerV2StatusResponse, error) {
+	if in.GetName() == "" {
+		return nil, status.Errorf(codes.InvalidArgument, "Name cannot be empty")
+	}
+
+	resp, err := i.client.Status(ctx, in.GetName())
+	if err != nil {
+		logger.Err(err).Warn("Unable to run action: Status")
+		return nil, status.Errorf(codes.Internal, "Unable to run action: Status: %s", err.Error())
+	}
+
+	return &pbSchedulerV2.SchedulerV2StatusResponse{
+		Release: newChartReleaseFromHelmRelease(resp),
+	}, nil
+}
+func (i *implementation) StatusObjects(ctx context.Context, in *pbSchedulerV2.SchedulerV2StatusObjectsRequest) (*pbSchedulerV2.SchedulerV2StatusObjectsResponse, error) {
+	if in.GetName() == "" {
+		return nil, status.Errorf(codes.InvalidArgument, "Name cannot be empty")
+	}
+
+	resp, objs, err := i.client.StatusObjects(ctx, in.GetName())
+	if err != nil {
+		logger.Err(err).Warn("Unable to run action: Status")
+		return nil, status.Errorf(codes.Internal, "Unable to run action: Status: %s", err.Error())
+	}
+
+	return &pbSchedulerV2.SchedulerV2StatusObjectsResponse{
+		Release: newChartReleaseFromHelmRelease(resp),
+		Objects: newReleaseInfoResourceObjectsFromResourceObjects(objs),
+	}, nil
+}
+
+func (i *implementation) Install(ctx context.Context, in *pbSchedulerV2.SchedulerV2InstallRequest) (*pbSchedulerV2.SchedulerV2InstallResponse, error) {
+	if in.GetName() == "" {
+		return nil, status.Errorf(codes.InvalidArgument, "Name cannot be empty")
+	}
+
+	var mods []util.Mod[action.Install]
+
+	mods = append(mods, in.GetOptions().Options()...)
+	mods = append(mods, func(action *action.Install) {
+		action.ReleaseName = in.GetName()
+		action.Namespace = i.cfg.Namespace
+
+		if action.Labels == nil {
+			action.Labels = map[string]string{}
+		}
+
+		action.Labels[LabelArangoDBDeploymentName] = i.cfg.Deployment
+	})
+
+	resp, err := i.client.Install(ctx, in.GetChart(), in.GetValues(), mods...)
+	if err != nil {
+		logger.Err(err).Warn("Unable to run action: Install")
+		return nil, status.Errorf(codes.Internal, "Unable to run action: Install: %s", err.Error())
+	}
+
+	return &pbSchedulerV2.SchedulerV2InstallResponse{
+		Release: newChartReleaseFromHelmRelease(resp),
+	}, nil
+}
+
+func (i *implementation) Upgrade(ctx context.Context, in *pbSchedulerV2.SchedulerV2UpgradeRequest) (*pbSchedulerV2.SchedulerV2UpgradeResponse, error) {
+	if in.GetName() == "" {
+		return nil, status.Errorf(codes.InvalidArgument, "Name cannot be empty")
+	}
+
+	var mods []util.Mod[action.Upgrade]
+
+	mods = append(mods, in.GetOptions().Options()...)
+	mods = append(mods, func(action *action.Upgrade) {
+		action.Namespace = i.cfg.Namespace
+
+		if action.Labels == nil {
+			action.Labels = map[string]string{}
+		}
+
+		action.Labels[LabelArangoDBDeploymentName] = i.cfg.Deployment
+	})
+
+	resp, err := i.client.Upgrade(ctx, in.GetName(), in.GetChart(), in.GetValues(), mods...)
+	if err != nil {
+		logger.Err(err).Warn("Unable to run action: Upgrade")
+		return nil, status.Errorf(codes.Internal, "Unable to run action: Upgrade: %s", err.Error())
+	}
+
+	var r pbSchedulerV2.SchedulerV2UpgradeResponse
+
+	if q := resp.Before; q != nil {
+		r.Before = newChartReleaseFromHelmRelease(q)
+	}
+
+	if q := resp.After; q != nil {
+		r.After = newChartReleaseFromHelmRelease(q)
+	}
+
+	return &r, nil
+}
+
+func (i *implementation) Uninstall(ctx context.Context, in *pbSchedulerV2.SchedulerV2UninstallRequest) (*pbSchedulerV2.SchedulerV2UninstallResponse, error) {
+	if in.GetName() == "" {
+		return nil, status.Errorf(codes.InvalidArgument, "Name cannot be empty")
+	}
+
+	var mods []util.Mod[action.Uninstall]
+
+	mods = append(mods, in.GetOptions().Options()...)
+
+	resp, err := i.client.Uninstall(ctx, in.GetName(), mods...)
+	if err != nil {
+		logger.Err(err).Warn("Unable to run action: Uninstall")
+		return nil, status.Errorf(codes.Internal, "Unable to run action: Uninstall: %s", err.Error())
+	}
+
+	return &pbSchedulerV2.SchedulerV2UninstallResponse{
+		Info:    resp.Info,
+		Release: newChartReleaseFromHelmRelease(&resp.Release),
+	}, nil
+}
+
+func (i *implementation) Test(ctx context.Context, in *pbSchedulerV2.SchedulerV2TestRequest) (*pbSchedulerV2.SchedulerV2TestResponse, error) {
+	if in.GetName() == "" {
+		return nil, status.Errorf(codes.InvalidArgument, "Name cannot be empty")
+	}
+
+	var mods []util.Mod[action.ReleaseTesting]
+
+	mods = append(mods, in.GetOptions().Options()...)
+
+	resp, err := i.client.Test(ctx, in.GetName(), mods...)
+	if err != nil {
+		logger.Err(err).Warn("Unable to run action: Test")
+		return nil, status.Errorf(codes.Internal, "Unable to run action: Test: %s", err.Error())
+	}
+
+	return &pbSchedulerV2.SchedulerV2TestResponse{
+		Release: newChartReleaseFromHelmRelease(resp),
+	}, nil
+}

integrations/scheduler/v2/implementation.go

@@ -0,0 +1,76 @@
+//
+// DISCLAIMER
+//
+// Copyright 2024 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 v2
+
+import (
+	"context"
+
+	"google.golang.org/grpc"
+
+	pbSchedulerV2 "github.com/arangodb/kube-arangodb/integrations/scheduler/v2/definition"
+	pbSharedV1 "github.com/arangodb/kube-arangodb/integrations/shared/v1/definition"
+	"github.com/arangodb/kube-arangodb/pkg/util/k8sutil/helm"
+	"github.com/arangodb/kube-arangodb/pkg/util/svc"
+)
+
+var _ pbSchedulerV2.SchedulerV2Server = &implementation{}
+var _ svc.Handler = &implementation{}
+
+func New(client helm.Client, cfg Configuration) (svc.Handler, error) {
+	return newInternal(client, cfg)
+}
+
+func newInternal(client helm.Client, c Configuration) (*implementation, error) {
+	if err := c.Validate(); err != nil {
+		return nil, err
+	}
+
+	return &implementation{
+		cfg:    c,
+		client: client,
+	}, nil
+}
+
+type implementation struct {
+	cfg Configuration
+
+	client helm.Client
+
+	pbSchedulerV2.UnimplementedSchedulerV2Server
+}
+
+func (i *implementation) Name() string {
+	return pbSchedulerV2.Name
+}
+
+func (i *implementation) Register(registrar *grpc.Server) {
+	pbSchedulerV2.RegisterSchedulerV2Server(registrar, i)
+}
+
+func (i *implementation) Health() svc.HealthState {
+	return svc.Healthy
+}
+
+func (i *implementation) InvalidateCache(ctx context.Context, in *pbSharedV1.Empty) (*pbSharedV1.Empty, error) {
+	i.client.Invalidate()
+
+	return &pbSharedV1.Empty{}, nil
+}