Label auto-pods with job name and number

Author: aranabhat

pipelines/lib/util.groovy

@@ -444,6 +444,66 @@ def saveCache(
 }
 
 
+def labelPod(){
+  if (env.NODE_NAME && (env.NODE_NAME =~ /(manager|snowflake)/)) {
+        echo "Skipping pod label: ${env.NODE_NAME} is a static/snowflake node."
+        return
+  }
+
+  def JOB = env.JOB_NAME ? env.JOB_NAME.replace('/', '.') : "unknown"
+    def BUILD_NUMBER = env.BUILD_NUMBER ? env.BUILD_NUMBER.toString() : "unknown"
+
+  def labels = [
+        "jenkins-job": JOB,
+        "jenkins-build": BUILD_NUMBER
+    ]
+  
+  def upstream = currentBuild.upstreamBuilds
+  def upstreamFields = ""
+  if (upstream) {
+    def tJob = upstream[0].projectName.replace('/', '.')
+    def tNum = upstream[0].number.toString()
+    echo "Upstream Trigger: ${tJob} #${tNum}"
+    labels["triggered-by"] = tJob
+    labels["triggered-build"] = tNum
+  }
+
+  def jsonLabels = groovy.json.JsonOutput.toJson([metadata: [labels: labels]])
+  echo "jsonLabels: ${jsonLabels}"
+
+  bash """
+  set -eu
+  set +x
+
+  SA=/var/run/secrets/kubernetes.io/serviceaccount
+  [ -r "\$SA/token" ] || { echo "not in k8s; skipping pod label"; exit 0; }
+  NS=\$(cat \$SA/namespace)
+  POD=\${HOSTNAME}
+  TOKEN=\$(cat \$SA/token)
+  CA=\$SA/ca.crt
+
+  cat <<EOF > /tmp/patch.json
+${jsonLabels}
+EOF
+
+  if curl -sS --fail \\
+    --cacert "\$CA" \\
+    -H "Authorization: Bearer \$TOKEN" \\
+    -H "Content-Type: application/merge-patch+json" \\
+    -X PATCH \\
+    "https://kubernetes.default.svc/api/v1/namespaces/\$NS/pods/\$POD" \\
+    --data-binary @/tmp/patch.json >/dev/null
+  then
+    echo "labeled for pod \$NS/\$POD"
+  else
+    echo "pod label skipped for \$NS/\$POD"
+    exit 0
+  fi
+  set -x
+  """
+
+}
+
 /**
  * Run a lsstsw build.
  *
@@ -2522,6 +2582,7 @@ def void nodeWrap(Closure run) {
 def void nodeWrap(String label, Closure run) {
   node(label) {
     printK8sVars()
+    labelPod()
     run()
   }
 }