swarming: send task tags to event_mon

appengine/swarming/event_mon_metrics.py

 # Copyright 2016 The LUCI Authors. All rights reserved.
 # Use of this source code is governed under the Apache License, Version 2.0
 # that can be found in the LICENSE file.
 
 import logging
 import time
 
 import gae_event_mon
 
 from server import task_result
 
 
 DIMENSIONS = (
     ('cores', int),
     ('cpu', unicode),
     ('device_os', unicode),
     ('device_type', unicode),
     ('gpu', unicode),
     ('hidpi', unicode),
     ('machine_type', unicode),
     ('os', unicode),
     ('pool', unicode),
     ('xcode_version', unicode),
     ('zone', unicode),
 )
 
 
+TAGS = {
+    'build_id': ['buildnumber'],
+    'buildername': ['buildername'],
+    'codereview': ['rietveld'],
+    'master': ['master'],
+    'name': ['name'],
+    'patch_project': ['patch_project'],
+    'project': ['project'],
+    'purpose': ['purpose'],
+    'slavename': ['slavename'],
+    'stepname': ['stepname'],
+}
+
+
 def _to_timestamp(dt):
   return int(time.mktime(dt.timetuple()))
 
 
 def _files_ref_to_proto(files_ref, proto):
   if files_ref.isolated:
     proto.isolated = files_ref.isolated
   if files_ref.isolatedserver:
     proto.isolatedserver = files_ref.isolatedserver
   if files_ref.namespace:
@@ skipping 95 matching lines @@
     _files_ref_to_proto(
         summary.outputs_ref, event.proto.swarming_task_event.outputs_ref)
 
   event.proto.swarming_task_event.cost_usd = summary.cost_usd
   if summary.cost_saved_usd:
     event.proto.swarming_task_event.cost_saved_usd = summary.cost_saved_usd
   if summary.deduped_from:
     event.proto.swarming_task_event.deduped_from = summary.deduped_from
   event.proto.swarming_task_event.try_number = summary.try_number
 
+  for tag in summary.tags:
+    if ':' not in tag:
+      logging.error('Unexpected tag: %r', tag)
+      continue
+    name, value = tag.split(':', 1)
+    for event_tag, task_tags in TAGS.iteritems():
+      if name in task_tags:
+        getattr(event.proto.swarming_task_event.tags, event_tag).append(value)
+
 
 def initialize():
   gae_event_mon.initialize('swarming')
 
 
 def send_task_event(summary):
   """Sends an event_mon event about a swarming task.
 
   Currently implemented as sending a HTTP request.
 
   Args:
     summary: TaskResultSummary object.
   """
   # Isolate rest of the app from monitoring pipeline issues. They should
   # not cause outage of swarming.
   try:
     event = gae_event_mon.Event('POINT')
     _task_summary_to_proto(summary, event)
     event.send()
   except Exception:
     logging.exception('Caught exception while sending event')