fix: skip MarkDrain and sent Event when --skip-drain

Author: ngsin

cmd/draino/draino.go

@@ -169,7 +169,8 @@ func main() {
 		kubernetes.NewEventRecorder(cs),
 		kubernetes.WithLogger(log),
 		kubernetes.WithDrainBuffer(*drainBuffer),
-		kubernetes.WithConditionsFilter(*conditions))
+		kubernetes.WithConditionsFilter(*conditions),
+		kubernetes.WithSkipDrainEvent(*skipDrain))
 
 	if *dryRun {
 		h = cache.FilteringResourceEventHandler{
@@ -179,7 +180,8 @@ func main() {
 				kubernetes.NewEventRecorder(cs),
 				kubernetes.WithLogger(log),
 				kubernetes.WithDrainBuffer(*drainBuffer),
-				kubernetes.WithConditionsFilter(*conditions)),
+				kubernetes.WithConditionsFilter(*conditions),
+				kubernetes.WithSkipDrainEvent(*skipDrain)),
 		}
 	}
 

internal/kubernetes/drainer.go

@@ -224,6 +224,12 @@ func (d *APICordonDrainer) Uncordon(n *core.Node, mutators ...nodeMutatorFn) err
 
 // MarkDrain set a condition on the node to mark that that drain is scheduled.
 func (d *APICordonDrainer) MarkDrain(n *core.Node, when, finish time.Time, failed bool) error {
+	// Do nothing if draining is not enabled.
+	if d.skipDrain {
+		d.l.Debug("Skipping drain because draining is disabled")
+		return nil
+	}
+
 	nodeName := n.Name
 	// Refresh the node object
 	freshNode, err := d.c.CoreV1().Nodes().Get(nodeName, meta.GetOptions{})

internal/kubernetes/eventhandler.go

@@ -80,11 +80,19 @@ type DrainingResourceEventHandler struct {
 	buffer                time.Duration
 
 	conditions []SuppliedCondition
+	skipDrain  bool
 }
 
 // DrainingResourceEventHandlerOption configures an DrainingResourceEventHandler.
 type DrainingResourceEventHandlerOption func(d *DrainingResourceEventHandler)
 
+// WithLogger configures if should skip drain in handler
+func WithSkipDrainEvent(skipDrain bool) DrainingResourceEventHandlerOption {
+	return func(h *DrainingResourceEventHandler) {
+		h.skipDrain = skipDrain
+	}
+}
+
 // WithLogger configures a DrainingResourceEventHandler to use the supplied
 // logger.
 func WithLogger(l *zap.Logger) DrainingResourceEventHandlerOption {
@@ -167,6 +175,10 @@ func (h *DrainingResourceEventHandler) HandleNode(n *core.Node) {
 		h.cordon(n, badConditions)
 	}
 
+	if h.skipDrain {
+		return
+	}
+
 	// Let's ensure that a drain is scheduled
 	hasSChedule, failedDrain := h.drainScheduler.HasSchedule(n.GetName())
 	if !hasSChedule {