From ce2180f24eb2a8741b9e04066fb7724ab2c2b27f Mon Sep 17 00:00:00 2001
From: Deepak <sah.sslpu@gmail.com>
Date: Fri, 30 Oct 2020 23:39:56 +0530
Subject: [PATCH] probe: add cordon controls inside tagger

---
 probe/kubernetes/controls.go | 1 +
 probe/kubernetes/reporter.go | 1 +
 2 files changed, 2 insertions(+)

diff --git a/probe/kubernetes/controls.go b/probe/kubernetes/controls.go
index 8d7f918f5..2ef12a184 100644
--- a/probe/kubernetes/controls.go
+++ b/probe/kubernetes/controls.go
@@ -529,6 +529,7 @@ func (r *Reporter) deregisterControls() {
 		ScaleUp,
 		ScaleDown,
 		CordonNode,
+		UncordonNode,
 	}
 	r.handlerRegistry.Batch(controls, nil)
 }
diff --git a/probe/kubernetes/reporter.go b/probe/kubernetes/reporter.go
index ad842a270..9ed12cb68 100644
--- a/probe/kubernetes/reporter.go
+++ b/probe/kubernetes/reporter.go
@@ -306,6 +306,7 @@ func (r *Tagger) Tag(rpt report.Report) (report.Report, error) {
 		controls := append(n.ActiveControls(), CordonNode, UncordonNode)
 		rpt.Host.Nodes[id] = n.WithLatestActiveControls(controls...)
 	}
+	rpt.Host.Controls.AddControls(CordonControl)
 	return rpt, nil
 }