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
47
48
49
50
51
52
53
54
55
56
57
58
|
# 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/.
import logging
import os
from concurrent import futures
import requests
from taskgraph.util.taskcluster import (
CONCURRENCY,
cancel_task,
list_task_group_incomplete_tasks,
)
from .registry import register_callback_action
logger = logging.getLogger(__name__)
@register_callback_action(
title="Cancel All",
name="cancel-all",
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):
def do_cancel_task(task_id):
logger.info(f"Cancelling task {task_id}")
try:
cancel_task(task_id, use_proxy=True)
except requests.HTTPError as e:
if e.response.status_code == 409:
# A 409 response indicates that this task is past its deadline. It
# cannot be cancelled at this time, but it's also not running
# anymore, so we can ignore this error.
logger.info(
f"Task {task_id} is past its deadline and cannot be cancelled."
)
return
raise
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(f"Cancelling {len(to_cancel)} tasks")
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()
|
