probe: add cordon controls inside tagger

This commit is contained in:
Deepak
2020-10-30 23:39:56 +05:30
parent e00b45a982
commit ce2180f24e
2 changed files with 2 additions and 0 deletions

View File

@@ -529,6 +529,7 @@ func (r *Reporter) deregisterControls() {
ScaleUp,
ScaleDown,
CordonNode,
UncordonNode,
}
r.handlerRegistry.Batch(controls, nil)
}

View File

@@ -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
}