clientutil: add func to find control plane nodes

add utilities to getch all the controlplane nodes, similarly to worker node which we support since forever. Signed-off-by: Francesco Romani <[email protected]>
k8stopologyawareschedwg · Jul 31, 2024 · c5094bb · c5094bb
1 parent da17968
commit c5094bb
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/pkg/clientutil/nodes/nodes.go b/pkg/clientutil/nodes/nodes.go
@@ -28,15 +28,19 @@ import (
 )
 
 const (
-	// RoleWorker contains the worker role
-	RoleWorker = "worker"
+	RoleControlPlane = "control-plane"
+	RoleWorker       = "worker"
 )
 
 const (
 	// LabelRole contains the key for the role label
 	LabelRole = "node-role.kubernetes.io"
 )
 
+func GetControlPlane(env *deployer.Environment) ([]corev1.Node, error) {
+	return GetByRole(env, RoleControlPlane)
+}
+
 func GetWorkers(env *deployer.Environment) ([]corev1.Node, error) {
 	return GetByRole(env, RoleWorker)
 }