Add defualt snapshotclass handling

cmd/csi-snapshotter/main.go

@@ -134,6 +134,7 @@ func main() {
 		*snapshotter,
 		factory.Volumesnapshot().V1alpha1().VolumeSnapshots(),
 		factory.Volumesnapshot().V1alpha1().VolumeSnapshotContents(),
+		factory.Volumesnapshot().V1alpha1().VolumeSnapshotClasses(),
 		*createSnapshotContentRetryCount,
 		*createSnapshotContentInterval,
 		csiConn,

pkg/controller/controller.go

@@ -51,6 +51,8 @@ type CSISnapshotController struct {
 	snapshotListerSynced cache.InformerSynced
 	contentLister        storagelisters.VolumeSnapshotContentLister
 	contentListerSynced  cache.InformerSynced
+	classLister			 storagelisters.VolumeSnapshotClassLister
+	classListerSynced	 cache.InformerSynced
 
 	snapshotStore cache.Store
 	contentStore  cache.Store
@@ -71,6 +73,7 @@ func NewCSISnapshotController(
 	snapshotterName string,
 	volumeSnapshotInformer storageinformers.VolumeSnapshotInformer,
 	volumeSnapshotContentInformer storageinformers.VolumeSnapshotContentInformer,
+	volumeSnapshotClassInformer storageinformers.VolumeSnapshotClassInformer,
 	createSnapshotContentRetryCount int,
 	createSnapshotContentInterval time.Duration,
 	conn connection.CSIConnection,
@@ -122,6 +125,9 @@ func NewCSISnapshotController(
 	ctrl.contentLister = volumeSnapshotContentInformer.Lister()
 	ctrl.contentListerSynced = volumeSnapshotContentInformer.Informer().HasSynced
 
+	ctrl.classLister = volumeSnapshotClassInformer.Lister()
+	ctrl.classListerSynced = volumeSnapshotClassInformer.Informer().HasSynced
+
 	return ctrl
 }
 

pkg/controller/csi_snapshot_controller.go

@@ -31,6 +31,7 @@ import (
 	ref "k8s.io/client-go/tools/reference"
 	"k8s.io/kubernetes/pkg/util/goroutinemap"
 	"k8s.io/kubernetes/pkg/util/goroutinemap/exponentialbackoff"
+	"k8s.io/apimachinery/pkg/labels"
 )
 
 // ==================================================================
@@ -74,6 +75,8 @@ import (
 
 const pvcKind = "PersistentVolumeClaim"
 
+const IsDefaultSnapshotClassAnnotation = "snapshot.storage.kubernetes.io/is-default-class"
+
 // syncContent deals with one key off the queue.  It returns false when it's time to quit.
 func (ctrl *CSISnapshotController) syncContent(content *crdv1.VolumeSnapshotContent) error {
 	glog.V(4).Infof("synchronizing VolumeSnapshotContent[%s]", content.Name)
@@ -674,12 +677,48 @@ func (ctrl *CSISnapshotController) getVolumeFromVolumeSnapshot(snapshot *crdv1.V
 func (ctrl *CSISnapshotController) GetClassFromVolumeSnapshot(snapshot *crdv1.VolumeSnapshot) (*crdv1.VolumeSnapshotClass, error) {
 	className := snapshot.Spec.VolumeSnapshotClassName
 	glog.V(5).Infof("getClassFromVolumeSnapshot [%s]: VolumeSnapshotClassName [%s]", snapshot.Name, className)
-	class, err := ctrl.clientset.VolumesnapshotV1alpha1().VolumeSnapshotClasses().Get(className, metav1.GetOptions{})
-	if err != nil {
-		glog.Errorf("failed to retrieve storage class %s from the API server: %q", className, err)
-		//return nil, fmt.Errorf("failed to retrieve storage class %s from the API server: %q", className, err)
+	if len(className) > 0 {
+		class, err := ctrl.clientset.VolumesnapshotV1alpha1().VolumeSnapshotClasses().Get(className, metav1.GetOptions{})
+		if err != nil {
+			glog.Errorf("failed to retrieve storage class %s from the API server: %q", className, err)
+			//return nil, fmt.Errorf("failed to retrieve storage class %s from the API server: %q", className, err)
+		}
+		glog.V(5).Infof("getClassFromVolumeSnapshot [%s]: VolumeSnapshotClassName [%s]", snapshot.Name, className)
+		return class, nil
+	} else {
+		// Find default snapshot class if available
+		list, err := ctrl.classLister.List(labels.Everything())
+        if err != nil {
+                return nil, err
+        }
+		pvc, err := ctrl.getClaimFromVolumeSnapshot(snapshot)
+		if err != nil {
+			return nil, err
+		}
+		storageclass, err := ctrl.client.StorageV1().StorageClasses().Get(*pvc.Spec.StorageClassName, metav1.GetOptions{})
+		if err != nil {
+			return nil, err
+		}
+		defaultClasses := []*crdv1.VolumeSnapshotClass{}
+
+        for _, class := range list {
+                if IsDefaultAnnotation(class.ObjectMeta) && storageclass.Provisioner == class.Snapshotter {
+                        defaultClasses = append(defaultClasses, class)
+                        glog.V(4).Infof("getDefaultClass added: %s", class.Name)
+                }
+        }
+
+        if len(defaultClasses) == 0 {
+                return nil, nil
+        }
+        if len(defaultClasses) > 1 {
+                glog.V(4).Infof("getDefaultClass %d defaults found", len(defaultClasses))
+                return nil, fmt.Errorf("%d default StorageClasses were found", len(defaultClasses))
+		}
+		glog.V(5).Infof("getClassFromVolumeSnapshot [%s]: default VolumeSnapshotClassName [%s]", snapshot.Name, defaultClasses[0])
+        return defaultClasses[0], nil
+
 	}
-	return class, nil
 }
 
 // getClaimFromVolumeSnapshot is a helper function to get PV from VolumeSnapshot.

pkg/controller/util.go

@@ -20,6 +20,7 @@ import (
 	"fmt"
 	"github.com/golang/glog"
 	crdv1 "github.com/kubernetes-csi/external-snapshotter/pkg/apis/volumesnapshot/v1alpha1"
+	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
 	"k8s.io/api/core/v1"
 	"k8s.io/apimachinery/pkg/api/meta"
 	"k8s.io/client-go/tools/cache"
@@ -110,3 +111,14 @@ func storeObjectUpdate(store cache.Store, obj interface{}, className string) (bo
 func GetSnapshotContentNameForSnapshot(snapshot *crdv1.VolumeSnapshot) string {
 	return "snapdata-" + string(snapshot.UID)
 }
+
+
+// IsDefaultAnnotation returns a boolean if
+// the annotation is set
+func IsDefaultAnnotation(obj metav1.ObjectMeta) bool {
+	if obj.Annotations[IsDefaultSnapshotClassAnnotation] == "true" {
+		return true
+	}
+
+	return false
+}