Merge pull request kubernetes-sigs#370 from sidharthsurana/feature/su…

…pport-maintenance-mode Add support for Machine maintenance window
jayunit100 · Jun 26, 2019 · 3ba76a0 · 3ba76a0
2 parents dbbf044 + cb4b5b5
commit 3ba76a0
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 0 deletions.
diff --git a/pkg/cloud/vsphere/actuators/machine/actuator.go b/pkg/cloud/vsphere/actuators/machine/actuator.go
@@ -78,6 +78,11 @@ func (a *Actuator) Create(
 		return err
 	}
 
+	if _, ok := machine.Annotations[constants.MaintenanceAnnotationLabel]; ok {
+		ctx.Logger.V(4).Info("skipping operations on machine", "reason", "annotation", "annotation-key", constants.MaintenanceAnnotationLabel)
+		return &clustererr.RequeueAfterError{RequeueAfter: constants.DefaultRequeue}
+	}
+
 	defer func() {
 		opErr = actuators.PatchAndHandleError(ctx, "Create", opErr)
 	}()
@@ -113,6 +118,11 @@ func (a *Actuator) Delete(
 		return err
 	}
 
+	if _, ok := machine.Annotations[constants.MaintenanceAnnotationLabel]; ok {
+		ctx.Logger.V(4).Info("skipping operations on machine", "reason", "annotation", "annotation-key", constants.MaintenanceAnnotationLabel)
+		return &clustererr.RequeueAfterError{RequeueAfter: constants.DefaultRequeue}
+	}
+
 	defer func() {
 		opErr = actuators.PatchAndHandleError(ctx, "Delete", opErr)
 	}()
@@ -179,6 +189,11 @@ func (a *Actuator) Exists(
 		return false, err
 	}
 
+	if _, ok := machine.Annotations[constants.MaintenanceAnnotationLabel]; ok {
+		ctx.Logger.V(4).Info("skipping operations on machine", "reason", "annotation", "annotation-key", constants.MaintenanceAnnotationLabel)
+		return false, &clustererr.RequeueAfterError{RequeueAfter: constants.DefaultRequeue}
+	}
+
 	defer func() {
 		opErr = actuators.PatchAndHandleError(ctx, "Exists", opErr)
 	}()

diff --git a/pkg/cloud/vsphere/constants/constants.go b/pkg/cloud/vsphere/constants/constants.go
@@ -40,4 +40,8 @@ const (
 	// ReadyAnnotationLabel is the annotation used to indicate a machine and/or
 	// cluster are ready.
 	ReadyAnnotationLabel = "capv.sigs.k8s.io/ready"
+
+	// MaintenanceAnnotationLabel is the annotation used to indicate a machine and/or
+	// cluster are in maintenance mode.
+	MaintenanceAnnotationLabel = "capv.sigs.k8s.io/maintenance"
 )