58 lines
1.9 KiB
Python
58 lines
1.9 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/.
|
|
|
|
|
|
import concurrent.futures as futures
|
|
import logging
|
|
import os
|
|
|
|
import requests
|
|
from taskgraph.util.taskcluster import CONCURRENCY, cancel_task
|
|
|
|
from gecko_taskgraph.util.taskcluster import list_task_group_incomplete_task_ids
|
|
|
|
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_task_ids(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()
|