[wsman-mk2] Emit workspace events

Author: Furisto

components/ws-daemon/pkg/controller/workspace_controller.go

@@ -26,6 +26,7 @@ import (
 	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
 	"k8s.io/apimachinery/pkg/types"
 	"k8s.io/apimachinery/pkg/util/wait"
+	"k8s.io/client-go/tools/record"
 	"k8s.io/client-go/util/retry"
 	ctrl "sigs.k8s.io/controller-runtime"
 	"sigs.k8s.io/controller-runtime/pkg/client"
@@ -58,9 +59,10 @@ type WorkspaceController struct {
 	operations              *WorkspaceOperations
 	metrics                 *workspaceMetrics
 	secretNamespace         string
+	recorder                record.EventRecorder
 }
 
-func NewWorkspaceController(c client.Client, nodeName, secretNamespace string, maxConcurrentReconciles int, ops *WorkspaceOperations, reg prometheus.Registerer) (*WorkspaceController, error) {
+func NewWorkspaceController(c client.Client, recorder record.EventRecorder, nodeName, secretNamespace string, maxConcurrentReconciles int, ops *WorkspaceOperations, reg prometheus.Registerer) (*WorkspaceController, error) {
 	metrics := newWorkspaceMetrics()
 	reg.Register(metrics)
 
@@ -71,6 +73,7 @@ func NewWorkspaceController(c client.Client, nodeName, secretNamespace string, m
 		operations:              ops,
 		metrics:                 metrics,
 		secretNamespace:         secretNamespace,
+		recorder:                recorder,
 	}, nil
 }
 
@@ -182,6 +185,7 @@ func (wsc *WorkspaceController) handleWorkspaceInit(ctx context.Context, ws *wor
 			wsc.metrics.recordInitializeTime(time.Since(initStart).Seconds(), ws)
 		}
 
+		wsc.emitEvent(ws, "Content init", initErr)
 		return ctrl.Result{}, err
 	}
 
@@ -277,6 +281,7 @@ func (wsc *WorkspaceController) handleWorkspaceStop(ctx context.Context, ws *wor
 		wsc.metrics.recordFinalizeTime(time.Since(disposeStart).Seconds(), ws)
 	}
 
+	wsc.emitEvent(ws, "Backup", disposeErr)
 	return ctrl.Result{}, err
 }
 
@@ -301,6 +306,12 @@ func (wsc *WorkspaceController) prepareInitializer(ctx context.Context, ws *work
 	return &init, nil
 }
 
+func (wsc *WorkspaceController) emitEvent(ws *workspacev1.Workspace, operation string, failure error) {
+	if failure != nil {
+		wsc.recorder.Eventf(ws, corev1.EventTypeWarning, "Failed", "%s failed: %s", operation, failure.Error())
+	}
+}
+
 func toWorkspaceGitStatus(status *csapi.GitStatus) *workspacev1.GitStatus {
 	if status == nil {
 		return nil

components/ws-daemon/pkg/daemon/daemon.go

@@ -210,7 +210,7 @@ func NewDaemon(config Config) (*Daemon, error) {
 		}
 
 		wsctrl, err := controller.NewWorkspaceController(
-			mgr.GetClient(), nodename, config.Runtime.SecretsNamespace, config.WorkspaceController.MaxConcurrentReconciles, workspaceOps, wrappedReg)
+			mgr.GetClient(), mgr.GetEventRecorderFor("workspace"), nodename, config.Runtime.SecretsNamespace, config.WorkspaceController.MaxConcurrentReconciles, workspaceOps, wrappedReg)
 		if err != nil {
 			return nil, err
 		}
@@ -219,7 +219,8 @@ func NewDaemon(config Config) (*Daemon, error) {
 			return nil, err
 		}
 
-		ssctrl := controller.NewSnapshotController(mgr.GetClient(), nodename, config.WorkspaceController.MaxConcurrentReconciles, workspaceOps)
+		ssctrl := controller.NewSnapshotController(
+			mgr.GetClient(), mgr.GetEventRecorderFor("snapshot"), nodename, config.WorkspaceController.MaxConcurrentReconciles, workspaceOps)
 		err = ssctrl.SetupWithManager(mgr)
 		if err != nil {
 			return nil, err

components/ws-manager-mk2/controllers/status.go

@@ -31,7 +31,7 @@ const (
 	containerUnknownExitCode = 255
 )
 
-func updateWorkspaceStatus(ctx context.Context, workspace *workspacev1.Workspace, pods corev1.PodList, cfg *config.Configuration) error {
+func (r *WorkspaceReconciler) updateWorkspaceStatus(ctx context.Context, workspace *workspacev1.Workspace, pods corev1.PodList, cfg *config.Configuration) error {
 	log := log.FromContext(ctx)
 
 	switch len(pods.Items) {
@@ -104,6 +104,8 @@ func updateWorkspaceStatus(ctx context.Context, workspace *workspacev1.Workspace
 			LastTransitionTime: metav1.Now(),
 			Message:            failure,
 		})
+
+		r.Recorder.Event(workspace, corev1.EventTypeWarning, "Failed", failure)
 	}
 
 	switch {

components/ws-manager-mk2/controllers/suite_test.go

@@ -105,7 +105,7 @@ var _ = BeforeSuite(func() {
 	Expect(err).ToNot(HaveOccurred())
 
 	conf := newTestConfig()
-	wsReconciler, err := NewWorkspaceReconciler(k8sManager.GetClient(), k8sManager.GetScheme(), &conf, metrics.Registry, &fakeMaintenance{enabled: false})
+	wsReconciler, err := NewWorkspaceReconciler(k8sManager.GetClient(), k8sManager.GetScheme(), k8sManager.GetEventRecorderFor("workspace"), &conf, metrics.Registry, &fakeMaintenance{enabled: false})
 	wsMetrics = wsReconciler.metrics
 	Expect(err).ToNot(HaveOccurred())
 	Expect(wsReconciler.SetupWithManager(k8sManager)).To(Succeed())

components/ws-manager-mk2/controllers/workspace_controller.go

@@ -16,6 +16,7 @@ import (
 	"k8s.io/apimachinery/pkg/runtime"
 	"k8s.io/apimachinery/pkg/types"
 	"k8s.io/apimachinery/pkg/util/wait"
+	"k8s.io/client-go/tools/record"
 	ctrl "sigs.k8s.io/controller-runtime"
 	"sigs.k8s.io/controller-runtime/pkg/client"
 	"sigs.k8s.io/controller-runtime/pkg/controller"
@@ -37,12 +38,13 @@ const (
 	maintenanceRequeue         = 1 * time.Minute
 )
 
-func NewWorkspaceReconciler(c client.Client, scheme *runtime.Scheme, cfg *config.Configuration, reg prometheus.Registerer, maintenance maintenance.Maintenance) (*WorkspaceReconciler, error) {
+func NewWorkspaceReconciler(c client.Client, scheme *runtime.Scheme, recorder record.EventRecorder, cfg *config.Configuration, reg prometheus.Registerer, maintenance maintenance.Maintenance) (*WorkspaceReconciler, error) {
 	reconciler := &WorkspaceReconciler{
 		Client:      c,
 		Scheme:      scheme,
 		Config:      cfg,
 		maintenance: maintenance,
+		Recorder:    recorder,
 	}
 
 	metrics, err := newControllerMetrics(reconciler)
@@ -63,6 +65,7 @@ type WorkspaceReconciler struct {
 	Config      *config.Configuration
 	metrics     *controllerMetrics
 	maintenance maintenance.Maintenance
+	Recorder    record.EventRecorder
 	OnReconcile func(ctx context.Context, ws *workspacev1.Workspace)
 }
 
@@ -114,12 +117,14 @@ func (r *WorkspaceReconciler) Reconcile(ctx context.Context, req ctrl.Request) (
 		return ctrl.Result{}, err
 	}
 
-	err = updateWorkspaceStatus(ctx, &workspace, workspacePods, r.Config)
+	oldStatus := workspace.Status
+	err = r.updateWorkspaceStatus(ctx, &workspace, workspacePods, r.Config)
 	if err != nil {
 		return ctrl.Result{}, err
 	}
 
 	r.updateMetrics(ctx, &workspace)
+	r.emitPhaseEvents(ctx, &workspace, oldStatus)
 
 	log.V(1).Info("updated workspace status", "status", workspace.Status)
 	err = r.Status().Update(ctx, &workspace)
@@ -185,6 +190,8 @@ func (r *WorkspaceReconciler) actOnStatus(ctx context.Context, workspace *worksp
 					log.Error(err, "Failed to patch PodStarts in workspace status")
 					return ctrl.Result{}, err
 				}
+
+				r.Recorder.Event(workspace, corev1.EventTypeNormal, "Creating", "")
 			}
 			r.metrics.rememberWorkspace(workspace, nil)
 
@@ -203,6 +210,7 @@ func (r *WorkspaceReconciler) actOnStatus(ctx context.Context, workspace *worksp
 
 			// Workspace might have already been in a deleting state,
 			// but not guaranteed, so try deleting anyway.
+			r.Recorder.Event(workspace, corev1.EventTypeNormal, "Deleting", "")
 			err := r.Client.Delete(ctx, workspace)
 			return ctrl.Result{}, client.IgnoreNotFound(err)
 		}
@@ -340,6 +348,20 @@ func (r *WorkspaceReconciler) updateMetrics(ctx context.Context, workspace *work
 	r.metrics.rememberWorkspace(workspace, &lastState)
 }
 
+func (r *WorkspaceReconciler) emitPhaseEvents(ctx context.Context, ws *workspacev1.Workspace, old workspacev1.WorkspaceStatus) {
+	if ws.Status.Phase == workspacev1.WorkspacePhaseInitializing && old.Phase != workspacev1.WorkspacePhaseInitializing {
+		r.Recorder.Event(ws, corev1.EventTypeNormal, "Initializing", "")
+	}
+
+	if ws.Status.Phase == workspacev1.WorkspacePhaseRunning && old.Phase != workspacev1.WorkspacePhaseRunning {
+		r.Recorder.Event(ws, corev1.EventTypeNormal, "Running", "")
+	}
+
+	if ws.Status.Phase == workspacev1.WorkspacePhaseStopping && old.Phase != workspacev1.WorkspacePhaseStopping {
+		r.Recorder.Event(ws, corev1.EventTypeNormal, "Stopping", "")
+	}
+}
+
 func (r *WorkspaceReconciler) deleteWorkspacePod(ctx context.Context, pod *corev1.Pod, reason string) (ctrl.Result, error) {
 	log := log.FromContext(ctx).WithValues("workspace", pod.Name, "reason", reason)
 	log.V(1).Info("deleting workspace pod")

components/ws-manager-mk2/main.go

@@ -113,13 +113,14 @@ func main() {
 		os.Exit(1)
 	}
 
-	maintenance, err := controllers.NewMaintenanceReconciler(mgr.GetClient())
+	maintenanceReconciler, err := controllers.NewMaintenanceReconciler(mgr.GetClient())
 	if err != nil {
 		setupLog.Error(err, "unable to create maintenance controller", "controller", "Maintenance")
 		os.Exit(1)
 	}
 
-	reconciler, err := controllers.NewWorkspaceReconciler(mgr.GetClient(), mgr.GetScheme(), &cfg.Manager, metrics.Registry, maintenance)
+	workspaceReconciler, err := controllers.NewWorkspaceReconciler(
+		mgr.GetClient(), mgr.GetScheme(), mgr.GetEventRecorderFor("workspace"), &cfg.Manager, metrics.Registry, maintenanceReconciler)
 	if err != nil {
 		setupLog.Error(err, "unable to create controller", "controller", "Workspace")
 		os.Exit(1)
@@ -132,22 +133,22 @@ func main() {
 		os.Exit(1)
 	}
 
-	wsmanService, err := setupGRPCService(cfg, mgr.GetClient(), activity, maintenance)
+	wsmanService, err := setupGRPCService(cfg, mgr.GetClient(), activity, maintenanceReconciler)
 	if err != nil {
 		setupLog.Error(err, "unable to start manager service")
 		os.Exit(1)
 	}
 
-	reconciler.OnReconcile = wsmanService.OnWorkspaceReconcile
-	if err = reconciler.SetupWithManager(mgr); err != nil {
+	workspaceReconciler.OnReconcile = wsmanService.OnWorkspaceReconcile
+	if err = workspaceReconciler.SetupWithManager(mgr); err != nil {
 		setupLog.Error(err, "unable to setup workspace controller with manager", "controller", "Workspace")
 		os.Exit(1)
 	}
 	if err = timeoutReconciler.SetupWithManager(mgr); err != nil {
 		setupLog.Error(err, "unable to setup timeout controller with manager", "controller", "Timeout")
 		os.Exit(1)
 	}
-	if err = maintenance.SetupWithManager(mgr); err != nil {
+	if err = maintenanceReconciler.SetupWithManager(mgr); err != nil {
 		setupLog.Error(err, "unable to setup maintenance controller with manager", "controller", "Maintenance")
 		os.Exit(1)
 	}