plugin-barman-cloud/test/e2e/internal/client/client.go
renovate[bot] 58f4fed40b
Some checks failed
Deploy Docusaurus to GitHub Pages / build (push) Failing after 4s
Deploy Docusaurus to GitHub Pages / deploy (push) Has been skipped
release-please / release-please (push) Failing after 3s
fix(deps): update module sigs.k8s.io/controller-runtime to v0.24.0 (#880)
Signed-off-by: renovate[bot] <29139614+renovate[bot]@users.noreply.github.com>
Signed-off-by: Leonardo Cecchi <leonardo.cecchi@gmail.com>
Co-authored-by: renovate[bot] <29139614+renovate[bot]@users.noreply.github.com>
Co-authored-by: Leonardo Cecchi <leonardo.cecchi@gmail.com>
2026-05-06 10:15:29 +02:00

102 lines
3.2 KiB
Go

/*
Copyright © contributors to CloudNativePG, established as
CloudNativePG a Series of LF Projects, LLC.
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.
SPDX-License-Identifier: Apache-2.0
*/
package client
import (
"fmt"
certmanagerv1 "github.com/cert-manager/cert-manager/pkg/apis/certmanager/v1"
cloudnativepgv1 "github.com/cloudnative-pg/api/pkg/api/v1"
admissionregistrationv1 "k8s.io/api/admissionregistration/v1"
appsv1 "k8s.io/api/apps/v1"
corev1 "k8s.io/api/core/v1"
rbacv1 "k8s.io/api/rbac/v1"
apiextensionsv1 "k8s.io/apiextensions-apiserver/pkg/apis/apiextensions/v1"
"k8s.io/client-go/kubernetes"
"k8s.io/client-go/rest"
"sigs.k8s.io/controller-runtime/pkg/client"
"sigs.k8s.io/controller-runtime/pkg/client/config"
pluginBarmanCloudV1 "github.com/cloudnative-pg/plugin-barman-cloud/api/v1"
)
// NewClient creates a new controller-runtime Kubernetes client.
//
//nolint:ireturn
func NewClient() (client.Client, *rest.Config, error) {
cfg, err := config.GetConfig()
if err != nil {
return nil, nil, fmt.Errorf("failed to get Kubernetes config: %w", err)
}
cl, err := client.New(cfg, client.Options{})
if err != nil {
return nil, nil, fmt.Errorf("failed to create Kubernetes client: %w", err)
}
if err := addScheme(cl); err != nil {
return nil, nil, fmt.Errorf("failed to add scheme: %w", err)
}
return cl, cfg, nil
}
// NewClientSet creates a new k8s client-go clientset.
func NewClientSet() (*kubernetes.Clientset, *rest.Config, error) {
cfg, err := config.GetConfig()
if err != nil {
return nil, nil, fmt.Errorf("failed to get Kubernetes config: %w", err)
}
clientSet, err := kubernetes.NewForConfig(cfg)
if err != nil {
return nil, nil, fmt.Errorf("failed to create Kubernetes client: %w", err)
}
return clientSet, cfg, nil
}
func addScheme(cl client.Client) error {
scheme := cl.Scheme()
if err := corev1.AddToScheme(scheme); err != nil {
return fmt.Errorf("failed to add core/v1 to scheme: %w", err)
}
if err := apiextensionsv1.AddToScheme(scheme); err != nil {
return fmt.Errorf("failed to add apiextensions/v1 to scheme: %w", err)
}
if err := admissionregistrationv1.AddToScheme(scheme); err != nil {
return fmt.Errorf("failed to add admissionregistration/v1 to scheme: %w", err)
}
if err := rbacv1.AddToScheme(scheme); err != nil {
return fmt.Errorf("failed to add rbac/v1 to scheme: %w", err)
}
if err := appsv1.AddToScheme(scheme); err != nil {
return fmt.Errorf("failed to add apps/v1 to scheme: %w", err)
}
if err := certmanagerv1.AddToScheme(scheme); err != nil {
return fmt.Errorf("failed to add cert-manager/v1 to scheme: %w", err)
}
if err := cloudnativepgv1.AddToScheme(scheme); err != nil {
return fmt.Errorf("failed to add cloudnativepg/v1 to scheme: %w", err)
}
pluginBarmanCloudV1.AddKnownTypes(scheme)
return nil
}