summaryrefslogtreecommitdiffstats
path: root/taskcluster/taskgraph/actions/cancel_all.py
blob: f4ea1e6f808440607f076af9e0def14fd9717e27 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
# -*- coding: utf-8 -*-

# 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/.

from __future__ import absolute_import, print_function, unicode_literals

import concurrent.futures as futures
import logging
import os

from taskgraph.util.taskcluster import (
    list_task_group_incomplete_tasks,
    cancel_task,
    CONCURRENCY,
)
from .registry import register_callback_action

logger = logging.getLogger(__name__)


@register_callback_action(
    title='Cancel All',
    name='cancel-all',
    generic=True,
    symbol='cAll',
    description=(
        'Cancel all running and pending tasks created by the decision task '
        'this action task is associated with.'
    ),
    order=400,
    context=[]
)
def cancel_all_action(parameters, graph_config, input, task_group_id, task_id, task):
    def do_cancel_task(task_id):
        logger.info('Cancelling task {}'.format(task_id))
        cancel_task(task_id, use_proxy=True)

    own_task_id = os.environ.get('TASK_ID', '')
    to_cancel = [t for t in list_task_group_incomplete_tasks(task_group_id) if t != own_task_id]
    logger.info("Cancelling {} tasks".format(len(to_cancel)))
    with futures.ThreadPoolExecutor(CONCURRENCY) as e:
        cancel_futs = [e.submit(do_cancel_task, t) for t in to_cancel]
        for f in futures.as_completed(cancel_futs):
            f.result()