More updates to the bucket claim controller

Author: mukhoakash

cmd/controller-manager/controller-manager.go

@@ -11,7 +11,6 @@ import (
 	"github.com/spf13/viper"
 
 	bucketcontroller "sigs.k8s.io/container-object-storage-interface-api/controller"
-	"sigs.k8s.io/container-object-storage-interface-controller/pkg/bucketaccessrequest"
 	"sigs.k8s.io/container-object-storage-interface-controller/pkg/bucketclaim"
 
 	"k8s.io/klog/v2"
@@ -68,6 +67,5 @@ func run(ctx context.Context, args []string) error {
 		return err
 	}
 	ctrl.AddBucketClaimListener(bucketclaim.NewBucketClaimListener())
-	ctrl.AddBucketAccessRequestListener(bucketaccessrequest.NewListener())
 	return ctrl.Run(ctx)
 }

pkg/bucketclaim/bucketclaim.go

@@ -17,10 +17,6 @@ import (
 	"sigs.k8s.io/container-object-storage-interface-controller/pkg/util"
 )
 
-const (
-	finalizer = "cosi.objectstorage.k8s.io/bucketclaim-protection"
-)
-
 // bucketClaimListener is a resource handler for bucket requests objects
 type bucketClaimListener struct {
 	kubeClient   kubeclientset.Interface
@@ -34,8 +30,8 @@ func NewBucketClaimListener() *bucketClaimListener {
 // Add creates a bucket in response to a bucketClaim
 func (b *bucketClaimListener) Add(ctx context.Context, bucketClaim *v1alpha1.BucketClaim) error {
 	klog.V(3).InfoS("Add BucketClaim",
-		"name", bucketClaim.Name,
-		"ns", bucketClaim.Namespace,
+		"name", bucketClaim.ObjectMeta.Name,
+		"ns", bucketClaim.ObjectMeta.Namespace,
 		"bucketClass", bucketClaim.Spec.BucketClassName,
 		"bucketPrefix", bucketClaim.Spec.BucketPrefix,
 	)
@@ -45,27 +41,27 @@ func (b *bucketClaimListener) Add(ctx context.Context, bucketClaim *v1alpha1.Buc
 		switch err {
 		case util.ErrInvalidBucketClass:
 			klog.ErrorS(util.ErrInvalidBucketClass,
-				"bucketClaim", bucketClaim.Name,
-				"ns", bucketClaim.Namespace,
+				"bucketClaim", bucketClaim.ObjectMeta.Name,
+				"ns", bucketClaim.ObjectMeta.Namespace,
 				"bucketClassName", bucketClaim.Spec.BucketClassName)
 		case util.ErrBucketAlreadyExists:
 			klog.V(3).InfoS("Bucket already exists",
-				"bucketClaim", bucketClaim.Name,
-				"ns", bucketClaim.Namespace,
+				"bucketClaim", bucketClaim.ObjectMeta.Name,
+				"ns", bucketClaim.ObjectMeta.Namespace,
 			)
 			return nil
 		default:
 			klog.ErrorS(err,
-				"name", bucketClaim.Name,
-				"ns", bucketClaim.Namespace,
+				"name", bucketClaim.ObjectMeta.Name,
+				"ns", bucketClaim.ObjectMeta.Namespace,
 				"err", err)
 		}
 		return err
 	}
 
 	klog.V(3).InfoS("Add BucketClaim success",
-		"name", bucketClaim.Name,
-		"ns", bucketClaim.Namespace)
+		"name", bucketClaim.ObjectMeta.Name,
+		"ns", bucketClaim.ObjectMeta.Namespace)
 	return nil
 }
 
@@ -74,14 +70,24 @@ func (b *bucketClaimListener) Update(ctx context.Context, old, new *v1alpha1.Buc
 	klog.V(3).InfoS("Update BucketClaim",
 		"name", old.Name,
 		"ns", old.Namespace)
+
+	if !new.GetDeletionTimestamp().IsZero() {
+		if controllerutil.ContainsFinalizer(bucketClaim, util.BucketClaimFinalizer) {
+			bucketName := bucketClaim.Status.BucketName
+			err := b.Buckets().Delete(ctx, bucketName, metav1.DeleteOptions{})
+			if err != nil {
+				return err
+			}
+		}
+	}
 	return nil
 }
 
 // Delete processes a bucket for which bucket request is deleted
 func (b *bucketClaimListener) Delete(ctx context.Context, bucketClaim *v1alpha1.BucketClaim) error {
 	klog.V(3).Infof("Delete BucketClaim  %v",
-		"name", bucketClaim.Name,
-		"ns", bucketClaim.Namespace)
+		"name", bucketClaim.ObjectMeta.Name,
+		"ns", bucketClaim.ObjectMeta.Namespace)
 
 	return nil
 }
@@ -93,50 +99,82 @@ func (b *bucketClaimListener) Delete(ctx context.Context, bucketClaim *v1alpha1.
 //    ErrBucketAlreadyExists - BucketClaim already processed
 //    non-nil err - Internal error                                [requeue'd with exponential backoff]
 func (b *bucketClaimListener) provisionBucketClaimOperation(ctx context.Context, bucketClaim *v1alpha1.BucketClaim) error {
-	bucketClassName := b.getBucketClass(bucketClaim)
-	bucketClass, err := b.BucketClasses().Get(ctx, bucketClassName, metav1.GetOptions{})
-	if err != nil {
-		klog.ErrorS(err, "Get Bucketclass Error", "name", bucketClassName)
-		return util.ErrInvalidBucketClass
-	}
-
 	if bucketClaim.Status.BucketReady {
 		return util.ErrBucketAlreadyExists
 	}
 
-	name = bucketClassName + string(bucketClaim.GetUID())
-
-	// create bucket
-	bucket := &v1alpha1.Bucket{}
-	bucket.Name = name
-	bucket.Status.BucketReady = false
-	bucket.Spec.DriverName = bucketClass.DriverName
-	bucket.Spec.BucketClassName = bucketClassName
-	bucket.Spec.DeletionPolicy = bucketClass.DeletionPolicy
-	bucket.Spec.BucketClaim = &v1.ObjectReference{
-		Name:      bucketClaim.Name,
-		Namespace: bucketClaim.Namespace,
-		UID:       bucketClaim.ObjectMeta.UID,
+	var bucketName string
+
+	if bucketClaim.Spec.ExistingBucketName != "" {
+		bucketName = bucketClaim.Spec.ExistingBucketName
+		bucket, err = b.Buckets().Get(ctx, bucketName, metav1.GetOptions{})
+		if err != nil {
+			klog.ErrorS(err, "Get Bucket with ExistingBucketName error", "name", existingBucketName)
+			return err
+		}
+
+		bucket.Spec.BucketClaim = &v1.ObjectReference{
+			Name:      bucketClaim.ObjectMeta.Name,
+			Namespace: bucketClaim.ObjectMeta.Namespace,
+			UID:       bucketClaim.ObjectMeta.UID,
+		}
+
+		_, err = b.Buckets().Update(ctx, bucket, metav1.UpdateOptions{})
+		if err != nil {
+			return err
+		}
+
+		bucketClaim.Status.BucketName = bucketName
+		bucketClaim.Status.BucketAvailable = true
+	} else {
+		bucketClassName := b.getBucketClass(bucketClaim)
+		bucketClass, err := b.BucketClasses().Get(ctx, bucketClassName, metav1.GetOptions{})
+		if err != nil {
+			klog.ErrorS(err, "Get Bucketclass Error", "name", bucketClassName)
+			return util.ErrInvalidBucketClass
+		}
+
+		bucketName = bucketClassName + string(bucketClaim.ObjectMeta.UID)
+
+		// create bucket
+		bucket := &v1alpha1.Bucket{}
+		bucket.Name = bucketName
+		bucket.Spec.DriverName = bucketClass.DriverName
+		bucket.Status.BucketReady = false
+		bucket.Spec.BucketClassName = bucketClassName
+		bucket.Spec.DeletionPolicy = bucketClass.DeletionPolicy
+		bucket.Spec.Parameters = util.CopySS(bucketClass.Parameters)
+
+		bucket.Spec.BucketClaim = &v1.ObjectReference{
+			Name:      bucketClaim.ObjectMeta.Name,
+			Namespace: bucketClaim.ObjectMeta.Namespace,
+			UID:       bucketClaim.ObjectMeta.UID,
+		}
+
+		bucket.Spec.Protocols = *bucketClaim.Spec.Protocol.DeepCopy()
+		bucket, err = b.Buckets().Create(ctx, bucket, metav1.CreateOptions{})
+		if err != nil && !errors.IsAlreadyExists(err) {
+			klog.ErrorS(err, "name", bucketName)
+			return err
+		}
+
+		bucketClaim.Status.BucketName = bucketName
+		bucketClaim.Status.BucketAvailable = false
 	}
-	bucket.Spec.Protocols = *bucketClass.Protocol.DeepCopy()
-	bucket.Spec.Parameters = util.CopySS(bucketClass.Parameters)
 
-	bucket, err = b.Buckets().Create(ctx, bucket, metav1.CreateOptions{})
-	if err != nil && !errors.IsAlreadyExists(err) {
-		klog.ErrorS(err, "name", bucket.Name)
+	_, err = b.BucketClaims(bucketClaim.ObjectMeta.Namespace).UpdateStatus(ctx, bucketClaim, metav1.UpdateOptions{})
+	if err != nil {
 		return err
 	}
 
-	// controllerutil.AddFinalizer(bucketClaim, finalizer)
-
-	bucketClaim.Status.BucketName = bucket.Name
-	bucketClaim.Status.BucketAvailable = false
-	_, err = b.BucketClaims(bucketClaim.Namespace).UpdateStatus(ctx, bucketClaim, metav1.UpdateOptions{})
+	// Add the finalizers so that bucketClaim is deleted
+	// only after the associated bucket is deleted.
+	controllerutil.AddFinalizer(bucketClaim, util.BucketClaimFinalizer)
+	_, err = b.BucketClaims(bucketClaim.ObjectMeta.Namespace).Update(ctx, bucketClaim, metav1.UpdateOptions{})
 	if err != nil {
 		return err
 	}
-
-	klog.Infof("Finished creating Bucket %v", bucket.Name)
+	klog.Infof("Finished creating Bucket %v", bucketName)
 	return nil
 }
 

pkg/util/const.go

@@ -4,6 +4,10 @@ import (
 	"errors"
 )
 
+const (
+	BucketClaimFinalizer = "cosi.objectstorage.k8s.io/bucketclaim-protection"
+)
+
 var (
 	// Error codes that the central controller will return
 	ErrBucketAlreadyExists        = errors.New("A bucket already existing that matches the bucket request")

pkg/util/util.go

@@ -3,7 +3,6 @@ package util
 import (
 	"context"
 	"encoding/json"
-	"errors"
 	"fmt"
 	"golang.org/x/time/rate"
 	"os"
@@ -228,65 +227,6 @@ func DeleteObjects(ctx context.Context, client bucketclientset.Interface, objs .
 	}
 }
 
-// CreateBucketRequest creates a bucket request object or return an existing bucket request object
-// This is used by bucket request unit tests
-func CreateBucketRequest(ctx context.Context, client bucketclientset.Interface, br *types.BucketRequest) (*types.BucketRequest, error) {
-	br, err := client.ObjectstorageV1alpha1().BucketRequests(br.Namespace).Create(ctx, br, metav1.CreateOptions{})
-	if (err != nil) && apierrors.IsAlreadyExists(err) {
-		br, err = client.ObjectstorageV1alpha1().BucketRequests(br.Namespace).Get(ctx, br.Name, metav1.GetOptions{})
-	}
-	return br, err
-}
-
-// CreateBucketAccessRequest creates a bucket access request object or return an existing bucket access request object
-// This is used by bucket access request unit tests
-func CreateBucketAccessRequest(ctx context.Context, client bucketclientset.Interface, bar *types.BucketAccessRequest) (*types.BucketAccessRequest, error) {
-	bar, err := client.ObjectstorageV1alpha1().BucketAccessRequests(bar.Namespace).Create(ctx, bar, metav1.CreateOptions{})
-	if (err != nil) && apierrors.IsAlreadyExists(err) {
-		bar, err = client.ObjectstorageV1alpha1().BucketAccessRequests(bar.Namespace).Get(ctx, bar.Name, metav1.GetOptions{})
-	}
-	return bar, err
-}
-
-// CreateBucketClass creates a bucket class object or return an existing bucket class object
-// This is used by bucket request unit tests
-func CreateBucketClass(ctx context.Context, client bucketclientset.Interface, bc *types.BucketClass) (*types.BucketClass, error) {
-	bc, err := client.ObjectstorageV1alpha1().BucketClasses().Create(ctx, bc, metav1.CreateOptions{})
-	if (err != nil) && apierrors.IsAlreadyExists(err) {
-		bc, err = client.ObjectstorageV1alpha1().BucketClasses().Get(ctx, bc.Name, metav1.GetOptions{})
-	}
-	return bc, err
-}
-
-// CreateBucketAccessClass creates a bucket access class object or return an existing bucket access class object
-// This is used by bucket request unit tests
-func CreateBucketAccessClass(ctx context.Context, client bucketclientset.Interface, bac *types.BucketAccessClass) (*types.BucketAccessClass, error) {
-	bac, err := client.ObjectstorageV1alpha1().BucketAccessClasses().Create(ctx, bac, metav1.CreateOptions{})
-	if (err != nil) && apierrors.IsAlreadyExists(err) {
-		bac, err = client.ObjectstorageV1alpha1().BucketAccessClasses().Get(ctx, bac.Name, metav1.GetOptions{})
-	}
-	return bac, err
-}
-
-// CreateBucket creates a bucket object or return an existing bucket object
-// This is used by bucket request unit tests
-func CreateBucket(ctx context.Context, client bucketclientset.Interface, b *types.Bucket) (*types.Bucket, error) {
-	b, err := client.ObjectstorageV1alpha1().Buckets().Create(ctx, b, metav1.CreateOptions{})
-	if (err != nil) && apierrors.IsAlreadyExists(err) {
-		b, err = client.ObjectstorageV1alpha1().Buckets().Get(ctx, b.Name, metav1.GetOptions{})
-	}
-	return b, err
-}
-
-// CreateBucketAccess creates a bucket access object or return an existing bucket access object
-// This is used by bucket access request unit tests
-func CreateBucketAccess(ctx context.Context, client bucketclientset.Interface, ba *types.BucketAccess) (*types.BucketAccess, error) {
-	ba, err := client.ObjectstorageV1alpha1().BucketAccesses().Create(ctx, ba, metav1.CreateOptions{})
-	if (err != nil) && apierrors.IsAlreadyExists(err) {
-		ba, err = client.ObjectstorageV1alpha1().BucketAccesses().Get(ctx, ba.Name, metav1.GetOptions{})
-	}
-	return ba, err
-}
 
 // getCRDClient returns CRD interface for managing CRD objects programmatically
 // Used by unit tests and functional tests