From dec299b9dbc22a971b963d1f6545b52d96dbdf19 Mon Sep 17 00:00:00 2001
From: Timothy Chavez <timothy.chavez@hp.com>
Date: Thu, 20 Aug 2015 00:14:19 -0500
Subject: [PATCH] Send node label with data status packet

Zuul can use this information to organize its build data by node type.
This will give us more granular insights into load and capacity.

Change-Id: Ibca938fcf8a65facd7e39dab4eb994dfc637722a
---
 .../java/hudson/plugins/gearman/StartJobWorker.java  | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/src/main/java/hudson/plugins/gearman/StartJobWorker.java b/src/main/java/hudson/plugins/gearman/StartJobWorker.java
index ecc0f0b..f3a817a 100644
--- a/src/main/java/hudson/plugins/gearman/StartJobWorker.java
+++ b/src/main/java/hudson/plugins/gearman/StartJobWorker.java
@@ -28,6 +28,7 @@ import hudson.model.Cause;
 import hudson.model.Computer;
 import hudson.model.Hudson;
 import hudson.model.Queue;
+import hudson.model.Node;
 import hudson.model.TextParameterValue;
 import hudson.model.queue.QueueTaskFuture;
 import hudson.slaves.OfflineCause;
@@ -99,6 +100,17 @@ public class StartJobWorker extends AbstractGearmanFunction {
        if (result != null) {
            data.put("result", result.toString());
        }
+
+       ArrayList<String> nodeLabels = new ArrayList<String>();
+       Node node = build.getBuiltOn();
+       if ( node != null ) {
+           Set<LabelAtom> nodeLabelAtoms = node.getAssignedLabels();
+           for (LabelAtom labelAtom : nodeLabelAtoms) {
+               nodeLabels.add(labelAtom.getDisplayName());
+           }
+       }
+       data.put("node_labels", nodeLabels);
+
        Gson gson = new Gson();
        return gson.toJson(data);
    }