pkg/k8sutil: utility function to set labels and annotations in one txn

coreos · Jan 18, 2018 · dedf1a2 · dedf1a2
1 parent 60bd412
commit dedf1a2
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/pkg/k8sutil/metadata.go b/pkg/k8sutil/metadata.go
@@ -80,6 +80,20 @@ func SetNodeAnnotations(nc v1core.NodeInterface, node string, m map[string]strin
 	})
 }
 
+// SetNodeAnnotationsLabels sets all keys in a and l to their respective values in
+// node's annotations and labels, likewise
+func SetNodeAnnotationsLabels(nc v1core.NodeInterface, node string, a, l map[string]string) error {
+	return UpdateNodeRetry(nc, node, func(n *v1api.Node) {
+		for k, v := range a {
+			n.Annotations[k] = v
+		}
+
+		for k, v := range l {
+			n.Labels[k] = v
+		}
+	})
+}
+
 // DeleteNodeLabels deletes all keys in ks
 func DeleteNodeLabels(nc v1core.NodeInterface, node string, ks []string) error {
 	return UpdateNodeRetry(nc, node, func(n *v1api.Node) {