Index: appengine/swarming/handlers_bot.py |
diff --git a/appengine/swarming/handlers_bot.py b/appengine/swarming/handlers_bot.py |
index b41fb60f416d11742d50993db18ec7c413893978..7bcce920b2dbe429c9410db4d03586fde2990dc2 100644 |
--- a/appengine/swarming/handlers_bot.py |
+++ b/appengine/swarming/handlers_bot.py |
@@ -575,7 +575,7 @@ class BotTaskUpdateHandler(_BotApiHandler): |
ACCEPTED_KEYS = { |
u'bot_overhead', u'cipd_stats', u'cost_usd', u'duration', u'exit_code', |
u'hard_timeout', u'id', u'io_timeout', u'isolated_stats', u'output', |
- u'output_chunk_start', u'outputs_ref', u'task_id', |
+ u'output_chunk_start', u'outputs_ref', u'cipd_pins', u'task_id', |
M-A Ruel
2016/08/24 02:13:44
keep ordered lists ordered
iannucci
2016/08/24 02:31:57
Done.
|
} |
REQUIRED_KEYS = {u'id', u'task_id'} |
@@ -608,6 +608,7 @@ class BotTaskUpdateHandler(_BotApiHandler): |
output = request.get('output') |
output_chunk_start = request.get('output_chunk_start') |
outputs_ref = request.get('outputs_ref') |
+ cipd_pins = request.get('cipd_pins') |
if (isolated_stats or cipd_stats) and bot_overhead is None: |
ereporter2.log_request( |
@@ -661,6 +662,9 @@ class BotTaskUpdateHandler(_BotApiHandler): |
if outputs_ref: |
outputs_ref = task_request.FilesRef(**outputs_ref) |
+ if cipd_pins: |
+ cipd_pins = [task_request.CipdPackage(**pin) for pin in cipd_pins] |
+ |
try: |
state = task_scheduler.bot_update_task( |
run_result_key=run_result_key, |
@@ -673,6 +677,7 @@ class BotTaskUpdateHandler(_BotApiHandler): |
io_timeout=io_timeout, |
cost_usd=cost_usd, |
outputs_ref=outputs_ref, |
+ cipd_pins=cipd_pins, |
performance_stats=performance_stats) |
if not state: |
logging.info('Failed to update, please retry') |