forked from mirrors/gecko-dev
7b13402d7e
Differential Revision: https://phabricator.services.mozilla.com/D1286 --HG-- extra : rebase_source : 3cb1080e5a94e3e9e5b5a3f6909b0e8d636de09c extra : source : 4a182c3447fa01c75fa8a231e067bebd83683da9
44 lines
1.4 KiB
Python
44 lines
1.4 KiB
Python
# This Source Code Form is subject to the terms of the Mozilla Public
|
|
# License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
# file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
|
"""
|
|
Adjust dependencies to not exceed MAX_DEPENDENCIES
|
|
"""
|
|
|
|
from __future__ import absolute_import, print_function, unicode_literals
|
|
from copy import deepcopy
|
|
|
|
from taskgraph.transforms.base import TransformSequence
|
|
import taskgraph.transforms.release_deps as release_deps
|
|
from taskgraph.util.treeherder import add_suffix
|
|
from taskgraph import MAX_DEPENDENCIES
|
|
|
|
transforms = TransformSequence()
|
|
|
|
|
|
def yield_job(orig_job, deps, count):
|
|
job = deepcopy(orig_job)
|
|
job['dependencies'] = deps
|
|
job['name'] = "{}-{}".format(orig_job['name'], count)
|
|
if 'treeherder' in job:
|
|
job['treeherder']['symbol'] = add_suffix(
|
|
job['treeherder']['symbol'], "-{}".format(count))
|
|
|
|
return job
|
|
|
|
|
|
@transforms.add
|
|
def add_dependencies(config, jobs):
|
|
for job in release_deps.add_dependencies(config, jobs):
|
|
count = 1
|
|
deps = {}
|
|
|
|
for dep_label in job['dependencies'].keys():
|
|
deps[dep_label] = dep_label
|
|
if len(deps) == MAX_DEPENDENCIES:
|
|
yield yield_job(job, deps, count)
|
|
deps = {}
|
|
count += 1
|
|
if deps:
|
|
yield yield_job(job, deps, count)
|
|
count += 1
|