mirror of
https://github.com/ceph/ceph-csi.git
synced 2024-12-11 15:40:24 +00:00
b1a4590967
There is no release for sigs.k8s.io/controller-runtime that supports Kubernetes v1.27. The main branch has all the required modifications, so we can use that for the time being. Signed-off-by: Niels de Vos <ndevos@ibm.com>
263 lines
9.2 KiB
Go
263 lines
9.2 KiB
Go
/*
|
|
Copyright 2020 The Kubernetes Authors.
|
|
|
|
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.
|
|
*/
|
|
|
|
package client
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
"k8s.io/apimachinery/pkg/api/meta"
|
|
"k8s.io/apimachinery/pkg/runtime"
|
|
"k8s.io/apimachinery/pkg/runtime/schema"
|
|
)
|
|
|
|
// NewNamespacedClient wraps an existing client enforcing the namespace value.
|
|
// All functions using this client will have the same namespace declared here.
|
|
func NewNamespacedClient(c Client, ns string) Client {
|
|
return &namespacedClient{
|
|
client: c,
|
|
namespace: ns,
|
|
}
|
|
}
|
|
|
|
var _ Client = &namespacedClient{}
|
|
|
|
// namespacedClient is a Client that wraps another Client in order to enforce the specified namespace value.
|
|
type namespacedClient struct {
|
|
namespace string
|
|
client Client
|
|
}
|
|
|
|
// Scheme returns the scheme this client is using.
|
|
func (n *namespacedClient) Scheme() *runtime.Scheme {
|
|
return n.client.Scheme()
|
|
}
|
|
|
|
// RESTMapper returns the scheme this client is using.
|
|
func (n *namespacedClient) RESTMapper() meta.RESTMapper {
|
|
return n.client.RESTMapper()
|
|
}
|
|
|
|
// GroupVersionKindFor returns the GroupVersionKind for the given object.
|
|
func (n *namespacedClient) GroupVersionKindFor(obj runtime.Object) (schema.GroupVersionKind, error) {
|
|
return n.client.GroupVersionKindFor(obj)
|
|
}
|
|
|
|
// IsObjectNamespaced returns true if the GroupVersionKind of the object is namespaced.
|
|
func (n *namespacedClient) IsObjectNamespaced(obj runtime.Object) (bool, error) {
|
|
return n.client.IsObjectNamespaced(obj)
|
|
}
|
|
|
|
// Create implements client.Client.
|
|
func (n *namespacedClient) Create(ctx context.Context, obj Object, opts ...CreateOption) error {
|
|
isNamespaceScoped, err := n.IsObjectNamespaced(obj)
|
|
if err != nil {
|
|
return fmt.Errorf("error finding the scope of the object: %w", err)
|
|
}
|
|
|
|
objectNamespace := obj.GetNamespace()
|
|
if objectNamespace != n.namespace && objectNamespace != "" {
|
|
return fmt.Errorf("namespace %s of the object %s does not match the namespace %s on the client", objectNamespace, obj.GetName(), n.namespace)
|
|
}
|
|
|
|
if isNamespaceScoped && objectNamespace == "" {
|
|
obj.SetNamespace(n.namespace)
|
|
}
|
|
return n.client.Create(ctx, obj, opts...)
|
|
}
|
|
|
|
// Update implements client.Client.
|
|
func (n *namespacedClient) Update(ctx context.Context, obj Object, opts ...UpdateOption) error {
|
|
isNamespaceScoped, err := n.IsObjectNamespaced(obj)
|
|
if err != nil {
|
|
return fmt.Errorf("error finding the scope of the object: %w", err)
|
|
}
|
|
|
|
objectNamespace := obj.GetNamespace()
|
|
if objectNamespace != n.namespace && objectNamespace != "" {
|
|
return fmt.Errorf("namespace %s of the object %s does not match the namespace %s on the client", objectNamespace, obj.GetName(), n.namespace)
|
|
}
|
|
|
|
if isNamespaceScoped && objectNamespace == "" {
|
|
obj.SetNamespace(n.namespace)
|
|
}
|
|
return n.client.Update(ctx, obj, opts...)
|
|
}
|
|
|
|
// Delete implements client.Client.
|
|
func (n *namespacedClient) Delete(ctx context.Context, obj Object, opts ...DeleteOption) error {
|
|
isNamespaceScoped, err := n.IsObjectNamespaced(obj)
|
|
if err != nil {
|
|
return fmt.Errorf("error finding the scope of the object: %w", err)
|
|
}
|
|
|
|
objectNamespace := obj.GetNamespace()
|
|
if objectNamespace != n.namespace && objectNamespace != "" {
|
|
return fmt.Errorf("namespace %s of the object %s does not match the namespace %s on the client", objectNamespace, obj.GetName(), n.namespace)
|
|
}
|
|
|
|
if isNamespaceScoped && objectNamespace == "" {
|
|
obj.SetNamespace(n.namespace)
|
|
}
|
|
return n.client.Delete(ctx, obj, opts...)
|
|
}
|
|
|
|
// DeleteAllOf implements client.Client.
|
|
func (n *namespacedClient) DeleteAllOf(ctx context.Context, obj Object, opts ...DeleteAllOfOption) error {
|
|
isNamespaceScoped, err := n.IsObjectNamespaced(obj)
|
|
if err != nil {
|
|
return fmt.Errorf("error finding the scope of the object: %w", err)
|
|
}
|
|
|
|
if isNamespaceScoped {
|
|
opts = append(opts, InNamespace(n.namespace))
|
|
}
|
|
return n.client.DeleteAllOf(ctx, obj, opts...)
|
|
}
|
|
|
|
// Patch implements client.Client.
|
|
func (n *namespacedClient) Patch(ctx context.Context, obj Object, patch Patch, opts ...PatchOption) error {
|
|
isNamespaceScoped, err := n.IsObjectNamespaced(obj)
|
|
if err != nil {
|
|
return fmt.Errorf("error finding the scope of the object: %w", err)
|
|
}
|
|
|
|
objectNamespace := obj.GetNamespace()
|
|
if objectNamespace != n.namespace && objectNamespace != "" {
|
|
return fmt.Errorf("namespace %s of the object %s does not match the namespace %s on the client", objectNamespace, obj.GetName(), n.namespace)
|
|
}
|
|
|
|
if isNamespaceScoped && objectNamespace == "" {
|
|
obj.SetNamespace(n.namespace)
|
|
}
|
|
return n.client.Patch(ctx, obj, patch, opts...)
|
|
}
|
|
|
|
// Get implements client.Client.
|
|
func (n *namespacedClient) Get(ctx context.Context, key ObjectKey, obj Object, opts ...GetOption) error {
|
|
isNamespaceScoped, err := n.IsObjectNamespaced(obj)
|
|
if err != nil {
|
|
return fmt.Errorf("error finding the scope of the object: %w", err)
|
|
}
|
|
if isNamespaceScoped {
|
|
if key.Namespace != "" && key.Namespace != n.namespace {
|
|
return fmt.Errorf("namespace %s provided for the object %s does not match the namespace %s on the client", key.Namespace, obj.GetName(), n.namespace)
|
|
}
|
|
key.Namespace = n.namespace
|
|
}
|
|
return n.client.Get(ctx, key, obj, opts...)
|
|
}
|
|
|
|
// List implements client.Client.
|
|
func (n *namespacedClient) List(ctx context.Context, obj ObjectList, opts ...ListOption) error {
|
|
if n.namespace != "" {
|
|
opts = append(opts, InNamespace(n.namespace))
|
|
}
|
|
return n.client.List(ctx, obj, opts...)
|
|
}
|
|
|
|
// Status implements client.StatusClient.
|
|
func (n *namespacedClient) Status() SubResourceWriter {
|
|
return n.SubResource("status")
|
|
}
|
|
|
|
// SubResource implements client.SubResourceClient.
|
|
func (n *namespacedClient) SubResource(subResource string) SubResourceClient {
|
|
return &namespacedClientSubResourceClient{client: n.client.SubResource(subResource), namespace: n.namespace, namespacedclient: n}
|
|
}
|
|
|
|
// ensure namespacedClientSubResourceClient implements client.SubResourceClient.
|
|
var _ SubResourceClient = &namespacedClientSubResourceClient{}
|
|
|
|
type namespacedClientSubResourceClient struct {
|
|
client SubResourceClient
|
|
namespace string
|
|
namespacedclient Client
|
|
}
|
|
|
|
func (nsw *namespacedClientSubResourceClient) Get(ctx context.Context, obj, subResource Object, opts ...SubResourceGetOption) error {
|
|
isNamespaceScoped, err := nsw.namespacedclient.IsObjectNamespaced(obj)
|
|
if err != nil {
|
|
return fmt.Errorf("error finding the scope of the object: %w", err)
|
|
}
|
|
|
|
objectNamespace := obj.GetNamespace()
|
|
if objectNamespace != nsw.namespace && objectNamespace != "" {
|
|
return fmt.Errorf("namespace %s of the object %s does not match the namespace %s on the client", objectNamespace, obj.GetName(), nsw.namespace)
|
|
}
|
|
|
|
if isNamespaceScoped && objectNamespace == "" {
|
|
obj.SetNamespace(nsw.namespace)
|
|
}
|
|
|
|
return nsw.client.Get(ctx, obj, subResource, opts...)
|
|
}
|
|
|
|
func (nsw *namespacedClientSubResourceClient) Create(ctx context.Context, obj, subResource Object, opts ...SubResourceCreateOption) error {
|
|
isNamespaceScoped, err := nsw.namespacedclient.IsObjectNamespaced(obj)
|
|
if err != nil {
|
|
return fmt.Errorf("error finding the scope of the object: %w", err)
|
|
}
|
|
|
|
objectNamespace := obj.GetNamespace()
|
|
if objectNamespace != nsw.namespace && objectNamespace != "" {
|
|
return fmt.Errorf("namespace %s of the object %s does not match the namespace %s on the client", objectNamespace, obj.GetName(), nsw.namespace)
|
|
}
|
|
|
|
if isNamespaceScoped && objectNamespace == "" {
|
|
obj.SetNamespace(nsw.namespace)
|
|
}
|
|
|
|
return nsw.client.Create(ctx, obj, subResource, opts...)
|
|
}
|
|
|
|
// Update implements client.SubResourceWriter.
|
|
func (nsw *namespacedClientSubResourceClient) Update(ctx context.Context, obj Object, opts ...SubResourceUpdateOption) error {
|
|
isNamespaceScoped, err := nsw.namespacedclient.IsObjectNamespaced(obj)
|
|
if err != nil {
|
|
return fmt.Errorf("error finding the scope of the object: %w", err)
|
|
}
|
|
|
|
objectNamespace := obj.GetNamespace()
|
|
if objectNamespace != nsw.namespace && objectNamespace != "" {
|
|
return fmt.Errorf("namespace %s of the object %s does not match the namespace %s on the client", objectNamespace, obj.GetName(), nsw.namespace)
|
|
}
|
|
|
|
if isNamespaceScoped && objectNamespace == "" {
|
|
obj.SetNamespace(nsw.namespace)
|
|
}
|
|
return nsw.client.Update(ctx, obj, opts...)
|
|
}
|
|
|
|
// Patch implements client.SubResourceWriter.
|
|
func (nsw *namespacedClientSubResourceClient) Patch(ctx context.Context, obj Object, patch Patch, opts ...SubResourcePatchOption) error {
|
|
isNamespaceScoped, err := nsw.namespacedclient.IsObjectNamespaced(obj)
|
|
if err != nil {
|
|
return fmt.Errorf("error finding the scope of the object: %w", err)
|
|
}
|
|
|
|
objectNamespace := obj.GetNamespace()
|
|
if objectNamespace != nsw.namespace && objectNamespace != "" {
|
|
return fmt.Errorf("namespace %s of the object %s does not match the namespace %s on the client", objectNamespace, obj.GetName(), nsw.namespace)
|
|
}
|
|
|
|
if isNamespaceScoped && objectNamespace == "" {
|
|
obj.SetNamespace(nsw.namespace)
|
|
}
|
|
return nsw.client.Patch(ctx, obj, patch, opts...)
|
|
}
|