Complete CIPD integration

appengine/swarming/server/task_result.py

 # Copyright 2014 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.
 
 """Task execution result models.
 
 This module doesn't do the scheduling itself. It only describes the entities to
 store tasks results.
 
 - TaskResultSummary represents the overall result for the TaskRequest taking in
@@ skipping 267 matching lines @@
 
   Parent is TaskRunResult. Key id is 1.
 
   Data from the client who uploaded the initial data is not tracked here, since
   it is generally operating on data that is used across multiple different
   tasks.
 
   This entity only exist for isolated tasks. For "raw command task", this entity
   is not stored, since there's nothing to note.
   """
-  # Miscellaneous overhead in second, in addition to the overhead from
-  # isolated_download.duration and isolated_upload.duration.
+  # Miscellaneous overhead in seconds, in addition to the overhead from
+  # package_installation.duration, isolated_download.duration and
+  # isolated_upload.duration
   bot_overhead = ndb.FloatProperty(indexed=False)
+  # Results installing CIPD packages before the task.
+  package_installation = ndb.LocalStructuredProperty(OperationStats)
   # Runtime dependencies download operation before the task.
   isolated_download = ndb.LocalStructuredProperty(OperationStats)
   # Results uploading operation after the task.
   isolated_upload = ndb.LocalStructuredProperty(OperationStats)
 
   @property
   def is_valid(self):
     return self.bot_overhead is not None
 
   def _pre_put_hook(self):
@@ skipping 135 matching lines @@
 
     Returns an empty instance if none is available.
     """
     # Keeps a cache. It's still paying the full latency cost of a DB fetch.
     if not hasattr(self, '_performance_stats_cache'):
       key = None if self.deduped_from else self.performance_stats_key
       # pylint: disable=attribute-defined-outside-init
       stats = (key.get() if key else None) or PerformanceStats()
       stats.isolated_download = stats.isolated_download or OperationStats()
       stats.isolated_upload = stats.isolated_upload or OperationStats()
+      stats.package_installation = (
+          stats.package_installation or OperationStats())
       self._performance_stats_cache = stats
     return self._performance_stats_cache
 
   @property
+  def overhead_package_installation(self):
+    """Returns the overhead from package installation in timedelta."""
+    perf = self.performance_stats
+    if perf.package_installation.duration is not None:
+      return datetime.timedelta(seconds=perf.package_installation.duration)
+
+  @property
   def overhead_isolated_inputs(self):
     """Returns the overhead from isolated setup in timedelta."""
     perf = self.performance_stats
     if perf.isolated_download.duration is not None:
       return datetime.timedelta(seconds=perf.isolated_download.duration)
 
   @property
   def overhead_isolated_outputs(self):
     """Returns the overhead from isolated results upload in timedelta."""
     perf = self.performance_stats
@@ skipping 674 matching lines @@
   if tags:
     # Add TaskResultSummary indexes if desired.
     if sort != 'created_ts':
       raise ValueError(
           'Add needed indexes for sort:%s and tags if desired' % sort)
     tags_filter = TaskResultSummary.tags == tags[0]
     for tag in tags[1:]:
       tags_filter = ndb.AND(tags_filter, TaskResultSummary.tags == tag)
     query = query.filter(tags_filter)
   return _filter_query(TaskResultSummary, query, start, end, sort, state)