27 lines
800 B
Python
27 lines
800 B
Python
|
import logging
|
||
|
|
|
||
|
|
from buildman.orchestrator import orchestrator_from_config, OrchestratorError
|
||
|
|
from util import slash_join
|
||
|
|
|
||
|
|
|
||
|
|
logger = logging.getLogger(__name__)
|
||
|
|
|
||
|
|
|
||
|
|
CANCEL_PREFIX = 'cancel/'
|
||
|
|
|
||
|
|
|
||
|
|
class OrchestratorCanceller(object):
|
||
|
|
""" An asynchronous way to cancel a build with any Orchestrator. """
|
||
|
|
def __init__(self, config):
|
||
|
|
self._orchestrator = orchestrator_from_config(config, canceller_only=True)
|
||
|
|
|
||
|
|
def try_cancel_build(self, build_uuid):
|
||
|
|
logger.info('Cancelling build %s', build_uuid)
|
||
|
|
cancel_key = slash_join(CANCEL_PREFIX, build_uuid)
|
||
|
|
try:
|
||
|
|
self._orchestrator.set_key_sync(cancel_key, build_uuid, expiration=60)
|
||
|
|
return True
|
||
|
|
except OrchestratorError:
|
||
|
|
logger.exception('Failed to write cancel action to redis with uuid %s', build_uuid)
|
||
|
|
return False
|