Provide runner labels from kube node annotations

oursky · Jul 4, 2023 · 812b2ce · 812b2ce
1 parent e83174b
commit 812b2ce
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/pkg/kube/controller.go b/pkg/kube/controller.go
@@ -131,6 +131,18 @@ func (p *ControllerProvider) RegisterAgent(
 	group := annotations[annotationRunnerGroup]
 	labels := strings.Split(annotations[annotationRunnerLabels], ",")
 
+	if pod.Spec.NodeName != "" {
+		node, err := p.kube.CoreV1().Nodes().Get(r.Context(), pod.Spec.NodeName, metav1.GetOptions{})
+		if err != nil {
+			p.logger.Error("failed to get node",
+				zap.String("nodeName", pod.Spec.NodeName),
+				zap.Error(err))
+		} else {
+			nodeRunnerLabels := strings.Split(node.Annotations[annotationRunnerLabels], ",")
+			labels = append(labels, nodeRunnerLabels...)
+		}
+	}
+
 	agent, err := p.state.makeAgent(pod, hostName)
 	if err != nil {
 		return nil, err