Add volume_snapshot_controller

Add volume_snapshot_controller to watch the VolumeSnapshot object.

Signed-off-by: JenTing Hsiao <[email protected]>

Author: jenting

components/ws-manager/cmd/run.go

@@ -200,6 +200,16 @@ var runCmd = &cobra.Command{
 			log.WithError(err).Fatal(err, "unable to create controller", "controller", "Pod")
 		}
 
+		err = (&manager.VolumeSnapshotReconciler{
+			Monitor: monitor,
+			Client:  mgr.GetClient(),
+			Log:     ctrl.Log.WithName("controllers").WithName("VolumeSnapshot"),
+			Scheme:  mgr.GetScheme(),
+		}).SetupWithManager(mgr)
+		if err != nil {
+			log.WithError(err).Fatal(err, "unable to create controller", "controller", "VolumeSnapshot")
+		}
+
 		if cfg.PProf.Addr != "" {
 			go pprof.Serve(cfg.PProf.Addr)
 		}

components/ws-manager/pkg/manager/volume_snapshot_controller.go

@@ -0,0 +1,53 @@
+// Copyright (c) 2022 Gitpod GmbH. All rights reserved.
+// Licensed under the GNU Affero General Public License (AGPL).
+// See License-AGPL.txt in the project root for license information.
+
+package manager
+
+import (
+	"context"
+
+	"github.com/go-logr/logr"
+	volumesnapshotv1 "github.com/kubernetes-csi/external-snapshotter/client/v4/apis/volumesnapshot/v1"
+	"k8s.io/apimachinery/pkg/api/errors"
+	"k8s.io/apimachinery/pkg/runtime"
+	"k8s.io/apimachinery/pkg/watch"
+	ctrl "sigs.k8s.io/controller-runtime"
+	"sigs.k8s.io/controller-runtime/pkg/client"
+	"sigs.k8s.io/controller-runtime/pkg/reconcile"
+)
+
+// VolumeSnapshotReconciler reconciles a VolumeSnapshot object
+type VolumeSnapshotReconciler struct {
+	client.Client
+	Log    logr.Logger
+	Scheme *runtime.Scheme
+
+	Monitor *Monitor
+}
+
+// Reconcile performs a reconciliation of a volume snapshot
+// For more details, check Reconcile and its Result here:
+// - https://pkg.go.dev/sigs.k8s.io/[email protected]/pkg/reconcile
+func (r *VolumeSnapshotReconciler) Reconcile(ctx context.Context, req ctrl.Request) (ctrl.Result, error) {
+	var vs volumesnapshotv1.VolumeSnapshot
+	err := r.Client.Get(context.Background(), req.NamespacedName, &vs)
+	if errors.IsNotFound(err) {
+		// volume snapshot is gone - that's ok
+		return reconcile.Result{}, nil
+	}
+
+	r.Monitor.eventpool.Add(vs.Name, watch.Event{
+		Type:   watch.Modified,
+		Object: &vs,
+	})
+
+	return ctrl.Result{}, nil
+}
+
+// SetupWithManager sets up the controller with the Manager.
+func (r *VolumeSnapshotReconciler) SetupWithManager(mgr ctrl.Manager) error {
+	return ctrl.NewControllerManagedBy(mgr).
+		For(&volumesnapshotv1.VolumeSnapshot{}).
+		Complete(r)
+}